From 3da0f680be0396ae07bc349c4dc82b24cd0f31a5 Mon Sep 17 00:00:00 2001
From: Rory Mitchell <r.a.mitchell.nz@gmail.com>
Date: Fri, 15 Mar 2019 13:18:11 +1300
Subject: [PATCH] Fix multi-GPU test failures

---
 src/linear/updater_gpu_coordinate.cu | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/linear/updater_gpu_coordinate.cu b/src/linear/updater_gpu_coordinate.cu
index 663276675d53..fa9146e18019 100644
--- 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");