From 9bc413f104d019edab38ec7979ea8477e7e3924a Mon Sep 17 00:00:00 2001 From: chenyu Date: Tue, 29 Jul 2025 13:29:37 -0400 Subject: [PATCH] remove ShapeTracker.to_uop [pr] (#11418) --- tinygrad/shape/shapetracker.py | 1 - tinygrad/uop/ops.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/tinygrad/shape/shapetracker.py b/tinygrad/shape/shapetracker.py index bdc83b802f..dc70fca5b6 100644 --- a/tinygrad/shape/shapetracker.py +++ b/tinygrad/shape/shapetracker.py @@ -86,7 +86,6 @@ class ShapeTracker: def reduce(self, axis:tuple[int, ...]) -> tuple[sint, ...]: return tuple(1 if i in axis else s for i,s in enumerate(self.shape)) - def to_uop(self) -> UOp: return UOp(Ops.VIEW, dtypes.void, (), self) def to_indexed_uops(self, _idxs:list[UOp]|tuple[UOp, ...]|None=None) -> tuple[UOp, UOp]: return views_to_indexed_uops(self.views, tuple(_idxs) if _idxs is not None else None) diff --git a/tinygrad/uop/ops.py b/tinygrad/uop/ops.py index 1a8ca202b0..0ebdb33b87 100644 --- a/tinygrad/uop/ops.py +++ b/tinygrad/uop/ops.py @@ -250,7 +250,7 @@ class UOp(MathTrait, metaclass=UOpMetaClass): ret = UOp(Ops.VCONST if isinstance(b, tuple) else Ops.CONST, dtype, arg=dtypes.as_const(b, dtype)) if shape is not None: from tinygrad.shape.shapetracker import ShapeTracker - ret = ret.replace(src=(ShapeTracker.from_shape(()).reshape((1,)*len(shape)).expand(shape).to_uop(),)) + ret = ret.replace(src=(UOp(Ops.VIEW, dtypes.void, (), ShapeTracker.from_shape(()).reshape((1,)*len(shape)).expand(shape)),)) if device is not None: ret = ret.replace(src=(UOp(Ops.DEVICE, arg=device).view(unwrap(ret.st)),)) return ret