From 696e4d20a12e73d63ccf6d574cf32666e388edf5 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Tue, 22 Aug 2023 11:34:34 -0700
Subject: [PATCH] fix KOPT=2 with variable shape

---
 tinygrad/codegen/optimizer.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tinygrad/codegen/optimizer.py b/tinygrad/codegen/optimizer.py
index fa5ebb8738..2c59d844e9 100644
--- a/tinygrad/codegen/optimizer.py
+++ b/tinygrad/codegen/optimizer.py
@@ -72,6 +72,9 @@ def kernel_optimize(k:Linearizer, create_k:Callable[[], Linearizer], to_prg):
 
   if global_db is not None and skey in global_db:
     choice = global_db[skey]
+  elif any(not isinstance(x, int) for x in k.full_shape):
+    # don't optimize variable shapes
+    choice = "BASELINE"
   else:
     # get baseline
     def get_baseline():