Skip to content

Commit 2454bf0

Browse files
committed
hotfix: remove shapetracker spam in viz
1 parent d488bbb commit 2454bf0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

Diff for: tinygrad/shape/shapetracker.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -115,8 +115,9 @@ def real_strides(self, ignore_valid=False) -> tuple[Optional[sint], ...]: return
115115
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]
116116

117117
def axis_is_masked(self, axis:int) -> bool:
118-
_, valid = self.to_indexed_uops()
119-
return axis in [x.arg for x in graph_rewrite(valid, symbolic_flat).toposort if x.op is Ops.RANGE]
118+
with Context(TRACK_MATCH_STATS=0):
119+
_, valid = self.to_indexed_uops()
120+
return axis in [x.arg for x in graph_rewrite(valid, symbolic_flat).toposort if x.op is Ops.RANGE]
120121

121122
def simplify(self) -> ShapeTracker:
122123
if len(self.views) >= 2 and (new_view := self.views[-2] + self.views[-1]) is not None:

0 commit comments

Comments
 (0)