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

optional l2 fetch

Browse source
This commit is contained in:
George Hotz 2025-03-28 18:12:48 +08:00
commit 45646fe102
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tinygrad/runtime/ops_dsp.py
View file

@ -193,6 +193,7 @@ def prefetch_l1(ld:UOp, idx:UOp):
return ld.replace(src=ld.src+(x1, x2))
def prefetch_l2(ld:UOp, idx:UOp):
if not getenv("PREFETCHL2", 0): return None
if ld.src[-1].op is Ops.CUSTOM: return None
ranges = sorted([x for x in ld.src[0].src[0].toposort if x.op is Ops.RANGE], key=lambda x: x.arg)
if len(ranges):