Skip to content

Commit a0b2cd0

Browse files
zou3519mawong-amd
authored andcommitted
[easy] Print number of needed GPUs in skip message (vllm-project#17594)
Signed-off-by: rzou <[email protected]>
1 parent d80dbd3 commit a0b2cd0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/compile/test_basic_correctness.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,8 @@ def test_compile_correctness(
103103
method = test_setting.method
104104
fullgraph = test_setting.fullgraph
105105
if cuda_device_count_stateless() != pp_size * tp_size:
106-
pytest.skip("Not correct CUDA devices for the test.")
106+
pytest.skip(f"Need exactly {pp_size}*{tp_size} CUDA gpus but got "
107+
f"{cuda_device_count_stateless()}")
107108

108109
with monkeypatch.context() as m:
109110
m.setenv("VLLM_ATTENTION_BACKEND", attn_backend)

0 commit comments

Comments
 (0)