From 09443604168a3c58c7ff43e92daa163f7a51f686 Mon Sep 17 00:00:00 2001
From: Rong Ou <rong.ou@gmail.com>
Date: Thu, 7 Mar 2019 13:53:09 -0800
Subject: [PATCH] minor fix: log InitDataOnce() only when it is actually called
 (#4206)

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

diff --git a/src/tree/updater_gpu_hist.cu b/src/tree/updater_gpu_hist.cu
index 408e89e61..f178b1f2b 100644
--- a/src/tree/updater_gpu_hist.cu
+++ b/src/tree/updater_gpu_hist.cu
@@ -1029,11 +1029,11 @@ class GPUHistMakerSpecialised{
   }
 
   void InitData(HostDeviceVector<GradientPair>* gpair, DMatrix* dmat) {
-    monitor_.Start("InitDataOnce", dist_.Devices());
     if (!initialised_) {
+      monitor_.Start("InitDataOnce", dist_.Devices());
       this->InitDataOnce(dmat);
+      monitor_.Stop("InitDataOnce", dist_.Devices());
     }
-    monitor_.Stop("InitDataOnce", dist_.Devices());
 
     column_sampler_.Init(info_->num_col_, param_.colsample_bynode,
                          param_.colsample_bylevel, param_.colsample_bytree);