Skip to content

Commit 0316720

Browse files
committed
update test
Signed-off-by: jiahanc <[email protected]>
1 parent 075f096 commit 0316720

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/moe/test_trtllm_gen_fused_moe.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2306,8 +2306,7 @@ def test_renormalize_routing(
23062306
"moe_impl",
23072307
[
23082308
pytest.param(FP4Moe(quant_mode=QuantMode.FP4_NVFP4_NVFP4), id="NvFP4xNvFP4"),
2309-
pytest.param(FP4Moe(quant_mode=QuantMode.FP4_MXFP4_MXFP8), id="MxFP4xMxFP8"),
2310-
pytest.param(FP4Moe(quant_mode=QuantMode.FP4_MXFP4_Bf16), id="MxFP4xBf16"),
2309+
pytest.param(FP4Moe(quant_mode=QuantMode.FP4_MXFP4_MXFP8), id="MxFP4xMxFP8")
23112310
],
23122311
)
23132312
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)