Skip to content

Commit 7c2134b

Browse files
authored
[torch.compile] force inductor threads (#10620)
Signed-off-by: Jee Jee Li <[email protected]>
1 parent a30a605 commit 7c2134b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

vllm/plugins/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
from contextlib import contextmanager
44
from typing import TYPE_CHECKING, Optional
55

6+
import torch
7+
68
import vllm.envs as envs
79

810
if TYPE_CHECKING:
@@ -26,7 +28,8 @@ def load_general_plugins():
2628

2729
# see https://github.com/vllm-project/vllm/issues/10480
2830
os.environ['TORCHINDUCTOR_COMPILE_THREADS'] = '1'
29-
31+
# see https://github.com/vllm-project/vllm/issues/10619
32+
torch._inductor.config.compile_threads = 1
3033
global plugins_loaded
3134
if plugins_loaded:
3235
return

0 commit comments

Comments
 (0)