xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-03-26 18:05:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update tests/compile/fullgraph/test_full_graph.py

Browse Source 
Co-authored-by: Luka Govedič <ProExpertProg@users.noreply.github.com>
Signed-off-by: baonudesifeizhai <85092850+baonudesifeizhai@users.noreply.github.com>
This commit is contained in:
baonudesifeizhai 2025-12-23 15:18:39 -05:00 committed by GitHub
parent 919ad836d3
commit 226c3dc98a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/compile/fullgraph/test_full_graph.py
View File

@ -195,26 +195,6 @@ def test_custom_compile_config(
):
pytest.skip("inductor graph partition is only available in PyTorch 2.9+")
# For get_raw_stream patch test: verify version and log context
if compilation_config.compile_sizes and len(compilation_config.compile_sizes) > 1:
# Verify torch version >= 2.9.0 for compile_sizes autotune
if not is_torch_equal_or_newer("2.9.0"):
pytest.skip(
f"compile_sizes autotune requires torch >= 2.9.0, "
f"got {torch.__version__}"
)
# Log version context for get_raw_stream patch testing
is_torch_2_9 = is_torch_equal("2.9.0") or is_torch_equal("2.9.1")
version_context = (
"2.9.x (get_raw_stream patch applied)"
if is_torch_2_9
else "2.10+ (get_raw_stream patch not needed)"
)
print(
f"Testing compile_sizes with torch {torch.__version__} ({version_context})"
)
print(f"MODEL={model}")
run_model(compilation_config, model, **model_kwargs)