We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d80dbd3 commit a0b2cd0Copy full SHA for a0b2cd0
tests/compile/test_basic_correctness.py
@@ -103,7 +103,8 @@ def test_compile_correctness(
103
method = test_setting.method
104
fullgraph = test_setting.fullgraph
105
if cuda_device_count_stateless() != pp_size * tp_size:
106
- pytest.skip("Not correct CUDA devices for the test.")
+ pytest.skip(f"Need exactly {pp_size}*{tp_size} CUDA gpus but got "
107
+ f"{cuda_device_count_stateless()}")
108
109
with monkeypatch.context() as m:
110
m.setenv("VLLM_ATTENTION_BACKEND", attn_backend)
0 commit comments