mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

fix KOPT=2 with variable shape

Browse source
This commit is contained in:
George Hotz 2023-08-22 11:34:34 -07:00
commit 696e4d20a1
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tinygrad/codegen/optimizer.py
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():