feat: set env variable OMP_NUM_THREADS

DIRACGrid · Nov 27, 2023 · df4e1ff · df4e1ff
1 parent eb8fe88
commit df4e1ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/DIRAC/Resources/Computing/InProcessComputingElement.py b/src/DIRAC/Resources/Computing/InProcessComputingElement.py
@@ -34,6 +34,8 @@ def submitJob(self, executableFile, proxy=None, inputs=None, **kwargs):
  :return: S_OK(payload exit code) / S_ERROR() if submission issue
  """
  payloadEnv = dict(os.environ)
+ if mp_threads := kwargs.get("numberOfProcessors"):
+ payloadEnv["OMP_NUM_THREADS"] = mp_threads
  payloadProxy = ""
  if proxy:
  self.log.verbose("Setting up proxy for payload")

diff --git a/src/DIRAC/Resources/Computing/PoolComputingElement.py b/src/DIRAC/Resources/Computing/PoolComputingElement.py
@@ -134,6 +134,7 @@ def submitJob(self, executableFile, proxy=None, inputs=None, **kwargs):
  # Here we define task kwargs: adding complex objects like thread.Lock can trigger errors in the task
  taskKwargs = {"InnerCESubmissionType": self.innerCESubmissionType}
  taskKwargs["jobDesc"] = kwargs.get("jobDesc", {})
+ taskKwargs["numberOfProcessors"] = kwargs.get("numberOfProcessors")
 
  # Submission
  future = self.pPool.submit(executeJob, executableFile, proxy, self.taskID, inputs, **taskKwargs)