From 931375ffeb86b1520090c21383ab2d38ba2aa5eb Mon Sep 17 00:00:00 2001 From: QI JUN Date: Wed, 7 Feb 2018 21:22:40 +0800 Subject: [PATCH] fix bug in memory optimization transpiler (#8233) --- python/paddle/v2/fluid/memory_optimization_transpiler.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/paddle/v2/fluid/memory_optimization_transpiler.py b/python/paddle/v2/fluid/memory_optimization_transpiler.py index 11e2cfb3cc..8bb8cf7b1a 100644 --- 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] -- GitLab