Skip to content

Commit

Permalink
Don't filter by defined symbols.
Browse files Browse the repository at this point in the history
  • Loading branch information
alexnick83 committed Nov 12, 2023
1 parent 22e3198 commit cbaade3
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions dace/codegen/targets/cuda.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,12 +202,13 @@ def preprocess(self, sdfg: SDFG) -> None:
and node.map.schedule in (dtypes.ScheduleType.GPU_Device, dtypes.ScheduleType.GPU_Persistent)):
if state.parent not in shared_transients:
shared_transients[state.parent] = state.parent.shared_transients()
sgraph = state.scope_subgraph(node)
used_symbols = sgraph.used_symbols(all_symbols=False)
arglist = sgraph.arglist(defined_syms, shared_transients[state.parent])
arglist = {k: v for k, v in arglist.items() if not k in defined_syms or k in used_symbols}
self._arglists[node] = arglist
# self._arglists[node] = state.scope_subgraph(node).arglist(defined_syms, shared_transients[state.parent])
# sgraph = state.scope_subgraph(node)
# used_symbols = sgraph.used_symbols(all_symbols=False)
# arglist = sgraph.arglist(defined_syms, shared_transients[state.parent])
# arglist = {k: v for k, v in arglist.items() if not k in defined_syms or k in used_symbols}
# self._arglists[node] = arglist
# TODO/NOTE: Did we change defined_syms?
self._arglists[node] = state.scope_subgraph(node).arglist(defined_syms, shared_transients[state.parent])

def _compute_pool_release(self, top_sdfg: SDFG):
"""
Expand Down

0 comments on commit cbaade3

Please sign in to comment.