Skip to content

Commit

Permalink
Merge branch 'host_map_host_data' of github.com:spcl/dace into host_m…
Browse files Browse the repository at this point in the history
…ap_host_data
  • Loading branch information
ThrudPrimrose committed Nov 1, 2024
2 parents a588061 + f1b81c4 commit 1d8eb53
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions dace/transformation/interstate/gpu_transform_sdfg.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ def can_be_applied(self, graph, expr_index, sdfg, permissive=False):
return True

def _get_marked_inputs_and_outputs(self, state, entry_node) -> list:
if (self.host_data is None or self.host_data == []) and (self.host_maps is None or self.host_maps == []):
if not self.host_data and not self.host_maps:
return []
marked_sources = [state.memlet_tree(e).root().edge.src for e in state.in_edges(entry_node)]
marked_sources = [sdutil.get_view_node(state, node) if isinstance(node, data.View) else node for node in marked_sources]
Expand All @@ -172,7 +172,7 @@ def _get_marked_inputs_and_outputs(self, state, entry_node) -> list:

def _output_or_input_is_marked_host(self, state, entry_node) -> bool:
marked_accesses = self._get_marked_inputs_and_outputs(state, entry_node)
return (len(marked_accesses) > 0)
return len(marked_accesses) > 0


def apply(self, _, sdfg: sd.SDFG):
Expand Down

0 comments on commit 1d8eb53

Please sign in to comment.