"docs/assets/vscode:/vscode.git/clone" did not exist on "2ec1c3f503988160d7303766da1458d19348f6a5"
Unverified Commit 77f0d465 authored by Richard Zou's avatar Richard Zou Committed by GitHub
Browse files

[BugFix] Allow use_cudagraph to work with dynamic VLLM_USE_V1 (#19390)


Signed-off-by: default avatarrzou <zou3519@gmail.com>
parent 22c3c0aa
......@@ -11,6 +11,16 @@ from vllm.config import (CompilationConfig, CompilationLevel, VllmConfig,
from .piecewise.test_simple import SillyModel
def test_use_cudagraphs_dynamic(monkeypatch):
assert vllm.envs.VLLM_USE_V1
vllm_config = VllmConfig()
assert vllm_config.compilation_config.use_cudagraph
monkeypatch.setenv('VLLM_USE_V1', '0')
vllm_config = VllmConfig()
assert not vllm_config.compilation_config.use_cudagraph
@pytest.mark.parametrize("enabled", [True, False])
def test_use_cudagraphs(enabled):
assert vllm.envs.VLLM_USE_V1
......
......@@ -3931,7 +3931,7 @@ class CompilationConfig:
constructor, e.g. `CompilationConfig(inductor_passes={"a": func})`."""
# CudaGraph compilation
use_cudagraph: bool = envs.VLLM_USE_V1
use_cudagraph: bool = field(default_factory=lambda: envs.VLLM_USE_V1)
"""Whether to use cudagraph inside compilation.
- False: cudagraph inside compilation is not used.
- True: cudagraph inside compilation is used. It requires
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment