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::Load(fname, silent != 0, true)); + *out = new std::shared_ptr(DMatrix::Load(fname, silent != 0, load_row_split)); API_END(); }