Merge pull request #15714 from dzhwinter/fix/memory_with_dtype

fix optimize with different dtype
6 years ago · 0936097c1e
parent abcefe7211 fdb4c9fbf5
commit 0936097c1e
1 changed files with 4 additions and 0 deletions
--- a/python/paddle/fluid/transpiler/memory_optimization_transpiler.py
+++ b/python/paddle/fluid/transpiler/memory_optimization_transpiler.py
@ -355,6 +355,10 @@ class ControlFlowGraph(object):
                                                 is_forward).dtype()
                        cache_dtype = self._find_var(block_desc, cache_var,
                                                     is_forward).dtype()
+                        if x_dtype != cache_dtype:
+                            if PRINT_LOG:
+                                print("x_dtype and cache_dtype are different")
+                            continue

                        if not compare_shape(x_shape, cache_shape, level):
                            continue