add assertion to check for double caching

inducer · Feb 27, 2025 · 9bc0a14 · 9bc0a14
1 parent 99b1d47
commit 9bc0a14
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytato/transform/__init__.py b/pytato/transform/__init__.py
@@ -349,6 +349,9 @@ def add(
             else:
                 key = self.get_key(key_inputs)
 
+        assert key not in self._expr_key_to_result, \
+            f"Cache entry is already present for key '{key}'."
+
         self._expr_key_to_result[key] = result
 
         return result