浏览代码

!4292 fix bugs in context_extend cc file

Merge pull request !4292 from liubuyu/bug_fix2
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 年前
父节点
当前提交
daefafbee6
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. +3
    -3
      mindspore/ccsrc/utils/context/context_extends.cc

+ 3
- 3
mindspore/ccsrc/utils/context/context_extends.cc 查看文件

@@ -329,7 +329,7 @@ bool FinalizeGe(const std::shared_ptr<MsContext> &ms_context_ptr, bool force) {
if (ge::GEFinalize() != ge::GRAPH_SUCCESS) {
MS_LOG(WARNING) << "Finalize GE failed!";
}
ms_context_ptr->set_pynative_ge_init(true);
ms_context_ptr->set_pynative_ge_init(fasle);
} else {
MS_LOG(INFO) << "Ge is used, no need to finalize, tsd reference = " << ms_context_ptr->ge_ref() << ".";
}
@@ -341,14 +341,14 @@ bool IsTsdOpened(const std::shared_ptr<MsContext> &ms_context_ptr) {
if (ms_context_ptr == nullptr) {
MS_LOG(EXCEPTION) << "nullptr";
}
return ms_context_ptr->tsd_ref();
return ms_context_ptr->IsTsdOpened();
}

bool IsGeInited(const std::shared_ptr<MsContext> &ms_context_ptr) {
if (ms_context_ptr == nullptr) {
MS_LOG(EXCEPTION) << "nullptr";
}
return ms_context_ptr->ge_ref();
return ms_context_ptr->IsGeInited();
}

// Register for device type.


正在加载...
取消
保存