mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-25 14:58:46 -05:00
fix KOPT=2 with variable shape
This commit is contained in:
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user