sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 4 Activity

oops

Browse Source
This commit is contained in:
geohot
2025-04-01 23:44:44 +08:00
parent 6ecaf11224
commit f3cb4c3eef
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/kernel.py
+1 -1
View File
@@ -447,7 +447,7 @@ class Kernel:
k.apply_opt(Opt(OptOps.UNROLL, 0, 0))
k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-3, 32))
if k.full_shape[len(k.full_shape)-4]%4 == 0:
if len(k.full_shape)-4 <= 8: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 0))
if k.full_shape[len(k.full_shape)-4] <= 8: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 0))
else: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 4))
# if this is small, swap it
# NOTE: this is breaking something (should be fixed w/o padto)