From 9448924d9e6e5ce20dfdb822ef00d3294a658e0f Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Wed, 8 Oct 2025 14:29:11 +0300 Subject: [PATCH] update gpt2 kernel count tests in CI=0 (#12523) --- test/models/test_real_world.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/models/test_real_world.py b/test/models/test_real_world.py index ccfccf45e9..0dc7e42b6d 100644 --- a/test/models/test_real_world.py +++ b/test/models/test_real_world.py @@ -94,7 +94,7 @@ class TestRealWorld(unittest.TestCase): @TinyJit def test(t, v): with Context(JIT=0): return model(t, v).realize() - helper_test("test_gpt2", lambda: (Tensor([[1,]]),Variable("pos", 1, 100).bind(1)), test, 0.23 if CI else 0.9, 160 if CI else 396, all_jitted=True) + helper_test("test_gpt2", lambda: (Tensor([[1,]]),Variable("pos", 1, 100).bind(1)), test, 0.23 if CI else 0.9, 160 if CI else 468, all_jitted=True) @unittest.skipIf(CI and Device.DEFAULT == "CPU", "slow") def test_train_mnist(self):