From 66d561aa712781ee7df2421a1c9706f853dd5a15 Mon Sep 17 00:00:00 2001 From: Zhangzefeng Date: Thu, 28 Mar 2024 14:06:34 +0800 Subject: [PATCH] Update test_rotary_emb_internlm.py --- tests/test_rotary_emb_internlm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_rotary_emb_internlm.py b/tests/test_rotary_emb_internlm.py index b7cecb18..26102897 100644 --- a/tests/test_rotary_emb_internlm.py +++ b/tests/test_rotary_emb_internlm.py @@ -17,7 +17,7 @@ def RotaryEmbTestFloat16() -> bool: ) res2 = ext.apply_rotary(input1, cos, sin, interleaved=interleaved, inplace=inplace) - # there is a little calculated error with ascend + # there is a little calculated error with ascend when dtype is float16 return torch.allclose(res1, res2, atol=1e-2, rtol=1e-3) def RotaryEmbTestFloat32() -> bool: