Return correct currrent block of a var (#19850)

5 years ago · 3e1e1fee26
parent 1bc285a53a
commit 3e1e1fee26
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/math_op_patch.py
+++ b/python/paddle/fluid/layers/math_op_patch.py
@ -40,7 +40,7 @@ def monkey_patch_variable():
        return dtype

    def current_block(var):
-        return var.block.program.current_block()
+        return var.block

    def create_new_tmp_var(block, dtype):
        tmp_name = unique_tmp_name()