|
|
|
@ -1252,9 +1252,14 @@ void PynativeExecutor::GradNetInner(const GradOperationPtr &grad, const py::obje
|
|
|
|
|
void PynativeExecutor::Clear(const std::string &flag) {
|
|
|
|
|
if (!flag.empty()) {
|
|
|
|
|
MS_LOG(DEBUG) << "Clear res";
|
|
|
|
|
(void)graph_map_.erase(flag);
|
|
|
|
|
(void)cell_graph_map_.erase(flag);
|
|
|
|
|
(void)cell_resource_map_.erase(flag);
|
|
|
|
|
auto key_value = std::find_if(graph_map_.begin(), graph_map_.end(),
|
|
|
|
|
[&flag](const auto &item) { return item.first.find(flag) != std::string::npos; });
|
|
|
|
|
if (key_value != graph_map_.end()) {
|
|
|
|
|
std::string key = key_value->first;
|
|
|
|
|
(void)graph_map_.erase(key);
|
|
|
|
|
(void)cell_graph_map_.erase(key);
|
|
|
|
|
(void)cell_resource_map_.erase(key);
|
|
|
|
|
}
|
|
|
|
|
Clean();
|
|
|
|
|
// Maybe exit in the pynative runing op, so need reset pynative flag.
|
|
|
|
|
auto ms_context = MsContext::GetInstance();
|
|
|
|
|