From 12b5b83821785194a6fedff98ec0fd4bcedffdb2 Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Sun, 23 Feb 2025 23:26:31 +0200 Subject: [PATCH] set TRACK_MATCH_STATS=0 for real_strides [pr] (#9216) --- tinygrad/shape/shapetracker.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tinygrad/shape/shapetracker.py b/tinygrad/shape/shapetracker.py index 3d245f7b7c..4802576e1a 100644 --- a/tinygrad/shape/shapetracker.py +++ b/tinygrad/shape/shapetracker.py @@ -112,7 +112,8 @@ class ShapeTracker: if all(len(x) == 0 for x in var_vals): return self, {} return ShapeTracker(tuple(unbound_views)), merge_dicts(var_vals) - def real_strides(self, ignore_valid=False) -> tuple[Optional[sint], ...]: return views_to_real_strides(self.views, ignore_valid) + def real_strides(self, ignore_valid=False) -> tuple[Optional[sint], ...]: + with Context(TRACK_MATCH_STATS=0): return views_to_real_strides(self.views, ignore_valid) 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: