From a4886c404a81c0aeb72834f68fa3d0fd50eea942 Mon Sep 17 00:00:00 2001
From: Livius <egyszeregy@freemail.hu>
Date: Fri, 14 May 2021 07:31:49 +0200
Subject: [PATCH] Fix compilation error on x86 (#6964)

Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
---
 src/c_api/c_api.cc | 5 +++--
 src/gbm/gbtree.cc  | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc
index f9a9bc15e..f7586c192 100644
--- a/src/c_api/c_api.cc
+++ b/src/c_api/c_api.cc
@@ -734,8 +734,9 @@ XGB_DLL int XGBoosterPredictFromCSR(BoosterHandle handle, char const *indptr,
   API_BEGIN();
   CHECK_HANDLE();
   std::shared_ptr<xgboost::data::CSRArrayAdapter> x{
-      new xgboost::data::CSRArrayAdapter{
-          StringView{indptr}, StringView{indices}, StringView{data}, cols}};
+      new xgboost::data::CSRArrayAdapter{StringView{indptr},
+                                         StringView{indices}, StringView{data},
+                                         static_cast<size_t>(cols)}};
   std::shared_ptr<DMatrix> p_m {nullptr};
   if (m) {
     p_m = *static_cast<std::shared_ptr<DMatrix> *>(m);
diff --git a/src/gbm/gbtree.cc b/src/gbm/gbtree.cc
index 50d5b963b..5c8299c83 100644
--- a/src/gbm/gbtree.cc
+++ b/src/gbm/gbtree.cc
@@ -231,7 +231,7 @@ void GBTree::DoBoost(DMatrix* p_fmat,
                     : in_gpair->DeviceIdx();
   auto out = MatrixView<float>(
       &predt->predictions,
-      {p_fmat->Info().num_row_, static_cast<size_t>(ngroup)}, device);
+      {static_cast<size_t>(p_fmat->Info().num_row_), static_cast<size_t>(ngroup)}, device);
   CHECK_NE(ngroup, 0);
   if (ngroup == 1) {
     std::vector<std::unique_ptr<RegTree>> ret;