Unverified Commit b90b0852 authored by Richard Zou's avatar Richard Zou Committed by GitHub
Browse files

[easy] Print number of needed GPUs in skip message (#17594)


Signed-off-by: default avatarrzou <zou3519@gmail.com>
parent 9352cdb5
......@@ -103,7 +103,8 @@ def test_compile_correctness(
method = test_setting.method
fullgraph = test_setting.fullgraph
if cuda_device_count_stateless() != pp_size * tp_size:
pytest.skip("Not correct CUDA devices for the test.")
pytest.skip(f"Need exactly {pp_size}*{tp_size} CUDA gpus but got "
f"{cuda_device_count_stateless()}")
with monkeypatch.context() as m:
m.setenv("VLLM_ATTENTION_BACKEND", attn_backend)
......
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