[RABIT] fix rabit in local mode

2016-01-12 21:34:26 -08:00
parent 05b958c178
commit 112d866dc9
2 changed files with 14 additions and 12 deletions
--- a/src/allreduce_robust.cc
+++ b/src/allreduce_robust.cc
@@ -79,7 +79,7 @@ void AllreduceRobust::Allreduce(void *sendrecvbuf_,
                                PreprocFunction prepare_fun,
                                void *prepare_arg) {
  // skip action in single node
-  if (world_size == 1) {
+  if (world_size == 1 || world_size == -1) {
    if (prepare_fun != NULL) prepare_fun(prepare_arg);
    return;
  }