Skip to content

Commit 489b32c

Browse files
committed
refact(net): simpilify del-instruction loop
1 parent b8377ca commit 489b32c

1 file changed

Lines changed: 7 additions & 10 deletions

File tree

graphkit/network.py

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -126,19 +126,16 @@ def compile(self):
126126
# Add instructions to delete predecessors as possible. A
127127
# predecessor may be deleted if it is a data placeholder that
128128
# is no longer needed by future Operations.
129-
for predecessor in self.graph.predecessors(node):
129+
for need in self.graph.pred[node]:
130130
if self._debug:
131-
print("checking if node %s can be deleted" % predecessor)
132-
predecessor_still_needed = False
131+
print("checking if node %s can be deleted" % need)
133132
for future_node in ordered_nodes[i+1:]:
134-
if isinstance(future_node, Operation):
135-
if predecessor in future_node.needs:
136-
predecessor_still_needed = True
137-
break
138-
if not predecessor_still_needed:
133+
if isinstance(future_node, Operation) and need in future_node.needs:
134+
break
135+
else:
139136
if self._debug:
140-
print(" adding delete instruction for %s" % predecessor)
141-
self.steps.append(DeleteInstruction(predecessor))
137+
print(" adding delete instruction for %s" % need)
138+
self.steps.append(DeleteInstruction(need))
142139

143140
else:
144141
raise TypeError("Unrecognized network graph node")

0 commit comments

Comments
 (0)