diff --git a/test/backend/test_arange.py b/test/backend/test_arange.py index c07c999b6f..faa8b48137 100644 --- a/test/backend/test_arange.py +++ b/test/backend/test_arange.py @@ -19,6 +19,7 @@ class TestArange(unittest.TestCase): self.assertEqual(self._get_flops(Tensor.arange(256), np.arange(256)), 0) self.assertEqual(self._get_flops(Tensor.arange(2560), np.arange(2560)), 0) + @unittest.skipIf(Device.DEFAULT == "CL", "flaky in CI") def test_arange_cumsum(self): np.testing.assert_equal(Tensor.arange(513).cumsum(0).numpy(), np.arange(513).cumsum())