From b90b0852e9011b48c3e8ff9009aa57e9428350d1 Mon Sep 17 00:00:00 2001 From: Richard Zou Date: Fri, 2 May 2025 18:27:43 -0400 Subject: [PATCH] [easy] Print number of needed GPUs in skip message (#17594) Signed-off-by: rzou --- tests/compile/test_basic_correctness.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/compile/test_basic_correctness.py b/tests/compile/test_basic_correctness.py index 0b76779b3a75..b6b45d1cbe88 100644 --- a/tests/compile/test_basic_correctness.py +++ b/tests/compile/test_basic_correctness.py @@ -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)