From 353e81fa8fa347f34d1b05f385d412f4d41d3548 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Mon, 20 Feb 2023 10:18:19 -0800
Subject: [PATCH] need hand opt with IMAGE==2

---
 tinygrad/llops/ops_gpu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/llops/ops_gpu.py b/tinygrad/llops/ops_gpu.py
index 9da7730d1b..97b5f2d591 100644
--- a/tinygrad/llops/ops_gpu.py
+++ b/tinygrad/llops/ops_gpu.py
@@ -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):