Skip to content

Commit f056d44

Browse files
authored
test: [nvbug: 5234494] skip_pre_ada for fp8 cases (#3718)
* skip_pre_ada for fp8 cases Signed-off-by: Ivy Zhang <[email protected]> * update Signed-off-by: Ivy Zhang <[email protected]> * update after rebase Signed-off-by: Ivy Zhang <[email protected]> --------- Signed-off-by: Ivy Zhang <[email protected]>
1 parent b0ac7c9 commit f056d44

File tree

3 files changed

+7
-3
lines changed

3 files changed

+7
-3
lines changed

tests/integration/defs/examples/test_llama.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4669,6 +4669,7 @@ def test_llm_llama_lookahead_xqa_fp8_1gpu(llama_example_root, llama_model_root,
46694669
venv_check_call(llm_venv, summary_cmd)
46704670

46714671

4672+
@skip_pre_ada
46724673
@pytest.mark.skip_less_device_memory(80000)
46734674
@pytest.mark.parametrize("code_llama_model_root", ['CodeLlama-7b-Instruct'],
46744675
indirect=True)

tests/integration/defs/examples/test_mistral.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -355,6 +355,7 @@ def test_llm_mistral_v1_smooth_quant_4gpus(llama_example_root,
355355
summary_cmd)
356356

357357

358+
@skip_pre_ada
358359
@pytest.mark.parametrize("run_type", ['inference', 'summarization'])
359360
@pytest.mark.parametrize("mistral_nemo_model_root", ['Mistral-Nemo-12b-Base'],
360361
indirect=True)

tests/integration/defs/test_e2e.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1605,9 +1605,11 @@ def test_ptp_quickstart_advanced_deepseek_r1_8gpus(llm_root, llm_venv,
16051605
pytest.param('Mixtral-8x7B-NVFP4',
16061606
'nvfp4-quantized/Mixtral-8x7B-Instruct-v0.1',
16071607
marks=skip_pre_blackwell),
1608-
pytest.param('Nemotron-Ultra-253B',
1609-
'nemotron-nas/Llama-3_1-Nemotron-Ultra-253B-v1',
1610-
marks=skip_pre_hopper),
1608+
pytest.param(
1609+
'Nemotron-Ultra-253B',
1610+
'nemotron-nas/Llama-3_1-Nemotron-Ultra-253B-v1',
1611+
marks=[skip_pre_hopper,
1612+
pytest.mark.skip_less_device_memory(140000)]),
16111613
])
16121614
def test_ptp_quickstart_advanced_8gpus(llm_root, llm_venv, model_name,
16131615
model_path):

0 commit comments

Comments
 (0)