Skip to content

Commit 08a06bf

Browse files
committed
fix by CR bot's comments
Signed-off-by: Lizhi Zhou <[email protected]>
1 parent 0721724 commit 08a06bf

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

tests/integration/defs/accuracy/test_disaggregated_serving.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -727,19 +727,20 @@ def test_auto_dtype(self, overlap_scheduler):
727727
@skip_pre_blackwell
728728
@pytest.mark.timeout(3600)
729729
class TestQwen3_30B_A3B(LlmapiAccuracyTestHarness):
730-
fp4_model = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_nvfp4_hf"
731-
fp8_model = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_fp8_hf"
732-
733-
@pytest.mark.parametrize("ctxpp,gentp", [(2, 2)], ids=["ctxpp2gentp2"])
734-
def test_mixed_ctx_gen_model(self, ctxpp, gentp):
735-
ctx_model = self.fp4_model
736-
gen_model = self.fp8_model
730+
FP4_MODEL = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_nvfp4_hf"
731+
FP8_MODEL = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_fp8_hf"
732+
733+
@pytest.mark.skip_less_device(4)
734+
@pytest.mark.parametrize("ctx_pp,gen_tp", [(2, 2)], ids=["ctxpp2gentp2"])
735+
def test_mixed_ctx_gen_model(self, ctx_pp, gen_tp):
736+
ctx_model = self.FP4_MODEL
737+
gen_model = self.FP8_MODEL
737738
return run_parallel_test("Qwen3/Qwen3-30B-A3B",
738739
ctx_model,
739-
ctx_pp=ctxpp,
740+
ctx_pp=ctx_pp,
740741
ctx_tp=1,
741742
gen_pp=1,
742-
gen_tp=gentp,
743+
gen_tp=gen_tp,
743744
test_sets=[GSM8K, MMLU],
744745
ctx_model=ctx_model,
745746
gen_model=gen_model)

0 commit comments

Comments
 (0)