diff --git a/src/gt4py/next/program_processors/runners/dace/transformations/gpu_utils.py b/src/gt4py/next/program_processors/runners/dace/transformations/gpu_utils.py index 6359cc1127..10ed652ec2 100644 --- a/src/gt4py/next/program_processors/runners/dace/transformations/gpu_utils.py +++ b/src/gt4py/next/program_processors/runners/dace/transformations/gpu_utils.py @@ -718,9 +718,9 @@ def apply(self, graph: Union[dace.SDFGState, dace.SDFG], sdfg: dace.SDFG) -> Non if not self.do_not_fuse: gtx_transformations.MapFusionSerial.apply_to( sdfg=sdfg, - map_exit_1=trivial_map_exit, - intermediate_access_node=access_node, - map_entry_2=second_map_entry, + first_map_exit=trivial_map_exit, + array=access_node, + second_map_entry=second_map_entry, verify=True, )