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

Fix multi-GPU test failures (#4259)

Browse Source
This commit is contained in:
Rory Mitchell 2019-03-15 14:40:43 +13:00 committed by GitHub
parent 4352fcdb15
commit 5465b73e7c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/linear/updater_gpu_coordinate.cu
View File

@ -201,10 +201,11 @@ class GPUCoordinateUpdater : public LinearUpdater {
monitor_.Stop("LazyInitShards");
monitor_.Start("UpdateGpair");
auto &in_gpair_host = in_gpair->ConstHostVector();
// Update gpair
dh::ExecuteIndexShards(&shards_, [&](int idx, std::unique_ptr<DeviceShard>& shard) {
if (!shard->IsEmpty()) {
shard->UpdateGpair(in_gpair->ConstHostVector(), model->param);
shard->UpdateGpair(in_gpair_host, model->param);
}
});
monitor_.Stop("UpdateGpair");