fix parallel peakflop usage (#51757)

This is required now once Distributed is not in the sysimage. Fixes #51756
JuliaLang · Oct 20, 2023 · 795d8d7 · 795d8d7
1 parent f2df1b4
commit 795d8d7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/stdlib/LinearAlgebra/src/LinearAlgebra.jl b/stdlib/LinearAlgebra/src/LinearAlgebra.jl
@@ -619,7 +619,9 @@ function peakflops(n::Integer=4096; eltype::DataType=Float64, ntrials::Integer=3
  if parallel
  let Distributed = Base.require(Base.PkgId(
  Base.UUID((0x8ba89e20_285c_5b6f, 0x9357_94700520ee1b)), "Distributed"))
- return sum(Distributed.pmap(peakflops, fill(n, Distributed.nworkers())))
+ nworkers = @invokelatest Distributed.nworkers()
+ results = @invokelatest Distributed.pmap(peakflops, fill(n, nworkers))
+ return sum(results)
  end
  else
  return 2*Float64(n)^3 / minimum(t)