Skip to content

Commit 103819b

Browse files
committed
修改rms_norm单测tol
1 parent 2220ad0 commit 103819b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/infiniop/rms_norm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545

4646
# Tolerance map for different data types
4747
_TOLERANCE_MAP = {
48-
InfiniDtype.F16: {"atol": 2e-3, "rtol": 2e-3},
48+
InfiniDtype.F16: {"atol": 1.5e-2, "rtol": 1.5e-2},
4949
InfiniDtype.BF16: {"atol": 8e-3, "rtol": 8e-3},
5050
}
5151

0 commit comments

Comments
 (0)