modify atol and rtol

DeepLink-org · Jan 9, 2025 · c39b8c5 · c39b8c5
1 parent 5565fbc
commit c39b8c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/internevo/test_rotary_embedding.py b/tests/internevo/test_rotary_embedding.py
@@ -41,7 +41,7 @@ def test_ApplyRotaryEmb():
                     in_place,
                 )
                 assert allclose(
-                    output_ref, output_ext, rtol=1e-2, atol=5e-2
+                    output_ref, output_ext
                 ), f"When input dtype is {input_dtype} and in_place is {in_place}, ApplyRotaryEmb fails to pass the forward test!"
                 assert allclose(
                     grad_ref, grad_ext