Skip to content

Commit 4b52054

Browse files
authored
[https://nvbugs/5541145][fix] Remove DeepSeekR1 test case from H20 to prevent OOM (#8610)
Signed-off-by: Jie Li <[email protected]>
1 parent d9d898e commit 4b52054

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
@@ -2156,6 +2156,7 @@ def test_nvfp4_multi_gpus_corner_case(self):
21562156

21572157
@pytest.mark.skip_less_mpi_world_size(8)
21582158
@skip_pre_hopper
2159+
@pytest.mark.skip_less_device_memory(140000)
21592160
@pytest.mark.parametrize(
21602161
"tp_size,pp_size,ep_size,mtp_nextn,fp8kv,attention_dp,cuda_graph,overlap_scheduler,max_batch_size,moe_backend",
21612162
[(8, 1, 4, 3, False, False, True, True, 1, "_DEFAULT"),

0 commit comments

Comments
 (0)