fix bug

7 years ago · fe2d590d21
parent 142fac18ec
commit fe2d590d21
1 changed files with 9 additions and 7 deletions
--- a/python/paddle/fluid/memory_optimization_transpiler.py
+++ b/python/paddle/fluid/memory_optimization_transpiler.py
@ -172,8 +172,10 @@ class ControlFlowGraph(object):
                                    block_desc, cache_var, is_forward).dtype()
                                # TODO(qijun): actually, we should compare dtype_to_size[x_dtype]
                                # and dtype_to_size[cache_dtype]
-                                if x_dtype == cache_dtype and PRINT_LOG:
-                                    print(("Hit Cache !!!! cache pool index "
+                                if x_dtype == cache_dtype:
+                                    if PRINT_LOG:
+                                        print(
+                                            ("Hit Cache !!!! cache pool index "
                                             "is %d, var name is %s, "
                                             "cached var name is %s, "
                                             "var shape is %s ") %