From c4ae6c7c51869d78206ea820a93252b6cbe43f17 Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Wed, 12 Jul 2023 12:38:45 +0800 Subject: [PATCH] ban numpy multithreading --- csst_common/pipeline.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/csst_common/pipeline.py b/csst_common/pipeline.py index bd8ad7c..bc07096 100644 --- a/csst_common/pipeline.py +++ b/csst_common/pipeline.py @@ -1,5 +1,5 @@ import json -import os.path +import os import subprocess import warnings @@ -59,6 +59,10 @@ class Pipeline: if filter_warnings: self.filter_warnings() + def ban_multithreading(self): + os.environ["OMP_NUM_THREADS"] = "1" + os.environ["MKL_NUM_THREADS"] = "1" + def set_test_env(self): os.environ["CSST_DFS_API_MODE"] = "cluster" os.environ["CSST_DFS_GATEWAY"] = "172.24.27.2:30880" -- GitLab