comment where future perf should go
MODIFY
tinygrad/llops/ops_gpu.py
tinygrad/llops/ops_gpu.py
@@ -246,6 +246,12 @@ class CLASTKernel(ASTKernel):
self.group_for_reduce.append(sz)
break
+ # TODO: this makes re_S3_32_3_3 at least 10x faster
+ #if self.first_reduce == 4 and self.shape_len == 7:
+ #self.group_for_reduce.append(112//2)
+ # TODO: this shouldn't have to be permuted
+ #self.reshape_and_permute(None, [0,1,2,3,6,4,5])
+
# if there's images in the latebufs, we have to make an axis the 4 storing one. this affects the kernel shape
self.upcast_in_mid_reduce = False
if any(isinstance(buf._buf, CLImage) for buf in self.bufs if buf not in self.earlybufs) and self.buftokens[0].typ != Types.FLOAT4:

comment where future perf should go