Skip to content

Commit 1aa8194

Browse files
committed
[https://nvbugs/5596343][test] Update test waive to get back some coverage
Signed-off-by: Dongfeng Yu <[email protected]>
1 parent a09b38a commit 1aa8194

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/integration/defs/accuracy/test_llm_api_pytorch.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3465,7 +3465,6 @@ def test_dummy_load_format(self):
34653465
task = GSM8K(model_name)
34663466
task.evaluate(llm, is_integration_test=True)
34673467

3468-
@pytest.mark.skip(reason="https://nvbugs/5596343")
34693468
@pytest.mark.skip_less_device(4)
34703469
@pytest.mark.parametrize(
34713470
"kv_cache_dtype",
@@ -3485,6 +3484,10 @@ def test_dummy_load_format(self):
34853484
def test_w4_4gpus(self, kv_cache_dtype, moe_backend, tp_size, pp_size,
34863485
ep_size, attention_dp, cuda_graph, overlap_scheduler,
34873486
mocker):
3487+
if get_sm_version() < 100:
3488+
pytest.skip(
3489+
"https://nvbugs/5596343: Skip Hopper due to accuracy issue.")
3490+
34883491
mocker.patch.object(GSM8K, "MAX_OUTPUT_LEN", 8192)
34893492
mocker.patch.dict(GSM8K.EVALUATE_KWARGS,
34903493
{"scores_filter": "exact_match,flexible-extract"})

0 commit comments

Comments
 (0)