fix mem opt

7 years ago · fcc1ffab5d
parent 2256fae45d
commit fcc1ffab5d
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/transpiler/memory_optimization_transpiler.py
+++ b/python/paddle/fluid/transpiler/memory_optimization_transpiler.py
@ -171,7 +171,7 @@ class ControlFlowGraph(object):
                self._live_out[i] |= self._live_in[s]
            self._live_in[i] = self._uses[i] | (
                self._live_out[i] - self._defs[i])
-            if live_in[i] != self._live_in[i]:
+            if live_in[i] != set(self._live_in[i]):
                for d in self._presuccessors[i]:
                    worklist.append(d)