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

[RABIT] fix rabit in local mode

Browse Source
This commit is contained in:
tqchen 2016-01-12 21:34:26 -08:00
parent 05b958c178
commit 112d866dc9
2 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/rabit/engine.h
View File

@ -41,6 +41,8 @@ class IEngine {
typedef void (ReduceFunction) (const void *src, typedef void (ReduceFunction) (const void *src,
void *dst, int count, void *dst, int count,
const MPI::Datatype &dtype); const MPI::Datatype &dtype);
/*! \brief virtual destructor */
virtual ~IEngine() {}
/*! /*!
* \brief performs in-place Allreduce, on sendrecvbuf * \brief performs in-place Allreduce, on sendrecvbuf
* this function is NOT thread-safe * this function is NOT thread-safe

2
src/allreduce_robust.cc
View File

@ -79,7 +79,7 @@ void AllreduceRobust::Allreduce(void *sendrecvbuf_,
PreprocFunction prepare_fun, PreprocFunction prepare_fun,
void *prepare_arg) { void *prepare_arg) {
// skip action in single node // skip action in single node
if (world_size == 1) { if (world_size == 1 || world_size == -1) {
if (prepare_fun != NULL) prepare_fun(prepare_arg); if (prepare_fun != NULL) prepare_fun(prepare_arg);
return; return;
} }