We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 94919c3 commit 5a6fec0Copy full SHA for 5a6fec0
1 file changed
tests/moe/test_trtllm_gen_routed_fused_moe.py
@@ -707,6 +707,7 @@ def test_trtllm_gen_fp8_mxfp8_routed_activation_parity(activation_type: int):
707
mismatch_pct = (~close).float().mean().item() * 100
708
assert mismatch_pct < 10, f"Mismatch percentage is {mismatch_pct:.2f}%"
709
710
+
711
@pytest.mark.parametrize("num_tokens", [8, 64])
712
@pytest.mark.parametrize("hidden_size", [1024, 2048])
713
@pytest.mark.parametrize("intermediate_size", [1024, 2048])
0 commit comments