Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix compilation error on x86 (#6964)

Browse Source 
Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
This commit is contained in:
Livius 2021-05-14 07:31:49 +02:00 committed by GitHub
parent f94f479358
commit a4886c404a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/c_api/c_api.cc
View File

@ -734,8 +734,9 @@ XGB_DLL int XGBoosterPredictFromCSR(BoosterHandle handle, char const *indptr,
API_BEGIN(); API_BEGIN();
CHECK_HANDLE(); CHECK_HANDLE();
std::shared_ptr<xgboost::data::CSRArrayAdapter> x{ std::shared_ptr<xgboost::data::CSRArrayAdapter> x{
new xgboost::data::CSRArrayAdapter{ new xgboost::data::CSRArrayAdapter{StringView{indptr},
StringView{indptr}, StringView{indices}, StringView{data}, cols}}; StringView{indices}, StringView{data},
static_cast<size_t>(cols)}};
std::shared_ptr<DMatrix> p_m {nullptr}; std::shared_ptr<DMatrix> p_m {nullptr};
if (m) { if (m) {
p_m = *static_cast<std::shared_ptr<DMatrix> *>(m); p_m = *static_cast<std::shared_ptr<DMatrix> *>(m);

2
src/gbm/gbtree.cc
View File

@ -231,7 +231,7 @@ void GBTree::DoBoost(DMatrix* p_fmat,
: in_gpair->DeviceIdx(); : in_gpair->DeviceIdx();
auto out = MatrixView<float>( auto out = MatrixView<float>(
&predt->predictions, &predt->predictions,
{p_fmat->Info().num_row_, static_cast<size_t>(ngroup)}, device); {static_cast<size_t>(p_fmat->Info().num_row_), static_cast<size_t>(ngroup)}, device);
CHECK_NE(ngroup, 0); CHECK_NE(ngroup, 0);
if (ngroup == 1) { if (ngroup == 1) {
std::vector<std::unique_ptr<RegTree>> ret; std::vector<std::unique_ptr<RegTree>> ret;