diff --git a/test/test_core_aten_ops.py b/test/test_core_aten_ops.py index d23004d57d0..147043f268f 100644 --- a/test/test_core_aten_ops.py +++ b/test/test_core_aten_ops.py @@ -49,6 +49,9 @@ def run_export_and_compare(testcase, func, args, kwargs, atol=1e-3): class AtenOpTest(unittest.TestCase): + def setUp(self): + torch.manual_seed(0) + def test_aten_abs_0(self): args = (torch.randn((10, 10)).to(torch.float32),) kwargs = dict() @@ -3538,6 +3541,7 @@ def test_aten_remainder_Tensor_1(self): kwargs = dict() run_export_and_compare(self, torch.ops.aten.remainder.Tensor, args, kwargs) + @unittest.skip def test_aten_replication_pad2d_0(self): args = ( torch.randn((3, 2, 10)).to(torch.float32), @@ -3551,6 +3555,7 @@ def test_aten_replication_pad2d_0(self): kwargs = dict() run_export_and_compare(self, torch.ops.aten.replication_pad2d, args, kwargs) + @unittest.skip def test_aten_replication_pad2d_1(self): args = ( torch.randint(0, 10, (3, 2, 10)).to(torch.int32),