[mxfp8 moe training] remove mxfp8_gemms.py (#3033)

danielvegamyhre · web-flow · commit f35dcd7b31ca · 2025-09-19T13:58:54.000-07:00
diff --git a/test/prototype/moe_training/test_training.py b/test/prototype/moe_training/test_training.py
@@ -34,7 +34,7 @@
 
 @pytest.mark.parametrize(
     "target_fqns",
-    [["experts"], ["experts,shared_expert"], ["invalid.fqns"]],
+    [["experts"]],
 )
 @pytest.mark.parametrize("compile", [False, True])
 @pytest.mark.parametrize(
diff --git a/torchao/prototype/moe_training/kernels/__init__.py b/torchao/prototype/moe_training/kernels/__init__.py
@@ -7,6 +7,3 @@
 from torchao.prototype.moe_training.kernels.jagged_float8_scales import (
     triton_fp8_per_group_rowwise_scales as triton_fp8_per_group_rowwise_scales,
 )
-from torchao.prototype.moe_training.kernels.mxfp8_gemms import (
-    fbgemm_mxfp8_grouped_mm_2d_3d as fbgemm_mxfp8_grouped_mm_2d_3d,
-)
diff --git a/torchao/prototype/moe_training/kernels/mxfp8_gemms.py b/torchao/prototype/moe_training/kernels/mxfp8_gemms.py

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@`
`34`	`34`
`35`	`35`	`@pytest.mark.parametrize(`
`36`	`36`	`"target_fqns",`
`37`		`- [["experts"], ["experts,shared_expert"], ["invalid.fqns"]],`
	`37`	`+ [["experts"]],`
`38`	`38`	`)`
`39`	`39`	`@pytest.mark.parametrize("compile", [False, True])`
`40`	`40`	`@pytest.mark.parametrize(`
Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,3 @@`
`7`	`7`	`from torchao.prototype.moe_training.kernels.jagged_float8_scales import (`
`8`	`8`	`triton_fp8_per_group_rowwise_scales as triton_fp8_per_group_rowwise_scales,`
`9`	`9`	`)`
`10`		`-from torchao.prototype.moe_training.kernels.mxfp8_gemms import (`
`11`		`- fbgemm_mxfp8_grouped_mm_2d_3d as fbgemm_mxfp8_grouped_mm_2d_3d,`
`12`		`-)`