Fix dygraph exit issue (#24347)

* add clean up; test=develop

* add import; test=develop
revert-24314-dev/fix_err_msg
hong 5 years ago committed by GitHub
parent 1ed8baf9c8
commit 5f65d9d58c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -19,6 +19,7 @@ import sys
import numpy as np import numpy as np
from paddle.fluid import core from paddle.fluid import core
from paddle.fluid import framework from paddle.fluid import framework
from paddle.fluid.multiprocess_utils import CleanupFuncRegistrar
from .tracer import Tracer from .tracer import Tracer
import logging import logging
import objgraph import objgraph
@ -134,6 +135,9 @@ def enable_dygraph(place=None):
_functional_dygraph_context_manager = guard(place=place) _functional_dygraph_context_manager = guard(place=place)
_functional_dygraph_context_manager.__enter__() _functional_dygraph_context_manager.__enter__()
# call disable_dygraph when Python exit
CleanupFuncRegistrar.register(disable_dygraph)
def disable_dygraph(): def disable_dygraph():
""" """

Loading…
Cancel
Save