From fdc68d0d65e664e077fb231d078b0a1ba223c1be Mon Sep 17 00:00:00 2001 From: Lukas Truemper Date: Mon, 30 Oct 2023 21:39:49 +0100 Subject: [PATCH] formatting done --- dace/transformation/dataflow/wcr_conversion.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/dace/transformation/dataflow/wcr_conversion.py b/dace/transformation/dataflow/wcr_conversion.py index 0dcb6e2fc5..85a4690912 100644 --- a/dace/transformation/dataflow/wcr_conversion.py +++ b/dace/transformation/dataflow/wcr_conversion.py @@ -11,6 +11,7 @@ from dace.transformation import helpers from dace import propagate_memlet + class AugAssignToWCR(transformation.SingleStateTransformation): """ Converts an augmented assignment ("a += b", "a = a + b") into a tasklet @@ -154,9 +155,7 @@ def apply(self, state: SDFGState, sdfg: SDFG): # If state fission is necessary to keep semantics, do it first if state.in_degree(input) > 0: - subgraph_nodes = set( - [e.src for e in state.bfs_edges(input, reverse=True)] - ) + subgraph_nodes = set([e.src for e in state.bfs_edges(input, reverse=True)]) subgraph_nodes.add(input) subgraph = StateSubgraphView(state, subgraph_nodes)