Fix compilation error due to 64-bit integer narrowing to size_t (#5250)

This commit is contained in:
Philip Hyunsu Cho
2020-01-30 21:32:15 -08:00
committed by GitHub
parent 472ded549d
commit adc795929a
2 changed files with 5 additions and 4 deletions

View File

@@ -311,7 +311,7 @@ XGB_DLL int XGDMatrixSliceDMatrixEx(DMatrixHandle handle,
DMatrix* dmat = static_cast<std::shared_ptr<DMatrix>*>(handle)->get();
CHECK(dynamic_cast<data::SimpleDMatrix*>(dmat))
<< "Slice only supported for SimpleDMatrix currently.";
data::DMatrixSliceAdapter adapter(dmat, {idxset, len});
data::DMatrixSliceAdapter adapter(dmat, {idxset, static_cast<size_t>(len)});
*out = new std::shared_ptr<DMatrix>(
DMatrix::Create(&adapter, std::numeric_limits<float>::quiet_NaN(), 1));
API_END();