set default blas num threads to Sys.CPU_THREADS / 2 (#45412)

Set default blas num threads to Sys.CPU_THREADS / 2 in absence of OPENBLAS_NUM_THREADS Co-authored-by: SamuraiAku <[email protected]>
JuliaLang · May 22, 2022 · 390503e · 390503e
1 parent a5438f9
commit 390503e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stdlib/LinearAlgebra/src/LinearAlgebra.jl b/stdlib/LinearAlgebra/src/LinearAlgebra.jl
@@ -580,6 +580,10 @@ function __init__()
     end
     # register a hook to disable BLAS threading
     Base.at_disable_library_threading(() -> BLAS.set_num_threads(1))
+
+    if !haskey(ENV, "OPENBLAS_NUM_THREADS")
+        BLAS.set_num_threads(max(1, Sys.CPU_THREADS ÷ 2))
+    end
 end
 
 end # module LinearAlgebra