Fix multi-GPU test failures (#4259)

dmlc · Mar 15, 2019 · 5465b73 · 5465b73
1 parent 4352fcd
commit 5465b73
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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");