[dace] Remove unrelated changes

edopao · Oct 2, 2023 · f0f9552 · f0f9552
1 parent 26f385c
commit f0f9552
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -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
@@ -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