From d62eaf46a71f272088f400ea64b17c515c1eb98d Mon Sep 17 00:00:00 2001 From: "flametbs@gmail.com" Date: Fri, 10 Feb 2023 01:21:36 +0800 Subject: [PATCH] cast data_index as size_t --- src/io/cuda/cuda_row_data.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/io/cuda/cuda_row_data.cpp b/src/io/cuda/cuda_row_data.cpp index 30bbb273193a..e614a986dbdd 100644 --- a/src/io/cuda/cuda_row_data.cpp +++ b/src/io/cuda/cuda_row_data.cpp @@ -323,7 +323,7 @@ void CUDARowData::GetDenseDataPartitioned(const BIN_TYPE* row_wise_data, std::ve const int partition_column_end = feature_partition_column_index_offsets_[i + 1]; const int num_columns_in_cur_partition = partition_column_end - partition_column_start; for (data_size_t data_index = start; data_index < end; ++data_index) { - const size_t data_offset = offset + data_index * num_columns_in_cur_partition; + const size_t data_offset = offset + static_cast(data_index) * num_columns_in_cur_partition; const size_t read_data_offset = static_cast(data_index) * num_total_columns; for (int column_index = 0; column_index < num_columns_in_cur_partition; ++column_index) { const size_t true_column_index = read_data_offset + column_index + partition_column_start;