From 2454bf01c331fceba235df0eb4ec05fe0be74586 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Mon, 27 Jan 2025 07:20:21 +0900 Subject: [PATCH] hotfix: remove shapetracker spam in viz --- tinygrad/shape/shapetracker.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tinygrad/shape/shapetracker.py b/tinygrad/shape/shapetracker.py index 4fff017555..6897a6cc68 100644 --- a/tinygrad/shape/shapetracker.py +++ b/tinygrad/shape/shapetracker.py @@ -115,8 +115,9 @@ class ShapeTracker: def unit_stride_axes(self, ignore_valid=False) -> list[int]: return [i for i,st in enumerate(self.real_strides(ignore_valid)) if st == 1] def axis_is_masked(self, axis:int) -> bool: - _, valid = self.to_indexed_uops() - return axis in [x.arg for x in graph_rewrite(valid, symbolic_flat).toposort if x.op is Ops.RANGE] + with Context(TRACK_MATCH_STATS=0): + _, valid = self.to_indexed_uops() + return axis in [x.arg for x in graph_rewrite(valid, symbolic_flat).toposort if x.op is Ops.RANGE] def simplify(self) -> ShapeTracker: if len(self.views) >= 2 and (new_view := self.views[-2] + self.views[-1]) is not None: