diff --git a/mindspore/ccsrc/pynative/pynative_execute.cc b/mindspore/ccsrc/pynative/pynative_execute.cc index d65807b9f6..75653ff5d2 100644 --- a/mindspore/ccsrc/pynative/pynative_execute.cc +++ b/mindspore/ccsrc/pynative/pynative_execute.cc @@ -937,6 +937,11 @@ void PynativeExecutor::Clear(const std::string &flag) { if (flag == "resource") { MS_LOG(INFO) << "Clear res"; 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; } MS_LOG(INFO) << "Clear";