Fix prune symbols removing used symbols

spcl · Jan 21, 2025 · 4fe276f · 4fe276f
1 parent 3509351
commit 4fe276f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dace/transformation/passes/prune_symbols.py b/dace/transformation/passes/prune_symbols.py
@@ -111,6 +111,8 @@ def used_symbols(self, sdfg: SDFG) -> Set[str]:
                         if node.code_exit.language != dtypes.Language.Python:
                             result |= symbolic.symbols_in_code(node.code_exit.as_string, sdfg.symbols.keys(),
                                                             node.ignored_symbols)
+            else:
+                result |= block.used_symbols(all_symbols=True, with_contents=False)
 
         for e in sdfg.all_interstate_edges():
             result |= e.data.free_symbols