github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-24 06:18:01 -05:00
Code Issues Packages Projects Releases 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
parent b925d05e22
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):