mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

need hand opt with IMAGE==2

Browse source
This commit is contained in:
George Hotz 2023-02-20 10:18:19 -08:00
commit 353e81fa8f
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tinygrad/llops/ops_gpu.py
View file

@ -231,7 +231,7 @@ class CLASTKernel(ASTKernel):
def codegen(self) -> Callable:
self.process()
self.upcast_in_mid_reduce = False
if not KOPT: self.hand_coded_optimizations()
if not KOPT or IMAGE == 2: self.hand_coded_optimizations()
# add a local buffer for multistage reduce
if len(self.group_for_reduce):