Merge pull request #138 from benenzhu/moe-patch-toerence

S1ro1 · web-flow · commit 8b0e3ea5648d · 2026-03-22T20:16:35.000-07:00
Update reference.py for MOE-mxfp4.
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=2e-2, atol=2e-2)
+check_implementation = make_match_reference(ref_kernel, rtol=5e-2, atol=5e-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=2e-2, atol=2e-2)`
	`299`	`+check_implementation = make_match_reference(ref_kernel, rtol=5e-2, atol=5e-2)`