Fix missing updates to the GPU cache dictionary/set

SeisSol · Sep 13, 2023 · f8ecf51 · f8ecf51
1 parent 9402324
commit f8ecf51
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yateto/ast/cost.py b/yateto/ast/cost.py
@@ -126,13 +126,13 @@ def estimate_Product(self, node):
     self._loaded_to_gpu_cache[node] = self._loaded_to_gpu_cache[left_term].union(self._loaded_to_gpu_cache[right_term])
 
     extra_cost = 0
-    if not right_term in self._loaded_to_gpu_cache:
+    if not right_term in self._loaded_to_gpu_cache[node]:
       self._loaded_to_gpu_cache[node].add(right_term)
       rbb = self._cache[right_term]
       extra_cost += rbb.size()
 
     if node.indices[self._lead_dim] != left_term.indices[self._lead_dim]:
-      if not node.leftTerm in self._loaded_to_gpu_cache:
+      if not node.leftTerm in self._loaded_to_gpu_cache[node]:
         self._loaded_to_gpu_cache[node].add(left_term)
         lbb = self._cache[left_term]
         extra_cost += lbb.size()