From f77b8a6993284e205cddf5a041d6337d3b2d23d3 Mon Sep 17 00:00:00 2001
From: Elliot Saba <staticfloat@gmail.com>
Date: Mon, 5 Dec 2022 14:40:25 -0500
Subject: [PATCH] Set `OPENBLAS_NUM_THREADS=1` on local Distributed workers
 (JuliaLang/julia#47803)

This should prevent LinearAlgebra from trying to increase our OpenBLAS
thread count in its `__init__()` method when we're not trying to enable
threaded BLAS.
---
 src/managers.jl | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/managers.jl b/src/managers.jl
index 03adfd1371d15c..57f58598e85dcf 100644
--- a/src/managers.jl
+++ b/src/managers.jl
@@ -487,6 +487,13 @@ function launch(manager::LocalManager, params::Dict, launched::Array, c::Conditi
     if get(env, "JULIA_DEPOT_PATH", nothing) === nothing
         env["JULIA_DEPOT_PATH"] = join(DEPOT_PATH, pathsep)
     end
+
+    # If we haven't explicitly asked for threaded BLAS, prevent OpenBLAS from starting
+    # up with multiple threads, thereby sucking up a bunch of wasted memory on Windows.
+    if !params[:enable_threaded_blas] &&
+       get(env, "OPENBLAS_NUM_THREADS", nothing) === nothing
+        env["OPENBLAS_NUM_THREADS"] = "1"
+    end
     # Set the active project on workers using JULIA_PROJECT.
     # Users can opt-out of this by (i) passing `env = ...` or (ii) passing
     # `--project=...` as `exeflags` to addprocs(...).