"vscode:/vscode.git/clone" did not exist on "d69ff0cbbbb07b571eeea62b4e2ce87b91cea387"
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( ...@@ -103,7 +103,8 @@ def test_compile_correctness(
method = test_setting.method method = test_setting.method
fullgraph = test_setting.fullgraph fullgraph = test_setting.fullgraph
if cuda_device_count_stateless() != pp_size * tp_size: 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: with monkeypatch.context() as m:
m.setenv("VLLM_ATTENTION_BACKEND", attn_backend) 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