Tighten moe-mxfp4 tolerance to rtol=2e-2, atol=2e-2

Mark Saroufim · Mark Saroufim · commit a846c7e9e8e4 · 2026-03-04T22:46:33.000-08:00
diff --git a/problems/amd_202602/moe-mxfp4/reference.py b/problems/amd_202602/moe-mxfp4/reference.py
@@ -296,4 +296,4 @@ def ref_kernel(data: input_t) -> output_t:
 
 
 
-check_implementation = make_match_reference(ref_kernel, rtol=5e-2, atol=5e-2)
+check_implementation = make_match_reference(ref_kernel, rtol=2e-2, atol=2e-2)

Original file line number	Diff line number	Diff line change
`@@ -296,4 +296,4 @@ def ref_kernel(data: input_t) -> output_t:`
`296`	`296`
`297`	`297`
`298`	`298`
`299`		`-check_implementation = make_match_reference(ref_kernel, rtol=5e-2, atol=5e-2)`
	`299`	`+check_implementation = make_match_reference(ref_kernel, rtol=2e-2, atol=2e-2)`