From 63f9c55156b67beb3f259a5bdb54d0a0035d27cc Mon Sep 17 00:00:00 2001
From: George Hotz <george@comma.ai>
Date: Thu, 20 Oct 2022 17:07:47 -0700
Subject: [PATCH] really dumb bug

---
 extra/thneed.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/extra/thneed.py b/extra/thneed.py
index 6c44074a3c..8b9edd2d82 100644
--- a/extra/thneed.py
+++ b/extra/thneed.py
@@ -281,15 +281,13 @@ class Thneed:
     local_cl_cache = []
     for prg, args in self.cl_cache:
       args = list(args)
-      # TODO: WTF is wrong with to_image?
-      if args[1] is None and len(args[0]) == 2 and 'to_image' not in prg.name:
+      if args[1] is None and len(args[0]) == 2:
         args[1] = [min(MAX_WORKGROUP, args[0][0]), 1]
         try:
           e = prg.clprg(CL().cl_queue, *args)
         except cl.LogicError:
           # INVALID_WORK_GROUP_SIZE
           args[1] = None
-          continue
 
       if args[1] is None and len(args[0]) == 3:
         runtimes = []