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

remove support for VIEW with two sources in ops [pr] (#9168)

Browse Source 
* only 1 src views can exist [pr]

* views can still exist without a base, this is a separate project
This commit is contained in:
qazal
2025-02-19 11:10:18 +01:00
committed by GitHub
parent a2afa523a0
commit 2fc8bf115d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/ops.py
+2 -2
View File
@@ -509,7 +509,7 @@ class UOp(MathTrait, metaclass=UOpMetaClass):
@property
def base(self) -> UOp:
if (self.op is Ops.VIEW and len(self.src) == 1) or self.op in GroupOp.Movement: return self.src[0].base
if (self.op is Ops.VIEW and len(self.src) != 0) or self.op in GroupOp.Movement: return self.src[0].base
return self
def view(self, new_st:ShapeTracker) -> UOp: return UOp(Ops.VIEW, self.dtype, (self.base,), new_st)
@@ -546,7 +546,7 @@ class UOp(MathTrait, metaclass=UOpMetaClass):
@property
def buf_uop(self) -> UOp:
if self.base.op is Ops.BUFFER: return self.base
assert self.base.op in {*GroupOp.Buffer, Ops.ASSIGN, Ops.VIEW}, f"buf_uop called on {self.op}"
assert self.base.op in {*GroupOp.Buffer, Ops.ASSIGN}, f"buf_uop called on {self.op}"
return self.src[0].buf_uop
@property
def buffer(self) -> Buffer: