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

minor l2fetch tweak

Browse Source
This commit is contained in:
geohot
2025-04-02 10:09:24 +08:00
parent 17f7b226cb
commit 147fc0e648
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_dsp.py
+1 -1
View File
@@ -236,7 +236,7 @@ def prefetch_l2(ld:UOp, idx:UOp):
nlen_uop = (nidx.substitute({ranges[-1]: ranges[-1].src[1], **zero_ranges}) -
nidx.substitute({ranges[-1]: ranges[-1].src[0], **zero_ranges})).simplify()
nidx = nidx.substitute({ranges[-1]: ranges[-1].src[0]})
buf_lines_total = (idx.src[0].dtype.size+127)//128
buf_lines_total = ((idx.src[0].dtype.size*idx.src[0].dtype.itemsize)+127)//128
if buf_lines_total < 8192//128:
# if the total buffer size is sub 8k, fetch it all
x1 = UOp(Ops.CUSTOM, dtypes.void, src=(idx.src[0], UOp.const(dtypes.int, buf_lines_total)),