add collision/duplication checks to CachedMapper/TransformMapper/Tran…

…sformMapperWithExtraArgs
inducer · Jan 28, 2025 · 6ac2ab3 · 6ac2ab3
1 parent 4fc8fa4
commit 6ac2ab3
Show file tree

Hide file tree

Showing 3 changed files with 314 additions and 22 deletions.
diff --git a/pytato/distributed/partition.py b/pytato/distributed/partition.py
@@ -296,7 +296,7 @@ def map_distributed_send(self, expr: DistributedSend) -> DistributedSend:
     def rec(self, expr: ArrayOrNames) -> ArrayOrNames:
         key = self._cache.get_key(expr)
         try:
-            return self._cache.retrieve(expr, key=key)
+            return self._cache_retrieve(expr, key=key)
         except KeyError:
             pass