From 89e774ce100a4ff5fb62fb5f295be499376721df Mon Sep 17 00:00:00 2001 From: Emilien Bauer Date: Wed, 21 Aug 2024 14:52:49 +0100 Subject: [PATCH] Dictionary tweak. --- devito/ir/xdsl_iet/cluster_to_ssa.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/devito/ir/xdsl_iet/cluster_to_ssa.py b/devito/ir/xdsl_iet/cluster_to_ssa.py index 48d2489d12..11c4d8a718 100644 --- a/devito/ir/xdsl_iet/cluster_to_ssa.py +++ b/devito/ir/xdsl_iet/cluster_to_ssa.py @@ -198,7 +198,7 @@ def _visit_math_nodes(self, dim: SteppingDimension, node: Expr, if output_indexed is not None: space_offsets = ([node.indices[d] - output_indexed.indices[d] for d in node.function.space_dimensions]) - temp = self.function_values[(node.function, time_offset)] + temp = self.apply_temps[(node.function, time_offset)] access = stencil.AccessOp.get(temp, space_offsets) return access.res # Otherwise, generate a load op @@ -378,8 +378,6 @@ def build_stencil_step(self, dim: SteppingDimension, eq: LoweredEq) -> None: apply_arg.name_hint = apply_op.name_hint.replace("temp", "blk") self.apply_temps = {k: v for k, v in zip(read_functions, apply.region.block.args)} - # Update the function values with the new temps - self.function_values |= self.apply_temps with ImplicitBuilder(apply.region.block): stencil.ReturnOp.get([self._visit_math_nodes(dim, eq.rhs, eq.lhs)])