Skip to content

Commit

Permalink
Dace backend: keep shape, remove stride symbols from domain visitor
Browse files Browse the repository at this point in the history
  • Loading branch information
edopao committed May 31, 2024
1 parent 342bd82 commit 34e8d20
Showing 1 changed file with 2 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 34e8d20

Please sign in to comment.