Browse Source

!1236 move set stage in finalize

From: @wangxiaotian22
Reviewed-by: @ji_chen,@wqtshg
Signed-off-by: @wqtshg
tags/v1.2.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
992c791c7c
1 changed files with 6 additions and 6 deletions
  1. +6
    -6
      ge/client/ge_api.cc

+ 6
- 6
ge/client/ge_api.cc View File

@@ -171,17 +171,17 @@ Status GEInitialize(const std::map<AscendString, AscendString> &options) {

// GE finalize, releasing all resources
Status GEFinalize() {
ErrorManager::GetInstance().SetStage(ErrorMessage::kFinalize, ErrorMessage::kFinalize);
GELOGT(TRACE_INIT, "GEFinalize start");

ErrorManager::GetInstance().GenWorkStreamIdDefault();
std::lock_guard<std::mutex> lock(g_ge_release_mutex);
// check init status
if (!g_ge_initialized) {
GELOGW("GEFinalize is called before GEInitialize");
GELOGW("[FINAL][FINAL]GEFinalize is called before GEInitialize");
return SUCCESS;
}

std::lock_guard<std::mutex> lock(g_ge_release_mutex);
ErrorManager::GetInstance().SetStage(ErrorMessage::kFinalize, ErrorMessage::kFinalize);
ErrorManager::GetInstance().GenWorkStreamIdDefault();
GELOGT(TRACE_INIT, "GEFinalize start");

// call Finalize
Status ret = SUCCESS;
Status middle_ret;


Loading…
Cancel
Save