Skip to content

Commit 40049bd

Browse files
[NPU]Fix rms norm (#1833)
1 parent 4387416 commit 40049bd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/npu/tests/unittests/test_rms_norm_npu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,8 @@ def check_result(self, golden_res, fused_res):
5252
rtol = 1e-03
5353
atol = 1e-03
5454
elif self.dtype == "bfloat16":
55-
rtol = 4e-3
56-
atol = 4e-3
55+
rtol = 1e-02
56+
atol = 1e-02
5757
else:
5858
self.assertTrue(
5959
False,

0 commit comments

Comments
 (0)