Skip to content

Commit 49c27fc

Browse files
committed
fix by CR bot's comments
Signed-off-by: Lizhi Zhou <[email protected]>
1 parent 423a7c3 commit 49c27fc

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
@@ -801,19 +801,20 @@ def test_auto_dtype(self, overlap_scheduler):
801801
@skip_pre_blackwell
802802
@pytest.mark.timeout(3600)
803803
class TestQwen3_30B_A3B(LlmapiAccuracyTestHarness):
804-
fp4_model = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_nvfp4_hf"
805-
fp8_model = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_fp8_hf"
806-
807-
@pytest.mark.parametrize("ctxpp,gentp", [(2, 2)], ids=["ctxpp2gentp2"])
808-
def test_mixed_ctx_gen_model(self, ctxpp, gentp):
809-
ctx_model = self.fp4_model
810-
gen_model = self.fp8_model
804+
FP4_MODEL = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_nvfp4_hf"
805+
FP8_MODEL = f"{llm_models_root()}/Qwen3/saved_models_Qwen3-30B-A3B_fp8_hf"
806+
807+
@pytest.mark.skip_less_device(4)
808+
@pytest.mark.parametrize("ctx_pp,gen_tp", [(2, 2)], ids=["ctxpp2gentp2"])
809+
def test_mixed_ctx_gen_model(self, ctx_pp, gen_tp):
810+
ctx_model = self.FP4_MODEL
811+
gen_model = self.FP8_MODEL
811812
return run_parallel_test("Qwen3/Qwen3-30B-A3B",
812813
ctx_model,
813-
ctx_pp=ctxpp,
814+
ctx_pp=ctx_pp,
814815
ctx_tp=1,
815816
gen_pp=1,
816-
gen_tp=gentp,
817+
gen_tp=gen_tp,
817818
test_sets=[GSM8K, MMLU],
818819
ctx_model=ctx_model,
819820
gen_model=gen_model)

0 commit comments

Comments
 (0)