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

change allreduce lib to rabit library, xgboost now run with rabit

Browse Source
This commit is contained in:
tqchen
2014-12-20 00:17:09 -08:00
parent 5ae99372d6
commit 8e16cc4617
28 changed files with 105 additions and 1206 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
wrapper/xgboost_wrapper.cpp
View File

@@ -83,21 +83,21 @@ using namespace xgboost::wrapper;
extern "C"{
void XGSyncInit(int argc, char *argv[]) {
sync::Init(argc, argv);
if (sync::IsDistributed()) {
std::string pname = xgboost::sync::GetProcessorName();
utils::Printf("distributed job start %s:%d\n", pname.c_str(), xgboost::sync::GetRank());
rabit::Init(argc, argv);
if (rabit::GetWorldSize() != 1) {
std::string pname = rabit::GetProcessorName();
utils::Printf("distributed job start %s:%d\n", pname.c_str(), rabit::GetRank());
}
}
void XGSyncFinalize(void) {
sync::Finalize();
rabit::Finalize();
}
int XGSyncGetRank(void) {
int rank = xgboost::sync::GetRank();
int rank = rabit::GetRank();
return rank;
}
int XGSyncGetWorldSize(void) {
return sync::GetWorldSize();
return rabit::GetWorldSize();
}
void* XGDMatrixCreateFromFile(const char *fname, int silent) {
return LoadDataMatrix(fname, silent != 0, false);