simple ignore mop noops [pr] (tinygrad#8263)

chenyuxyz · Dec 15, 2024 · 53603c4 · 53603c4
1 parent e1518f1
commit 53603c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tinygrad/ops.py b/tinygrad/ops.py
@@ -482,7 +482,7 @@ def view(self, new_st:ShapeTracker) -> UOp:
 
   def _mop(self, op:Ops, arg):
     ret = UOp(op, self.dtype, (self,), arg)
-    ret.st  # pylint: disable=pointless-statement
+    if self.st == ret.st: return self  # ignore NOOPs, also check ret.st
     return ret
 
   def reshape(self, arg:Tuple[sint, ...]): return self._mop(Ops.RESHAPE, arg)