diff --git a/test/test_schedule.py b/test/test_schedule.py index 46cbb1ba61a2..4ea0a5b95a37 100644 --- a/test/test_schedule.py +++ b/test/test_schedule.py @@ -1608,7 +1608,7 @@ def f(a): return a.item() r, et = timeit(f, a) self.assertEqual(r, val) - self.assertLess(et, 1e3) + self.assertLess(et, 1200) def test_no_rewrite_elementwise(self): bufs = [UOp(UOps.DEFINE_GLOBAL, dtypes.int.ptr(), (), i) for i in range(3)]