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

e kernels

Browse source
This commit is contained in:
George Hotz 2025-03-26 16:49:23 +08:00
commit 9e19cdfbbe
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
extra/replay_pkl.py
View file

@ -135,8 +135,11 @@ if __name__ == "__main__":
k.apply_opt(Opt(OptOps.UNROLL, 0, 8))
k.apply_opt(Opt(OptOps.UPCAST, 2, 32))
if k.full_shape[1]%4 == 0: k.apply_opt(Opt(OptOps.UPCAST, 1, 4))
elif len(k.full_shape) == 1 and k.full_shape[0]%128 == 0:
k.apply_opt(Opt(OptOps.UPCAST, 0, 128))
elif len(k.full_shape) == 1:
for sz in [128,64,32]:
if k.full_shape[0]%sz == 0:
k.apply_opt(Opt(OptOps.UPCAST, 0, sz))
break
"""
#if knum in [7, 11, 14, 18]: