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

…sformMapperWithExtraArgs
inducer · Jan 10, 2025 · f90083b · f90083b
1 parent 073f504
commit f90083b
Show file tree

Hide file tree

Showing 4 changed files with 349 additions and 35 deletions.
diff --git a/pytato/distributed/partition.py b/pytato/distributed/partition.py
@@ -349,7 +349,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