Fix testcase for cuda10

2023-02-20 17:44:36 +08:00 · 2023-02-20 17:44:36 +08:00 · 5720796ae1
parent 42fa3f67af
commit 5720796ae1
1 changed files with 1 additions and 1 deletions
--- a/tests/st/fallback/test_graph_fallback_runtime.py
+++ b/tests/st/fallback/test_graph_fallback_runtime.py
@ -575,7 +575,7 @@ def test_net_dict_2_grad():
    outputs1 = ops.grad(net)(x)
    dict_lenet = DictLeNet()
    outputs2 = ops.grad(dict_lenet)(x)
-    assert np.all(outputs1.asnumpy() == outputs2.asnumpy())
+    assert np.allclose(outputs1.asnumpy(), outputs2.asnumpy())


@pytest.mark.level0