未验证 提交 fa07857a 编写于 作者: G Guoxia Wang 提交者: GitHub

[AMP OP&Test] update bf16 atol=0.1 to atol=0.02 (#52121)

上级 5a9d59c5
...@@ -472,7 +472,7 @@ class TestBF16ScaleBiasLayerNorm(unittest.TestCase): ...@@ -472,7 +472,7 @@ class TestBF16ScaleBiasLayerNorm(unittest.TestCase):
) )
def assert_equal(x, y): def assert_equal(x, y):
np.testing.assert_allclose(x, y, rtol=1e-05, atol=0.1) np.testing.assert_allclose(x, y, rtol=1e-05, atol=3e-2)
assert_equal(y_np_1, y_np_2) assert_equal(y_np_1, y_np_2)
assert_equal(x_g_np_1, x_g_np_2) assert_equal(x_g_np_1, x_g_np_2)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册