MoE test - Add FP4_NVFP4_NVFP4 to allowed quant_mode with logits_dtype of fp32

amitz-nv · amitz-nv · commit e9df6f8b9730 · 2026-03-31T16:11:39.000Z
Signed-off-by: amitz-nv &lt;203509407+amitz-nv@users.noreply.github.com&gt;
diff --git a/tests/moe/utils.py b/tests/moe/utils.py
@@ -170,6 +170,7 @@ def skip_checks(
         )
 
     if logits_dtype == torch.float32 and moe_impl.quant_mode not in [
+        QuantMode.FP4_NVFP4_NVFP4,
         QuantMode.FP8_PER_TENSOR,
         QuantMode.FP8_BLOCK_SCALE_DEEPSEEK,
         QuantMode.FP8_BLOCK_SCALE_MXFP8,

Original file line number	Diff line number	Diff line change
`@@ -170,6 +170,7 @@ def skip_checks(`
`170`	`170`	`)`
`171`	`171`
`172`	`172`	`if logits_dtype == torch.float32 and moe_impl.quant_mode not in [`
	`173`	`+ QuantMode.FP4_NVFP4_NVFP4,`
`173`	`174`	`QuantMode.FP8_PER_TENSOR,`
`174`	`175`	`QuantMode.FP8_BLOCK_SCALE_DEEPSEEK,`
`175`	`176`	`QuantMode.FP8_BLOCK_SCALE_MXFP8,`