Skip to content

Commit ff6c040

Browse files
authored
Fix dygraph exit issue (#24347) (#24838)
* add clean up; test=develop * add import; test=develop
1 parent c0b7241 commit ff6c040

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

python/paddle/fluid/dygraph/base.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import numpy as np
2020
from paddle.fluid import core
2121
from paddle.fluid import framework
22+
from paddle.fluid.multiprocess_utils import CleanupFuncRegistrar
2223
from .tracer import Tracer
2324
import logging
2425
import objgraph
@@ -182,6 +183,9 @@ def enable_dygraph(place=None):
182183
_functional_dygraph_context_manager = guard(place=place)
183184
_functional_dygraph_context_manager.__enter__()
184185

186+
# call disable_dygraph when Python exit
187+
CleanupFuncRegistrar.register(disable_dygraph)
188+
185189

186190
def disable_dygraph():
187191
"""

0 commit comments

Comments
 (0)