wl1318
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Only release runtime resource in GRAPH_MODE

This commit is contained in:
caifubi 2020-05-28 22:15:29 +08:00
parent f6ad679ef9
commit 6c491b8d3e
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/ccsrc/session/kernel_graph.cc
View File

@ -712,6 +712,14 @@ void KernelGraph::UpdateCallRealInput() {
std::string KernelGraph::ToString() const { return std::string("kernel_graph_").append(std::to_string(graph_id_)); }
KernelGraph::~KernelGraph() { device::KernelRuntimeManager::Instance().ClearGraphResource(graph_id_); }
KernelGraph::~KernelGraph() {
auto context = MsContext::GetInstance();
if (!context) {
return;
}
if (context->execution_mode() == kGraphMode) {
device::KernelRuntimeManager::Instance().ClearGraphResource(graph_id_);
}
}
} // namespace session
} // namespace mindspore