obscura/vllm
2
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2026-08-22 21:50:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm][CI] Skip unbacked dynamic shapes tests on PyTorch < 2.11 (#44256)

Browse Source 
Signed-off-by: JartX <[email protected]>
This commit is contained in:
JartX
2026-06-01 19:09:01 -05:00
committed by GitHub
parent 8c3cc98cff
commit 48c0d13e65
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/compile/test_dynamic_shapes_compilation.py
+9
View File
@@ -55,6 +55,15 @@ def test_dynamic_shapes_compilation(
evaluate_guards,
):
"""Test that all dynamic shapes types compile successfully"""
if shapes_type == DynamicShapesType.UNBACKED and not is_torch_equal_or_newer(
"2.11.0"
):
# NOTE[ROCm]: shape_id (used by Qwen2/Llama to relate input dims) only
# landed in torch 2.11, but the ROCm CI still runs torch 2.10.x. On
# older torch there's no way to express it, so unbacked shapes go
# data-dependent and compilation blows up -- nothing to test.
pytest.skip("unbacked dynamic shapes with shape_id require torch>=2.11")
if evaluate_guards and shapes_type == DynamicShapesType.UNBACKED:
pytest.skip("unbacked dynamic shapes do not add guards")