From 43346b319fa701a23aaf1c82cc3011c9aa0d2c06 Mon Sep 17 00:00:00 2001 From: Edoardo Paone Date: Wed, 31 Jan 2024 14:43:21 +0100 Subject: [PATCH] [dace] Fix formatting --- .../runners/dace_iterator/itir_to_tasklet.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py index 5b9f48bfcd..053f8b0c7e 100644 --- a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py +++ b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_tasklet.py @@ -1192,9 +1192,11 @@ def _visit_reduce(self, node: itir.FunCall): ) input_mapping = { - param: dace.Memlet(data=arg.value.data, subset="0") - if arg.value.desc(self.context.body).shape == (1,) - else create_memlet_at(arg.value.data, nreduce_index) + param: ( + dace.Memlet(data=arg.value.data, subset="0") + if arg.value.desc(self.context.body).shape == (1,) + else create_memlet_at(arg.value.data, nreduce_index) + ) for (param, _), arg in zip(inner_inputs, args) } output_mapping = {