Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
7c2134be
Unverified
Commit
7c2134be
authored
Nov 25, 2024
by
Jee Jee Li
Committed by
GitHub
Nov 24, 2024
Browse files
[torch.compile] force inductor threads (#10620)
Signed-off-by:
Jee Jee Li
<
pandaleefree@gmail.com
>
parent
a30a605d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
vllm/plugins/__init__.py
vllm/plugins/__init__.py
+4
-1
No files found.
vllm/plugins/__init__.py
View file @
7c2134be
...
...
@@ -3,6 +3,8 @@ import os
from
contextlib
import
contextmanager
from
typing
import
TYPE_CHECKING
,
Optional
import
torch
import
vllm.envs
as
envs
if
TYPE_CHECKING
:
...
...
@@ -26,7 +28,8 @@ def load_general_plugins():
# see https://github.com/vllm-project/vllm/issues/10480
os
.
environ
[
'TORCHINDUCTOR_COMPILE_THREADS'
]
=
'1'
# see https://github.com/vllm-project/vllm/issues/10619
torch
.
_inductor
.
config
.
compile_threads
=
1
global
plugins_loaded
if
plugins_loaded
:
return
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment