From 63990705b5897811914d9b2bc200221a1b5de31b Mon Sep 17 00:00:00 2001 From: chenyu Date: Mon, 15 Jul 2024 20:09:38 -0400 Subject: [PATCH] test kernel opts case for 4 local and 4 groups (#5499) make sure local grouped dim is correct --- test/test_linearizer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/test_linearizer.py b/test/test_linearizer.py index ffd6935b..61e6d718 100644 --- a/test/test_linearizer.py +++ b/test/test_linearizer.py @@ -1172,6 +1172,10 @@ class TestKernelOpts(unittest.TestCase): [Opt(OptOps.LOCAL, 0, 2), Opt(OptOps.GROUPTOP, 0, 64)], # Checking how it works with locals + grouped reduce + upcasts [Opt(OptOps.LOCAL, 0, 2), Opt(OptOps.GROUPTOP, 0, 2), Opt(OptOps.UPCAST, 0, 8), Opt(OptOps.UNROLL, 1, 4)], + # many local + many group + [Opt(OptOps.GROUP, 0, 2)] * 4, + [Opt(OptOps.LOCAL, 0, 2)] * 4, + [Opt(OptOps.LOCAL, 0, 2), Opt(OptOps.GROUP, 0, 2)] * 4, ]) @unittest.skipIf(CI and Device.DEFAULT in {"AMD"}, "AMD CI doesn't support multiple sync threads yet")