From dd82b28e20bc63e35cd978b770b0cf78c27f738f Mon Sep 17 00:00:00 2001
From: Rory Mitchell <r.a.mitchell.nz@gmail.com>
Date: Sat, 17 Feb 2018 12:11:48 +1300
Subject: [PATCH] Update GPU code with dmatrix changes (#3117)

---
 src/tree/updater_gpu.cu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/tree/updater_gpu.cu b/src/tree/updater_gpu.cu
index 50a466539..cee7c3e88 100644
--- a/src/tree/updater_gpu.cu
+++ b/src/tree/updater_gpu.cu
@@ -651,9 +651,9 @@ class GPUMaker : public TreeUpdater {
     fId->reserve(nCols * nRows);
     // in case you end up with a DMatrix having no column access
     // then make sure to enable that before copying the data!
-    if (!dmat->HaveColAccess()) {
+    if (!dmat->HaveColAccess(true)) {
       const std::vector<bool> enable(nCols, true);
-      dmat->InitColAccess(enable, 1, nRows);
+      dmat->InitColAccess(enable, 1, nRows, true);
     }
     dmlc::DataIter<ColBatch>* iter = dmat->ColIterator();
     iter->BeforeFirst();