Browse Source

reset pynative flag in the resource clear

tags/v0.6.0-beta
limingqi107 5 years ago
parent
commit
66eefc0271
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      mindspore/ccsrc/pynative/pynative_execute.cc

+ 5
- 0
mindspore/ccsrc/pynative/pynative_execute.cc View File

@@ -902,6 +902,11 @@ void PynativeExecutor::Clear(const std::string &flag) {
if (flag == "resource") { if (flag == "resource") {
MS_LOG(INFO) << "Clear res"; MS_LOG(INFO) << "Clear res";
Clean(); Clean();
// Maybe exit in the pynative runing op, so need reset pynative flag.
auto ms_context = MsContext::GetInstance();
if (ms_context != nullptr) {
ms_context->set_enable_pynative_infer(false);
}
return; return;
} }
MS_LOG(INFO) << "Clear"; MS_LOG(INFO) << "Clear";


Loading…
Cancel
Save