fix unit test

xadupre · Jun 4, 2024 · 6a3c8b0 · 6a3c8b0
1 parent f24ff44
commit 6a3c8b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/cuda/test_cudaops.py b/test/cuda/test_cudaops.py
@@ -216,7 +216,7 @@ def _addmul_shared_input_cuda(self, itype, op_type, shapea=(3, 2, 3), shapeb=(3,
         sess = _ort.InferenceSession(model2.SerializeToString(), opts, providers=["CUDAExecutionProvider"])
         got = sess.run(None, feeds1)
         for i in range(2):
-            self.assertEqualArray(expected[i], got[i])
+            assert_almost_equal(expected[i], got[i])
 
     @unittest.skipIf(not has_cuda(), reason="CUDA is missing")
     def test_add_shared_input_cuda(self):