Skip to content

Commit fce137e

Browse files
committed
add pytest flag
Signed-off-by: jiant <[email protected]>
1 parent dd3d55a commit fce137e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/integration/defs/accuracy/test_llm_api_pytorch.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4181,6 +4181,7 @@ def test_bf16_4gpu(self, tp_size, pp_size, ep_size, cuda_graph,
41814181
task.evaluate(llm)
41824182

41834183
@skip_pre_blackwell
4184+
@pytest.mark.skip_less_device(4)
41844185
@pytest.mark.parametrize("moe_backend", ["CUTLASS", "TRTLLM"],
41854186
ids=["cutlass", "trtllm"])
41864187
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)