diff --git a/vllm/platforms/cpu.py b/vllm/platforms/cpu.py index 179ee6a7d24..a9216c2322e 100644 --- a/vllm/platforms/cpu.py +++ b/vllm/platforms/cpu.py @@ -115,6 +115,9 @@ def check_and_update_config(cls, vllm_config: VllmConfig) -> None: # Environment variables for CPU executor # + # Set default threads num for OpenMP parallel + os.environ["OMP_NUM_THREADS"] = str(torch.get_num_threads()) + # Disable torch async compiling which won't work with daemonic processes os.environ["TORCHINDUCTOR_COMPILE_THREADS"] = "1"