set number of openblas threads to 1 while precompiling

JuliaLang · Sep 16, 2022 · 76b5914 · 76b5914
1 parent 35191ad
commit 76b5914
Showing 1 changed file with 16 additions and 14 deletions.
diff --git a/base/loading.jl b/base/loading.jl
@@ -1683,21 +1683,23 @@ function create_expr_cache(pkg::PkgId, input::String, output::String, concrete_d
     end
     deps_eltype = sprint(show, eltype(concrete_deps); context = :module=>nothing)
     deps = deps_eltype * "[" * join(deps_strs, ",") * "]"
-    trace = isassigned(PRECOMPILE_TRACE_COMPILE) ? `--trace-compile=$(PRECOMPILE_TRACE_COMPILE[])` : ``
-    io = open(pipeline(`$(julia_cmd()::Cmd) -O0
-                       --output-ji $output --output-incremental=yes
-                       --startup-file=no --history-file=no --warn-overwrite=yes
-                       --color=$(have_color === nothing ? "auto" : have_color ? "yes" : "no")
-                       $trace
-                       -`, stderr = internal_stderr, stdout = internal_stdout),
-              "w", stdout)
-    # write data over stdin to avoid the (unlikely) case of exceeding max command line size
-    write(io.in, """
-        Base.include_package_for_output($(pkg_str(pkg)), $(repr(abspath(input))), $(repr(depot_path)), $(repr(dl_load_path)),
+    withenv("OPENBLAS_NUM_THREADS" => 1) do
+        trace = isassigned(PRECOMPILE_TRACE_COMPILE) ? `--trace-compile=$(PRECOMPILE_TRACE_COMPILE[])` : ``
+        io = open(pipeline(`$(julia_cmd()::Cmd) -O0
+                        --output-ji $output --output-incremental=yes
+                        --startup-file=no --history-file=no --warn-overwrite=yes
+                        --color=$(have_color === nothing ? "auto" : have_color ? "yes" : "no")
+                        $trace
+                        -`, stderr = internal_stderr, stdout = internal_stdout),
+                "w", stdout)
+        # write data over stdin to avoid the (unlikely) case of exceeding max command line size
+        write(io.in, """
+        @time Base.include_package_for_output($(pkg_str(pkg)), $(repr(abspath(input))), $(repr(depot_path)), $(repr(dl_load_path)),
             $(repr(load_path)), $deps, $(repr(source_path(nothing))))
-        """)
-    close(io.in)
-    return io
+            """)
+        close(io.in)
+        return io
+    end
 end
 
 function compilecache_dir(pkg::PkgId)