From a2a69225f2f0bbf35f105336ce42a0058afdcecf Mon Sep 17 00:00:00 2001 From: Emilien Bauer Date: Wed, 21 Aug 2024 15:01:10 +0100 Subject: [PATCH] Sync --- devito/ir/xdsl_iet/cluster_to_ssa.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/devito/ir/xdsl_iet/cluster_to_ssa.py b/devito/ir/xdsl_iet/cluster_to_ssa.py index f2f2383bb6..11c4d8a718 100644 --- 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