From 2ccdb39351d09547c8449e6741430c73327bb034 Mon Sep 17 00:00:00 2001
From: "junpeng.li" <junpeng.li@accenture.com>
Date: Tue, 18 Oct 2022 11:04:52 +0800
Subject: [PATCH] modify R for #5540

---
 R-package/src/lightgbm_R.cpp | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/R-package/src/lightgbm_R.cpp b/R-package/src/lightgbm_R.cpp
index 82956daef4b9..a9cb70a15711 100644
--- a/R-package/src/lightgbm_R.cpp
+++ b/R-package/src/lightgbm_R.cpp
@@ -198,7 +198,7 @@ SEXP LGBM_DatasetCreateFromMat_R(SEXP data,
   SEXP reference) {
   R_API_BEGIN();
   SEXP ret = PROTECT(R_MakeExternalPtr(nullptr, R_NilValue, R_NilValue));
-  int32_t nrow = static_cast<int32_t>(Rf_asInteger(num_row));
+  int64_t nrow = static_cast<int64_t>(Rf_asInteger(num_row));
   int32_t ncol = static_cast<int32_t>(Rf_asInteger(num_col));
   double* p_mat = REAL(data);
   const char* parameters_ptr = CHAR(PROTECT(Rf_asChar(parameters)));
@@ -223,7 +223,7 @@ SEXP LGBM_DatasetGetSubset_R(SEXP handle,
   R_API_BEGIN();
   _AssertDatasetHandleNotNull(handle);
   SEXP ret = PROTECT(R_MakeExternalPtr(nullptr, R_NilValue, R_NilValue));
-  int32_t len = static_cast<int32_t>(Rf_asInteger(len_used_row_indices));
+  int64_t len = static_cast<int64_t>(Rf_asInteger(len_used_row_indices));
   std::vector<int32_t> idxvec(len);
   // convert from one-based to zero-based index
 #pragma omp parallel for schedule(static, 512) if (len >= 1024)
@@ -335,7 +335,7 @@ SEXP LGBM_DatasetSetField_R(SEXP handle,
   SEXP num_element) {
   R_API_BEGIN();
   _AssertDatasetHandleNotNull(handle);
-  int len = Rf_asInteger(num_element);
+  int64_t len = Rf_asInteger(num_element);
   const char* name = CHAR(PROTECT(Rf_asChar(field_name)));
   if (!strcmp("group", name) || !strcmp("query", name)) {
     std::vector<int32_t> vec(len);
@@ -349,7 +349,7 @@ SEXP LGBM_DatasetSetField_R(SEXP handle,
   } else {
     std::vector<float> vec(len);
 #pragma omp parallel for schedule(static, 512) if (len >= 1024)
-    for (int i = 0; i < len; ++i) {
+    for (int64_t i = 0; i < len; ++i) {
       vec[i] = static_cast<float>(REAL(field_data)[i]);
     }
     CHECK_CALL(LGBM_DatasetSetField(R_ExternalPtrAddr(handle), name, vec.data(), len, C_API_DTYPE_FLOAT32));
@@ -365,7 +365,7 @@ SEXP LGBM_DatasetGetField_R(SEXP handle,
   R_API_BEGIN();
   _AssertDatasetHandleNotNull(handle);
   const char* name = CHAR(PROTECT(Rf_asChar(field_name)));
-  int out_len = 0;
+  int64_t out_len = 0;
   int out_type = 0;
   const void* res;
   CHECK_CALL(LGBM_DatasetGetField(R_ExternalPtrAddr(handle), name, &out_len, &res, &out_type));
@@ -373,19 +373,19 @@ SEXP LGBM_DatasetGetField_R(SEXP handle,
     auto p_data = reinterpret_cast<const int32_t*>(res);
     // convert from boundaries to size
 #pragma omp parallel for schedule(static, 512) if (out_len >= 1024)
-    for (int i = 0; i < out_len - 1; ++i) {
+    for (int64_t i = 0; i < out_len - 1; ++i) {
       INTEGER(field_data)[i] = p_data[i + 1] - p_data[i];
     }
   } else if (!strcmp("init_score", name)) {
     auto p_data = reinterpret_cast<const double*>(res);
 #pragma omp parallel for schedule(static, 512) if (out_len >= 1024)
-    for (int i = 0; i < out_len; ++i) {
+    for (int64_t i = 0; i < out_len; ++i) {
       REAL(field_data)[i] = p_data[i];
     }
   } else {
     auto p_data = reinterpret_cast<const float*>(res);
 #pragma omp parallel for schedule(static, 512) if (out_len >= 1024)
-    for (int i = 0; i < out_len; ++i) {
+    for (int64_t i = 0; i < out_len; ++i) {
       REAL(field_data)[i] = p_data[i];
     }
   }
@@ -427,7 +427,7 @@ SEXP LGBM_DatasetUpdateParamChecking_R(SEXP old_params,
 SEXP LGBM_DatasetGetNumData_R(SEXP handle, SEXP out) {
   R_API_BEGIN();
   _AssertDatasetHandleNotNull(handle);
-  int nrow;
+  int64_t nrow;
   CHECK_CALL(LGBM_DatasetGetNumData(R_ExternalPtrAddr(handle), &nrow));
   INTEGER(out)[0] = nrow;
   return R_NilValue;
@@ -957,7 +957,7 @@ SEXP LGBM_BoosterPredictForMat_R(SEXP handle,
   R_API_BEGIN();
   _AssertBoosterHandleNotNull(handle);
   int pred_type = GetPredictType(is_rawscore, is_leafidx, is_predcontrib);
-  int32_t nrow = static_cast<int32_t>(Rf_asInteger(num_row));
+  int64_t nrow = static_cast<int64_t>(Rf_asInteger(num_row));
   int32_t ncol = static_cast<int32_t>(Rf_asInteger(num_col));
   const double* p_mat = REAL(data);
   double* ptr_ret = REAL(out_result);