fix compiler_py bug (#16122)

test=develop
6 years ago · 2f1b3afa6f
parent db120b9392
commit 2f1b3afa6f
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/compiler.py
+++ b/python/paddle/fluid/compiler.py
@ -37,7 +37,7 @@ def _place_obj(place):

 def _is_pserver_mode(main_program):
    main = main_program if main_program \
-        else default_main_program()
+        else framework.default_main_program()
    for op in main.global_block().ops:
        if op.type in ["send", "recv"]:
            return True