From 5850a2558a0f6d9fc17334391f8d6ffd6a10684f Mon Sep 17 00:00:00 2001
From: hlsc <hlsc@qq.com>
Date: Sun, 29 Jul 2018 08:21:30 +0800
Subject: [PATCH] fix DMatrix load_row_split bug (#3431)

---
 src/c_api/c_api.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc
index c4c293846..77822c294 100644
--- a/src/c_api/c_api.cc
+++ b/src/c_api/c_api.cc
@@ -213,11 +213,13 @@ int XGDMatrixCreateFromFile(const char *fname,
                             int silent,
                             DMatrixHandle *out) {
   API_BEGIN();
+  bool load_row_split = false;
   if (rabit::IsDistributed()) {
     LOG(CONSOLE) << "XGBoost distributed mode detected, "
                  << "will split data among workers";
+    load_row_split = true;
   }
-  *out = new std::shared_ptr<DMatrix>(DMatrix::Load(fname, silent != 0, true));
+  *out = new std::shared_ptr<DMatrix>(DMatrix::Load(fname, silent != 0, load_row_split));
   API_END();
 }