add IsNaN

Author: xadupre

_unittests/ut_torch_models/test_validate_models.py

@@ -30,7 +30,7 @@ def test_validate_microsoft_phi4_reasoning(self):
             stop_if_static=2 if pv.Version(torch.__version__) > pv.Version("2.6.1") else 0,
             dump_folder="dump_test/validate_microsoft_phi4_reasoning",
         )
-        self.assertLess(summary["disc_onnx_ort_run_abs"], 1e-5)
+        self.assertLess(summary["disc_onnx_ort_run_abs"], 2e-5)
         self.assertIn("onnx_filename", data)
 
     @requires_transformers("4.53")
@@ -51,7 +51,7 @@ def test_validate_microsoft_phi3_mini_128k(self):
             stop_if_static=2 if pv.Version(torch.__version__) > pv.Version("2.6.1") else 0,
             dump_folder="dump_test/validate_microsoft_phi3_mini_128k",
         )
-        self.assertLess(summary["disc_onnx_ort_run_abs"], 1e-5)
+        self.assertLess(summary["disc_onnx_ort_run_abs"], 2e-5)
         self.assertIn("onnx_filename", data)
 
 

onnx_diagnostic/reference/torch_ops/__init__.py

@@ -45,6 +45,7 @@
     Erf_9,
     Exp_1,
     Identity_1,
+    IsNaN_9,
     Log_1,
     Neg_1,
     Not_1,

onnx_diagnostic/reference/torch_ops/unary_ops.py

@@ -37,6 +37,13 @@ def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(x.tensor)
 
 
+class IsNaN_9(OpRunKernel):
+    """IsNaN"""
+
+    def run(self, x: OpRunTensor) -> OpRunTensor:
+        return OpRunTensor(x.tensor.isnan())
+
+
 class Log_1(OpRunKernel):
     """Log"""