From f0f95525a93a3fed41c4c409a0703af8b91bfe24 Mon Sep 17 00:00:00 2001 From: Edoardo Paone Date: Mon, 2 Oct 2023 14:51:48 +0200 Subject: [PATCH] [dace] Remove unrelated changes --- .../program_processors/runners/dace_iterator/itir_to_sdfg.py | 4 ++-- .../runners/dace_iterator/itir_to_tasklet.py | 5 +---- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_sdfg.py b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_sdfg.py index 6104915810..92d9b598ce 100644 --- a/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_sdfg.py +++ b/src/gt4py/next/program_processors/runners/dace_iterator/itir_to_sdfg.py @@ -244,8 +244,8 @@ def visit_FencilDefinition(self, node: itir.FencilDefinition): def visit_StencilClosure( self, node: itir.StencilClosure, array_table: dict[str, dace.data.Array] ) -> dace.SDFG: - assert ItirToSDFG._check_no_lifts(node), "Lifts not supported" - assert ItirToSDFG._check_shift_offsets_are_literals(node), "Shift offsets must be literals" + assert ItirToSDFG._check_no_lifts(node) + assert ItirToSDFG._check_shift_offsets_are_literals(node) neighbor_tables = filter_neighbor_tables(self.offset_provider) input_names = [str(inp.id) for inp in node.inputs] 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 49c128089c..118ab7a3a9 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 @@ -713,10 +713,7 @@ def _visit_indirect_addressing(self, node: itir.FunCall) -> IteratorExpr: element = tail[1].value assert isinstance(element, int) - assert isinstance( - self.offset_provider[offset], NeighborTableOffsetProvider - ), "StridedNeighborOffsetProvider not supported" - table = self.offset_provider[offset] + table: NeighborTableOffsetProvider = self.offset_provider[offset] shifted_dim = table.origin_axis.value target_dim = table.neighbor_axis.value