Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

shutdown for multiple times (#99)

Browse Source
This commit is contained in:
Nan Zhu 2019-07-16 12:41:39 -07:00 committed by GitHub
parent 65b718a5e7
commit dba32d54d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/engine.cc
View File

@ -57,14 +57,16 @@ bool Init(int argc, char *argv[]) {
/*! \brief finalize syncrhonization module */
bool Finalize() {
ThreadLocalEntry* e = EngineThreadLocal::Get();
utils::Check(e->engine.get() != nullptr,
"rabit::Finalize engine is not initialized or already been finalized.");
if (e->engine->Shutdown()) {
e->engine.reset(nullptr);
e->initialized = false;
return true;
if (e->engine.get() != nullptr) {
if (e->engine->Shutdown()) {
e->engine.reset(nullptr);
e->initialized = false;
return true;
} else {
return false;
}
} else {
return false;
return true;
}
}