Fix traversal for defined symbols

spcl · Nov 13, 2024 · 40d4a12 · 40d4a12
1 parent 7e4bc3d
commit 40d4a12
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dace/sdfg/utils.py b/dace/sdfg/utils.py
@@ -1557,6 +1557,10 @@ def _tswds_cf_region(
         for _, b in region.branches:
             yield from _tswds_cf_region(sdfg, b, symbols, recursive)
         return
+    elif isinstance(region, LoopRegion):
+        # Add the own loop variable to the defined symbols, if present.
+        loop_syms = region.new_symbols(symbols)
+        symbols.update({k: v for k, v in loop_syms.items() if v is not None})
 
     # Add symbols from inter-state edges along the state machine
     start_region = region.start_block