Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!4292 fix bugs in context_extend cc file 

Merge pull request !4292 from liubuyu/bug_fix2
This commit is contained in:
mindspore-ci-bot 2020-08-11 21:55:30 +08:00 committed by Gitee
parent 5ac8c40e91 aead81941f
commit daefafbee6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/utils/context/context_extends.cc
View File

@ -329,7 +329,7 @@ bool FinalizeGe(const std::shared_ptr<MsContext> &ms_context_ptr, bool force) {
if (ge::GEFinalize() != ge::GRAPH_SUCCESS) { if (ge::GEFinalize() != ge::GRAPH_SUCCESS) {
MS_LOG(WARNING) << "Finalize GE failed!"; MS_LOG(WARNING) << "Finalize GE failed!";
} }
ms_context_ptr->set_pynative_ge_init(true); ms_context_ptr->set_pynative_ge_init(fasle);
} else { } else {
MS_LOG(INFO) << "Ge is used, no need to finalize, tsd reference = " << ms_context_ptr->ge_ref() << "."; 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) { if (ms_context_ptr == nullptr) {
MS_LOG(EXCEPTION) << "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) { bool IsGeInited(const std::shared_ptr<MsContext> &ms_context_ptr) {
if (ms_context_ptr == nullptr) { if (ms_context_ptr == nullptr) {
MS_LOG(EXCEPTION) << "nullptr"; MS_LOG(EXCEPTION) << "nullptr";
} }
return ms_context_ptr->ge_ref(); return ms_context_ptr->IsGeInited();
} }
// Register for device type. // Register for device type.