From 30c3a11b6d6cd05edc8b074028338b98affc4511 Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Sun, 30 Jul 2023 12:29:50 +0800 Subject: [PATCH] ban numpy multithreading --- csst_common/pipeline.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/csst_common/pipeline.py b/csst_common/pipeline.py index b2131aa..851df60 100644 --- a/csst_common/pipeline.py +++ b/csst_common/pipeline.py @@ -74,7 +74,10 @@ class Pipeline: def ban_multithreading(self): os.environ["OMP_NUM_THREADS"] = "1" + os.environ["OPENBLAS_NUM_THREADS"] = "1" os.environ["MKL_NUM_THREADS"] = "1" + os.environ["VECLIB_MAXIMUM_THREADS"] = "1" + os.environ["NUMEXPR_NUM_THREADS"] = "1" def set_test_env(self): os.environ["CSST_DFS_API_MODE"] = "cluster" -- GitLab