From cfea4dbe85fbceb7b0397d45573ef075aeaabc6e Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Sun, 21 Dec 2014 04:35:32 -0800
Subject: [PATCH] fix rabit for single node without initialization

---
 src/allreduce_base.cc | 4 +++-
 src/allreduce_base.h  | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/allreduce_base.cc b/src/allreduce_base.cc
index b8b3ed0de..f2a104a90 100644
--- a/src/allreduce_base.cc
+++ b/src/allreduce_base.cc
@@ -21,7 +21,7 @@ AllreduceBase::AllreduceBase(void) {
   host_uri = "";
   slave_port = 9010;
   nport_trial = 1000;
-  rank = -1;
+  rank = 0;
   world_size = -1;
   hadoop_mode = 0;
   version_number = 0;
@@ -50,6 +50,8 @@ void AllreduceBase::Init(void) {
       this->SetParam("rabit_world_size", num_task);
     }
   }
+  // clear the setting before start reconnection
+  this->rank = -1;
   //---------------------
   // start socket
   utils::Socket::Startup();
diff --git a/src/allreduce_base.h b/src/allreduce_base.h
index f26b5ec45..bede2c228 100644
--- a/src/allreduce_base.h
+++ b/src/allreduce_base.h
@@ -58,6 +58,7 @@ class AllreduceBase : public IEngine {
   }
   /*! \brief get rank */
   virtual int GetWorldSize(void) const {
+    if (world_size == -1) return 1;
     return world_size;
   }
   /*! \brief get rank */