From cdc036b752df52c32d43329a9b9c92ae05debbec Mon Sep 17 00:00:00 2001
From: Ray Kim <msca8h@naver.com>
Date: Thu, 15 Mar 2018 05:40:24 +0900
Subject: [PATCH] Fixed performance bug (#3171)

Minor performance improvements to gpu predictor
---
 src/predictor/gpu_predictor.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/predictor/gpu_predictor.cu b/src/predictor/gpu_predictor.cu
index 255888324..9d04610b8 100644
--- a/src/predictor/gpu_predictor.cu
+++ b/src/predictor/gpu_predictor.cu
@@ -277,7 +277,7 @@ class GPUPredictor : public xgboost::Predictor {
     CHECK_EQ(model.param.size_leaf_vector, 0);
     // Copy decision trees to device
     thrust::host_vector<size_t> h_tree_segments;
-    h_tree_segments.reserve((tree_end - tree_end) + 1);
+    h_tree_segments.reserve((tree_end - tree_begin) + 1);
     size_t sum = 0;
     h_tree_segments.push_back(sum);
     for (auto tree_idx = tree_begin; tree_idx < tree_end; tree_idx++) {