nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

ignore that

Browse Source
This commit is contained in:
geohot
2025-08-13 09:43:08 -07:00
parent cbf85fbfd0
commit e3d8185ba4
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/uop/ops.py
-1
View File
@@ -144,7 +144,6 @@ class UOp(MathTrait, metaclass=UOpMetaClass):
if self.op in GroupOp.Movement: return unwrap(self.src[0].st).mop(self.op, self.arg)
# CONST with a DEVICE has a shape of ()
if self.op is Ops.CONST and len(self.src) and self.src[0].op is Ops.DEVICE: return ShapeTracker.from_shape(())
#if self.op is Ops.CONTIGUOUS and self.tag == 1: return ShapeTracker.from_shape(self.arg)
# BufferOps and ASSIGN flow ShapeTracker from a direct edge
if self.op in {Ops.STORE, Ops.ASSIGN, Ops.LOAD}: return self.src[0].st
if self.op in GroupOp.Buffer: return views[0] if (views:=[x.st for x in self.src if x.op is Ops.VIEW]) else None