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

Merge branch 'master' into cstyle_new_style

Browse Source
This commit is contained in:
George Hotz
2026-06-07 12:07:24 -07:00
committed by GitHub
parent 8bed25fbdd 0336ba8eb1
commit ba1ef70bfd
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/uop/ops.py
View File

@@ -771,9 +771,7 @@ class UOp(OpMixin, metaclass=UOpMetaClass):
@recursive_property
def addrspace(self) -> AddrSpace|None:
if self.op is Ops.PARAM: return self.arg.addrspace
if self.op is Ops.BUFFER:
if isinstance(self.arg, ParamArg): return self.arg.addrspace
return AddrSpace.GLOBAL
if self.op is Ops.BUFFER: return self.arg.addrspace if isinstance(self.arg, ParamArg) else AddrSpace.GLOBAL
if self.op is Ops.DEFINE_LOCAL: return AddrSpace.LOCAL
if self.op is Ops.DEFINE_REG: return AddrSpace.REG
if self.op is Ops.LOAD: return AddrSpace.REG # LOAD brings things into registers
@@ -870,6 +868,7 @@ class UOp(OpMixin, metaclass=UOpMetaClass):
def realized(self) -> Buffer|MultiBuffer|None:
# only these can be realized
if self.op not in (Ops.BUFFER, Ops.MSTACK): return None
# ParamArg is LOCAL/REG and never realized
if self.op is Ops.BUFFER and isinstance(self.arg, ParamArg): return None
# LUNIQUEs are never realized
if self.op_in_backward_slice_with_self(Ops.LUNIQUE): return None