p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!24562 Enable MD cleanup at module exit 

Merge pull request !24562 from luoyang/cleanup
This commit is contained in:
i-robot 2021-10-11 08:09:30 +00:00 committed by Gitee
parent fd0f1e3eaa be71ae982b
commit ebf2cc4041
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ccsrc/pipeline/jit/init.cc
View File

@ -302,16 +302,15 @@ PYBIND11_MODULE(_c_expression, m) {
.def("reset_algo_parameters", &CostModelContext::ResetAlgoParameters, "Reset the AlgoParameters.");
(void)py::module::import("atexit").attr("register")(py::cpp_function{[&]() -> void {
#ifdef ENABLE_MINDDATA
MS_LOG(INFO) << "Start releasing dataset handles...";
py::module iterators = py::module::import("mindspore.dataset.engine.iterators");
(void)iterators.attr("_cleanup")();
MS_LOG(INFO) << "End release dataset handles.";
#endif
// only in case that c++ calling python interface, ClearResAtexit should be called.
if (mindspore::parse::python_adapter::IsPythonEnv()) {
mindspore::pipeline::ClearResAtexit();
#ifdef ENABLE_MINDDATA
MS_LOG(INFO) << "Start releasing dataset handles...";
py::module iterators = py::module::import("mindspore.dataset.engine.iterators");
(void)iterators.attr("_cleanup")();
MS_LOG(INFO) << "End release dataset handles.";
#endif
}
}});