optimize data flow analysis (#8271)

8 years ago · b5ffe5bce2
parent 73f1b47189
commit b5ffe5bce2
1 changed files with 3 additions and 4 deletions
--- a/python/paddle/v2/fluid/memory_optimization_transpiler.py
+++ b/python/paddle/v2/fluid/memory_optimization_transpiler.py
@ -92,14 +92,13 @@ class ControlFlowGraph(object):
        live_in = defaultdict(set)
        live_out = defaultdict(set)
        while True:
-            for i in range(self.op_size):
+            for i in range(self.op_size, 0, -1):
                live_in[i] = set(self._live_in[i])
                live_out[i] = set(self._live_out[i])
-                self._live_in[i] = self._uses[i] | (
-                    self._live_out[i] - self._defs[i])
                for s in self._successors[i]:
                    self._live_out[i] |= self._live_in[s]
-
+                self._live_in[i] = self._uses[i] | (
+                    self._live_out[i] - self._defs[i])
            if self._reach_fixed_point(live_in, live_out):
                break