File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments