Fix multi-GPU test failures (#4259)

2019-03-15 14:40:43 +13:00
parent 4352fcdb15
commit 5465b73e7c
1 changed files with 2 additions and 1 deletions
--- a/src/linear/updater_gpu_coordinate.cu
+++ b/src/linear/updater_gpu_coordinate.cu
@@ -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");