[Bugfix] set OMP_NUM_THREADS to 1 by default for multiprocessing (#6109)

Signed-off-by: Travis Johnson <[email protected]>
Co-authored-by: Nick Hill <[email protected]>

Author: tjohnson31415

vllm/executor/multiproc_gpu_executor.py

@@ -37,6 +37,11 @@ def _init_executor(self) -> None:
         # Disable torch async compiling which won't work with daemonic processes
         os.environ["TORCHINDUCTOR_COMPILE_THREADS"] = "1"
 
+        # Set OMP_NUM_THREADS to 1 if it is not set explicitly, avoids CPU
+        # contention amongst the shards
+        if "OMP_NUM_THREADS" not in os.environ:
+            os.environ["OMP_NUM_THREADS"] = "1"
+
         assert world_size <= cuda_device_count_stateless(), (
             "please set tensor_parallel_size to less than max local gpu count")