Skip to content

Commit 7382df0

Browse files
danielhua23root
andauthored
Enhance mla numeric check mechanism (#137)
Co-authored-by: root <root@smci355-ccs-aus-n06-05.cs-aus.dcgpu>
1 parent c29c9a2 commit 7382df0

3 files changed

Lines changed: 280 additions & 138 deletions

File tree

problems/amd_202602/mixed-mla/reference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,4 +369,4 @@ def ref_kernel(data: input_t) -> output_t:
369369
)
370370

371371

372-
check_implementation = make_match_reference(ref_kernel, rtol=2e-02, atol=2e-02)
372+
check_implementation = make_match_reference(ref_kernel, rtol=1e-01, atol=1e-01)

0 commit comments

Comments
 (0)