API(program_guard) error message enhancement (#23452)

* API(program_guard) error message enhancement test=develop

* fix Cycle import test=develop

* polish code according comment test=develop
revert-23830-2.0-beta
Aurelius84 5 years ago committed by GitHub
parent a00d0cb9f9
commit c9b3bdbfcb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -4987,12 +4987,12 @@ def program_guard(main_program, startup_program=None):
data = fluid.data(name='image', shape=[None, 784, 784], dtype='float32')
"""
if not isinstance(main_program, Program):
raise TypeError("main_program should be Program")
from .data_feeder import check_type
check_type(main_program, 'main_program', Program, 'fluid.program_guard')
main_program = switch_main_program(main_program)
if startup_program is not None:
if not isinstance(startup_program, Program):
raise TypeError("startup_program should be Program")
check_type(startup_program, 'startup_program', Program,
'fluid.program_guard')
startup_program = switch_startup_program(startup_program)
yield
switch_main_program(main_program)

Loading…
Cancel
Save