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

add distributed checking

Browse Source
This commit is contained in:
tqchen 2016-04-11 15:43:01 -07:00
parent be50e7b632
commit 849b20b7c8
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/rabit/c_api.h
View File

@ -46,6 +46,9 @@ RABIT_DLL int RabitGetRank();
/*! \brief get total number of process */ /*! \brief get total number of process */
RABIT_DLL int RabitGetWorldSize(); RABIT_DLL int RabitGetWorldSize();
/*! \brief get rank of current process */
RABIT_DLL int RabitIsDistributed();
/*! /*!
* \brief print the msg to the tracker, * \brief print the msg to the tracker,
* this function can be used to communicate the information of the progress to * this function can be used to communicate the information of the progress to

4
src/c_api.cc
View File

@ -178,6 +178,10 @@ int RabitGetWorldSize() {
return rabit::GetWorldSize(); return rabit::GetWorldSize();
} }
int RabitIsDistributed() {
return rabit::IsDistributed();
}
void RabitTrackerPrint(const char *msg) { void RabitTrackerPrint(const char *msg) {
std::string m(msg); std::string m(msg);
rabit::TrackerPrint(m); rabit::TrackerPrint(m);