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

…sformMapperWithExtraArgs
inducer · Sep 24, 2024 · 63c29f3 · 63c29f3
1 parent 4a206ad
commit 63c29f3
Show file tree

Hide file tree

Showing 4 changed files with 362 additions and 32 deletions.
diff --git a/pytato/distributed/partition.py b/pytato/distributed/partition.py
@@ -346,7 +346,7 @@ def map_distributed_send(self, expr: DistributedSend) -> DistributedSend:
     def rec(self, expr: ArrayOrNames) -> ArrayOrNames:  # type: ignore[override]
         key = self._cache.get_key(expr)
         try:
-            return self._cache.retrieve(expr, key=key)
+            return self._cache_retrieve(expr, key=key)
         except KeyError:
             pass