new change for mpi

2014-10-16 15:12:10 -07:00
parent a21df0770d
commit 0cf2dd39ea
8 changed files with 65 additions and 8 deletions
--- a/wrapper/xgboost_wrapper.cpp
+++ b/wrapper/xgboost_wrapper.cpp
@@ -44,7 +44,7 @@ class Booster: public learner::BoostLearner {
    for (bst_omp_uint j = 0; j < ndata; ++j) {
      gpair_[j] = bst_gpair(grad[j], hess[j]);
    }
-    gbm_->DoBoost(train.fmat(), train.info.info, &gpair_);
+    gbm_->DoBoost(train.fmat(), this->FindBufferOffset(train), train.info.info, &gpair_);
  }
  inline void CheckInitModel(void) {
    if (!init_model) {