nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

e kernels

Browse Source
This commit is contained in:
geohot
2025-03-26 16:49:23 +08:00
parent f7b38fa94c
commit 9e19cdfbbe
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/replay_pkl.py
+5 -2
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]: