github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

simpler unbind_view [pr] (#10636)

Browse Source
This commit is contained in:
chenyu
2025-06-05 01:03:27 -04:00
committed by GitHub
parent d0969f5a1f
commit f6d7db25b7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/engine/schedule.py
View File

@@ -21,7 +21,8 @@ def unbind_view(ctx:list[dict[Variable, int]], x:UOp):
if any(x.op is Ops.BIND for x in st.vars()):
st, var_vals = st.unbind()
ctx.append(var_vals)
return x.replace(arg=st) if st != x.st else None
return x.replace(arg=st)
return None
def unbind_bind(ctx:list[dict[Variable, int]], x:UOp):
var, val = x.unbind()