atol

Author: xadupre

_unittests/ut_light_api/test_backend_export.py

@@ -244,7 +244,19 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
         raise NotImplementedError("Unable to run the model node by node.")
 
 
-backend_test = onnx.backend.test.BackendTest(ExportBackend, __name__)
+dft_atol = 1e-3 if sys.platform != "linux" else 1e-5
+backend_test = onnx.backend.test.BackendTest(
+    ExportBackend,
+    __name__,
+    test_kwargs={
+        "test_dft": {"atol": dft_atol},
+        "test_dft_axis": {"atol": dft_atol},
+        "test_dft_axis_opset19": {"atol": dft_atol},
+        "test_dft_inverse": {"atol": dft_atol},
+        "test_dft_inverse_opset19": {"atol": dft_atol},
+        "test_dft_opset19": {"atol": dft_atol},
+    },
+)
 
 # The following tests are too slow with the reference implementation (Conv).
 backend_test.exclude(

_unittests/ut_reference/test_backend_extended_reference_evaluator.py

@@ -1,5 +1,6 @@
 import os
 import platform
+import sys
 import unittest
 from typing import Any
 import numpy
@@ -78,10 +79,21 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
         raise NotImplementedError("Unable to run the model node by node.")
 
 
+dft_atol = 1e-3 if sys.platform != "linux" else 1e-5
 backend_test = onnx.backend.test.BackendTest(
-    ExtendedReferenceEvaluatorBackend, __name__
+    ExtendedReferenceEvaluatorBackend,
+    __name__,
+    test_kwargs={
+        "test_dft": {"atol": dft_atol},
+        "test_dft_axis": {"atol": dft_atol},
+        "test_dft_axis_opset19": {"atol": dft_atol},
+        "test_dft_inverse": {"atol": dft_atol},
+        "test_dft_inverse_opset19": {"atol": dft_atol},
+        "test_dft_opset19": {"atol": dft_atol},
+    },
 )
 
+
 if os.getenv("APPVEYOR"):
     backend_test.exclude("(test_vgg19|test_zfnet)")
 if platform.architecture()[0] == "32bit":