xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-14 05:04:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CI/Build][Bugfix] Fix CPU backend default threads num (#13077)

Browse Source
This commit is contained in:
Li, Jiang 2025-02-12 00:55:56 +08:00 committed by GitHub
parent 2b25b7d2e1
commit 565c1efa65
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/platforms/cpu.py
View File

@ -115,6 +115,9 @@ class CpuPlatform(Platform):
# Environment variables for CPU executor # 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 # Disable torch async compiling which won't work with daemonic processes
os.environ["TORCHINDUCTOR_COMPILE_THREADS"] = "1" os.environ["TORCHINDUCTOR_COMPILE_THREADS"] = "1"