Merge pull request #15756 from dzhwinter/picked/fix_py_opt

cherry picked. test=develop

Author: dzhwinter

python/paddle/fluid/transpiler/memory_optimization_transpiler.py

@@ -355,6 +355,10 @@ def compare_shape(x_shape, cache_shape, opt_level):
                                                  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