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

Fixes compilation errors on MSVC x86 targets (#8823)

Browse Source
This commit is contained in:
Mauro Leggieri
2023-02-25 22:20:28 +03:00
committed by GitHub
parent a65ad0bd9c
commit 90c0633a28
4 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/objective/quantile_obj.cu
View File

@@ -80,8 +80,11 @@ class QuantileRegression : public ObjFunction {
linalg::ElementWiseKernel(
ctx_, gpair, [=] XGBOOST_DEVICE(std::size_t i, GradientPair const&) mutable {
auto idx = linalg::UnravelIndex(
i, {n_samples, static_cast<SizeT>(alpha.size()), n_targets / alpha.size()});
auto idx = linalg::UnravelIndex(static_cast<std::size_t>(i),
{static_cast<std::size_t>(n_samples),
static_cast<std::size_t>(alpha.size()),
static_cast<std::size_t>(n_targets / alpha.size())});
// std::tie is not available for cuda kernel.
std::size_t sample_id = std::get<0>(idx);
std::size_t quantile_id = std::get<1>(idx);