From 287a198c4fc16ee5189b4f19f0743c53a3a02142 Mon Sep 17 00:00:00 2001 From: chenyu Date: Thu, 17 Oct 2024 11:08:50 -0400 Subject: [PATCH] increase test_strongly_connected_DAG threshold (#7131) flaky --- test/test_schedule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)]