From d453b0dcf72d50501bc59309a971c172ef148e31 Mon Sep 17 00:00:00 2001 From: dzhwinter Date: Thu, 14 Feb 2019 16:02:38 +0800 Subject: [PATCH] add details. test=develop --- .../paddle/fluid/transpiler/memory_optimization_transpiler.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/paddle/fluid/transpiler/memory_optimization_transpiler.py b/python/paddle/fluid/transpiler/memory_optimization_transpiler.py index 52c1aea28..047e0832b 100755 --- 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_dtyp are different") + continue if not compare_shape(x_shape, cache_shape, level): continue -- GitLab