Relase runtime resource only in graph_mode

pull/1619/head
caifubi 5 years ago
parent 5f2bf782f1
commit a3b80d34d1

@ -719,6 +719,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

Loading…
Cancel
Save