atol

Author: xadupre

_unittests/ut_torch_export_patches/test_patch_torch.py

@@ -63,7 +63,7 @@ def test_vmap_tuple(self):
         x, y = torch.randn(2, 5), torch.randn(5)
         expected = torch.vmap(torch.dot, in_dims=(0, None))(x, y)
         got = patched_vmap(torch.dot, in_dims=(0, None))(x, y)
-        self.assertEqualArray(expected, got)
+        self.assertEqualArray(expected, got, atol=1e-5)
 
     def test_vmap_transformers_scenario_vmap(self):
         def padding_mask_function(padding_mask: torch.Tensor) -> Callable: