call Mapper.rec from CachedMapper too just to avoid copy/paste errors

inducer · Feb 25, 2025 · 99b1d47 · 99b1d47
1 parent 4ff4017
commit 99b1d47
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytato/transform/__init__.py b/pytato/transform/__init__.py
@@ -418,7 +418,7 @@ def rec(self, expr: ArrayOrNames, *args: P.args, **kwargs: P.kwargs) -> ResultT:
         except KeyError:
             return self._cache.add(
                 (expr, args, kwargs),
-                super().rec(expr, *args, **kwargs),
+                Mapper.rec(self, expr, *args, **kwargs),
                 key=key)
 
     def rec_function_definition(
@@ -430,7 +430,7 @@ def rec_function_definition(
         except KeyError:
             return self._function_cache.add(
                 (expr, args, kwargs),
-                super().rec_function_definition(expr, *args, **kwargs),
+                Mapper.rec_function_definition(self, expr, *args, **kwargs),
                 key=key)
 
     def clone_for_callee(