add tolerance to specific tests

Author: danielenricocahall

keras/src/ops/nn_test.py

@@ -1324,14 +1324,17 @@ def test_polar(self):
 
 
 class NNOpsCorrectnessTest(testing.TestCase):
-    def assertAllClose(self, x1, x2, atol=1e-6, rtol=1e-6, msg=None):
+    def setUp(self):
+        super().setUp()
         if backend.backend() == "openvino":
-            # OpenVINO seems to use lower precision for some operations,
-            # or employs some different algorithms that wind up with
-            # slightly different results. To address this, we relax
-            # the tolerances for OpenVINO backend.
-            atol = 1e-3
-        super().assertAllClose(x1, x2, atol=atol, rtol=rtol, msg=msg)
+            import openvino as ov
+            import openvino.properties.hint as hints
+
+            core = ov.Core()
+            core.set_property(
+                "CPU",
+                {hints.execution_mode: hints.ExecutionMode.PERFORMANCE},
+            )
 
     def test_relu(self):
         x = np.array([-1, 0, 1, 2, 3], dtype=np.float32)
@@ -1360,7 +1363,13 @@ def test_softplus(self):
 
     def test_softsign(self):
         x = np.array([-1, 0, 1, 2, 3], dtype=np.float32)
-        self.assertAllClose(knn.softsign(x), [-0.5, 0, 0.5, 0.6666667, 0.75])
+        if backend.backend() == "openvino":
+            kwargs = {"atol": 1e-3}
+        else:
+            kwargs = {}
+        self.assertAllClose(
+            knn.softsign(x), [-0.5, 0, 0.5, 0.6666667, 0.75], **kwargs
+        )
 
     def test_silu(self):
         x = np.array([-1, 0, 1, 2, 3], dtype=np.float32)
@@ -1378,41 +1387,57 @@ def test_log_sigmoid(self):
 
     def test_leaky_relu(self):
         x = np.array([-1, 0, 1, 2, 3], dtype=np.float32)
-        self.assertAllClose(
-            knn.leaky_relu(x),
-            [-0.2, 0, 1, 2, 3],
-        )
+        if backend.backend() == "openvino":
+            kwargs = {"atol": 1e-3}
+        else:
+            kwargs = {}
+        self.assertAllClose(knn.leaky_relu(x), [-0.2, 0, 1, 2, 3], **kwargs)
 
     def test_hard_sigmoid(self):
         x = np.array([-1, 0, 1, 2, 3], dtype=np.float32)
+        if backend.backend() == "openvino":
+            kwargs = {"atol": 1e-3}
+        else:
+            kwargs = {}
         self.assertAllClose(
             knn.hard_sigmoid(x),
             [0.33333334, 0.5, 0.6666667, 0.8333334, 1.0],
+            **kwargs,
         )
 
     def test_hard_silu(self):
         x = np.array([-3, -2, -1, 0, 1, 2, 3], dtype=np.float32)
+        if backend.backend() == "openvino":
+            kwargs = {"atol": 1e-3}
+        else:
+            kwargs = {}
         self.assertAllClose(
             knn.hard_silu(x),
             [-0.0, -0.333333, -0.333333, 0.0, 0.6666667, 1.6666667, 3.0],
+            **kwargs,
         )
 
     def test_elu(self):
         x = np.array([-1, 0, 1, 2, 3], dtype=np.float32)
+        if backend.backend() == "openvino":
+            kwargs = {"atol": 1e-3}
+        else:
+            kwargs = {}
+        self.assertAllClose(knn.elu(x), [-0.63212055, 0, 1, 2, 3], **kwargs)
         self.assertAllClose(
-            knn.elu(x),
-            [-0.63212055, 0, 1, 2, 3],
-        )
-        self.assertAllClose(
-            knn.elu(x, alpha=0.5),
-            [-0.31606027, 0, 1, 2, 3],
+            knn.elu(x, alpha=0.5), [-0.31606027, 0, 1, 2, 3], **kwargs
         )
 
     def test_selu(self):
         x = np.array([-1, 0, 1, 2, 3], dtype=np.float32)
+        if backend.backend() == "openvino":
+            kwargs = {"atol": 1e-3}
+        else:
+            kwargs = {}
         self.assertAllClose(
             knn.selu(x),
             [-1.1113307, 0.0, 1.050701, 2.101402, 3.152103],
+            **kwargs,
         )
 
     def test_gelu(self):
@@ -2288,7 +2313,11 @@ def test_ctc_loss(self):
         output_length = np.array([3, 2])
 
         result = knn.ctc_loss(labels, outputs, label_length, output_length)
-        self.assertAllClose(result, np.array([3.4411672, 1.91680186]))
+        if backend.backend() == "openvino":
+            kwargs = {"atol": 1e-3}
+        else:
+            kwargs = {}
+        self.assertAllClose(result, np.array([3.4411672, 1.91680186]), **kwargs)
 
     def test_ctc_decode(self):
         inputs = np.array(