From 2018021c1754ff744df277bdea635f3926e9e8d7 Mon Sep 17 00:00:00 2001 From: Federico Stagni Date: Mon, 27 Nov 2023 17:20:29 +0100 Subject: [PATCH] feat: set env variable OMP_NUM_THREADS --- src/DIRAC/Resources/Computing/InProcessComputingElement.py | 2 +- src/DIRAC/Resources/Computing/SingularityComputingElement.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/DIRAC/Resources/Computing/InProcessComputingElement.py b/src/DIRAC/Resources/Computing/InProcessComputingElement.py index 39b3976284e..618973b81da 100755 --- a/src/DIRAC/Resources/Computing/InProcessComputingElement.py +++ b/src/DIRAC/Resources/Computing/InProcessComputingElement.py @@ -35,7 +35,7 @@ def submitJob(self, executableFile, proxy=None, inputs=None, **kwargs): """ payloadEnv = dict(os.environ) if mp_threads := kwargs.get("numberOfProcessors"): - payloadEnv["OMP_NUM_THREADS"] = mp_threads + payloadEnv["OMP_NUM_THREADS"] = str(mp_threads) payloadProxy = "" if proxy: self.log.verbose("Setting up proxy for payload") diff --git a/src/DIRAC/Resources/Computing/SingularityComputingElement.py b/src/DIRAC/Resources/Computing/SingularityComputingElement.py index f31f053fdef..f1eee9f373d 100644 --- a/src/DIRAC/Resources/Computing/SingularityComputingElement.py +++ b/src/DIRAC/Resources/Computing/SingularityComputingElement.py @@ -94,6 +94,7 @@ r"Xrd.*", r"DIRAC_.*", r"BEARER_TOKEN.*", + r"OMP_NUM_TREADS", ] ENV_VAR_WHITELIST = re.compile(r"^(" + r"|".join(ENV_VAR_WHITELIST) + r")$") @@ -374,6 +375,9 @@ def submitJob(self, executableFile, proxy=None, **kwargs): baseDir = ret["baseDir"] tmpDir = ret["tmpDir"] + if mp_threads := kwargs.get("numberOfProcessors"): + os.environ["OMP_NUM_THREADS"] = str(mp_threads) + if proxy: payloadProxyLoc = ret["proxyLocation"]