Sync

xdslproject · Aug 21, 2024 · a2a6922 · a2a6922
1 parent 9c10636
commit a2a6922
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/devito/ir/xdsl_iet/cluster_to_ssa.py b/devito/ir/xdsl_iet/cluster_to_ssa.py
@@ -384,12 +384,13 @@ def build_stencil_step(self, dim: SteppingDimension, eq: LoweredEq) -> None:
 
         lb = stencil.IndexAttr.get(*([0] * len(shape)))
         ub = stencil.IndexAttr.get(*shape)
-        stencil.StoreOp.get(
+
+        store = stencil.StoreOp.get(
             apply.res[0],
-            self.function_args[self.out_time_buffer],
+            self.function_values[self.out_time_buffer],
             stencil.StencilBoundsAttr(zip(lb, ub)),
         )
-        load = stencil.LoadOp.get(self.function_args[self.out_time_buffer])
+        load = stencil.LoadOp.get(self.function_values[self.out_time_buffer])
         load.res.name_hint = f"{write_function.name}_t{self.out_time_buffer[1]}_temp"  # noqa
         self.temps[self.out_time_buffer] = load.res