really dumb bug
MODIFY
extra/thneed.py
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 = []

really dumb bug