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

Fix linear gpu input (#6255)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-10-19 12:02:36 +08:00
committed by GitHub
parent cdcdab98b8
commit 5037abeb86
3 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/gbm/gblinear.cc
View File

@@ -231,7 +231,8 @@ class GBLinear : public GradientBooster {
// start collecting the prediction
const int ngroup = model_.learner_model_param->num_output_group;
preds.resize(p_fmat->Info().num_row_ * ngroup);
for (const auto &batch : p_fmat->GetBatches<SparsePage>()) {
for (const auto &page : p_fmat->GetBatches<SparsePage>()) {
auto const& batch = page.GetView();
// output convention: nrow * k, where nrow is number of rows
// k is number of group
// parallel over local batch
@@ -241,7 +242,7 @@ class GBLinear : public GradientBooster {
}
#pragma omp parallel for schedule(static)
for (omp_ulong i = 0; i < nsize; ++i) {
const size_t ridx = batch.base_rowid + i;
const size_t ridx = page.base_rowid + i;
// loop over output groups
for (int gid = 0; gid < ngroup; ++gid) {
bst_float margin =