github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

oops, i didn't mean to change that

Browse Source
This commit is contained in:
George Hotz
2025-10-14 20:12:47 +08:00
parent accee5d840
commit 3b0b3dcff3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/uop/ops.py
View File

@@ -233,7 +233,7 @@ class UOp(MathTrait, metaclass=UOpMetaClass):
return None
# some ops init the shape
case Ops.CONST | Ops.DEFINE_VAR | Ops.BIND: return () if len(self.src) and self.src[0].op is Ops.DEVICE else None
case Ops.CONST | Ops.DEFINE_VAR | Ops.BIND: return () if self._device is not None else None
case Ops.BUFFER: return (self.arg,)
case Ops.BUFFER_VIEW: return (self.arg[0],)
case Ops.BUFFERIZE: return tuple([int(r.vmax+1) for r in self.src[1:]])