From 34e8d20e549851c6ab2252d38c0c64db734419e1 Mon Sep 17 00:00:00 2001 From: Edoardo Paone Date: Fri, 31 May 2024 09:07:08 +0200 Subject: [PATCH] Dace backend: keep shape, remove stride symbols from domain visitor --- .../runners/dace_iterator/itir_to_sdfg.py | 7 ++----- 1 file changed, 2 insertions(+), 5 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 a5eeeed666..d213fc3f0b 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 @@ -455,14 +455,11 @@ def visit_StencilClosure( program_arg_syms[name] = SymbolExpr(name, dtype) else: assert isinstance(type_, ts.FieldType) - # make shape and stride symbols available as arguments to domain visitor + # make shape symbols (corresponding to field size) available as arguments to domain visitor if name in input_names or name in output_names: field_symbols = [ val - for val in [ - *closure_sdfg.arrays[name].shape, - *closure_sdfg.arrays[name].strides, - ] + for val in closure_sdfg.arrays[name].shape if isinstance(val, dace.symbol) and str(val) not in input_names ] for sym in field_symbols: