diff --git a/tests/integration/defs/accuracy/test_disaggregated_serving.py b/tests/integration/defs/accuracy/test_disaggregated_serving.py index 46d4f51a712c..bacdedba92e0 100644 --- a/tests/integration/defs/accuracy/test_disaggregated_serving.py +++ b/tests/integration/defs/accuracy/test_disaggregated_serving.py @@ -2282,10 +2282,20 @@ def _make_configs(self, use_py_transceiver: bool = False): return ctx_server_config, gen_server_config, disaggregated_server_config @pytest.mark.skip_less_device(8) - @parametrize_with_ids("use_py_transceiver", [True, False]) - @parametrize_with_ids("block_reuse", [True, False]) - @parametrize_with_ids("mtp_nextn", [0, 1, 3]) - def test_auto_dtype(self, use_py_transceiver, block_reuse, mtp_nextn): + @pytest.mark.parametrize( + "mtp_nextn,block_reuse,use_py_transceiver", + [ + (0, False, False), + (0, False, True), + (3, True, False), + ], + ids=[ + "mtp_nextn=0-block_reuse=False-use_py_transceiver=False", + "mtp_nextn=0-block_reuse=False-use_py_transceiver=True", + "mtp_nextn=3-block_reuse=True-use_py_transceiver=False", + ], + ) + def test_auto_dtype(self, mtp_nextn, block_reuse, use_py_transceiver): if use_py_transceiver and block_reuse: pytest.skip("Python transceiver does not support block reuse") diff --git a/tests/integration/test_lists/waives.txt b/tests/integration/test_lists/waives.txt index e0da833bc85b..9093a7e17b1b 100644 --- a/tests/integration/test_lists/waives.txt +++ b/tests/integration/test_lists/waives.txt @@ -12,8 +12,6 @@ accuracy/test_disaggregated_serving.py::TestLlama3_1_8BInstruct::test_tp_pp_symm accuracy/test_disaggregated_serving.py::TestLlama3_1_8BInstruct::test_tp_pp_symmetric[GSM8K-tp2pp2] SKIP (https://nvbugs/6428069) accuracy/test_disaggregated_serving.py::TestLlama3_1_8BInstruct::test_tp_pp_symmetric[MMLU-tp1pp2] SKIP (https://nvbugs/6427411) accuracy/test_disaggregated_serving.py::TestLlama3_1_8BInstruct::test_tp_pp_symmetric[MMLU-tp2pp2] SKIP (https://nvbugs/6428069) -accuracy/test_disaggregated_serving.py::TestNemotron3Super120B::test_auto_dtype[mtp_nextn=0-block_reuse=False-use_py_transceiver=True] SKIP (https://nvbugs/6478726) -accuracy/test_disaggregated_serving.py::TestQwen3NextInstruct::test_auto_dtype[use_py_transceiver=False] SKIP (https://nvbugs/6427411) accuracy/test_disaggregated_serving.py::TestQwen3_30B_A3B::test_mixed_ctx_gen_model[ctxpp2gentp2] SKIP (https://nvbugs/5748664) accuracy/test_epd_disagg_multimodal.py::TestVideoMMEEPD::test_disaggregated_videomme[nemotron_nano_v3_omni_fp8] SKIP (https://nvbugs/6478692) accuracy/test_llm_api_autodeploy.py::TestNemotronNanoV3::test_accuracy[nvfp4-4-attn_dp_off-trtllm] SKIP (https://nvbugs/6367792)