fix KOPT=2 with variable shape

This commit is contained in:
George Hotz
2023-08-22 11:34:34 -07:00
parent de1fcc418f
commit 696e4d20a1

View File

@@ -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():