mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-11 23:46:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' into shrink_in_render

Browse Source
This commit is contained in:
George Hotz
2026-06-01 19:25:15 -07:00
committed by GitHub
parent 2bcd46d946 20242fdf1d
commit a35c1c9c38
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/uop/ops.py
View File

@@ -1116,7 +1116,11 @@ class ProgramInfo:
if u.op is Ops.DEFINE_VAR: _vars.append(u)
if u.op is Ops.PARAM: _globals.append(u.arg.slot)
if u.op in (Ops.STORE, Ops.LOAD):
<<<<<<< shrink_in_render
if (idx:=u.src[0]).op in (Ops.INDEX, Ops.SHRINK):
=======
if (idx:=u.src[0]).op in (Ops.INDEX, Ops.SHRINK) or (u.src[0].op is Ops.CAST and (idx:=u.src[0].src[0]).op is Ops.INDEX):
>>>>>>> master
if (buf:=idx.src[0]).op is Ops.PARAM: (outs if u.op is Ops.STORE else ins).append(buf.arg.slot)
if u.op is Ops.SPECIAL:
if u.arg[0] == 'i': local_size = None