fix bug in memory optimization transpiler (#8233)

7 years ago · 931375ffeb
parent 6c3b78b7cf
commit 931375ffeb
1 changed files with 3 additions and 0 deletions
--- a/python/paddle/v2/fluid/memory_optimization_transpiler.py
+++ b/python/paddle/v2/fluid/memory_optimization_transpiler.py
@ -155,6 +155,9 @@ class ControlFlowGraph(object):
                    for x in defs_can_optimize
                ]
                for x, x_shape in out_pair:
+                    # If x is both in uses and defs, it can not be optimized!
+                    if x in self._uses[i]:
+                        continue
                    for index, cache_pair in enumerate(self.pool):
                        cache_var = cache_pair[0]
                        cache_shape = cache_pair[1]