fix atol

xadupre · xadupre · commit 6705737b4b78 · 2025-06-10T12:41:15.000+02:00
diff --git a/_unittests/ut_reference/test_torch_onnx_evaluator.py b/_unittests/ut_reference/test_torch_onnx_evaluator.py
@@ -1468,7 +1468,7 @@ def run(self, x, scale, bias=None):
         )
         expected = torch_sess.run(None, feeds)
         got = torch_sess_custom.run(None, feeds)
-        self.assertEqualAny(expected, got)
+        self.assertEqualAny(expected, got, atol=1e-3)
         self.assertEqual([1], LayerNormalizationOrt._shared)
 
 

Original file line number	Diff line number	Diff line change
`@@ -1468,7 +1468,7 @@ def run(self, x, scale, bias=None):`
`1468`	`1468`	`)`
`1469`	`1469`	`expected = torch_sess.run(None, feeds)`
`1470`	`1470`	`got = torch_sess_custom.run(None, feeds)`
`1471`		`- self.assertEqualAny(expected, got)`
	`1471`	`+ self.assertEqualAny(expected, got, atol=1e-3)`
`1472`	`1472`	`self.assertEqual([1], LayerNormalizationOrt._shared)`
`1473`	`1473`
`1474`	`1474`