未验证 提交 75b73400 编写于 作者: Z zhangkaihuo 提交者: GitHub

fix sparse_attention unittest (#47547)

上级 eb100c7b
...@@ -405,7 +405,7 @@ class TestSparseAttentionAPI(unittest.TestCase): ...@@ -405,7 +405,7 @@ class TestSparseAttentionAPI(unittest.TestCase):
) )
np.testing.assert_allclose( np.testing.assert_allclose(
fetches_result, expected_result, rtol=1e-05, atol=1e-05 fetches_result[0], expected_result, rtol=1e-05, atol=1e-05
) )
def test_dygraph(self): def test_dygraph(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册