From 9e4b4568860c592237108a3e42fc901b95b8cc50 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Wed, 9 Aug 2023 04:08:05 -0700 Subject: [PATCH] Fix read out of bounds in string concatenate --- cpp/src/strings/copying/concatenate.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cpp/src/strings/copying/concatenate.cu b/cpp/src/strings/copying/concatenate.cu index 287910c9a6f..fb9434c8899 100644 --- a/cpp/src/strings/copying/concatenate.cu +++ b/cpp/src/strings/copying/concatenate.cu @@ -121,7 +121,7 @@ __global__ void fused_concatenate_string_offset_kernel(column_device_view const* bitmask_type* output_mask, size_type* out_valid_count) { - size_type output_index = threadIdx.x + blockIdx.x * blockDim.x; + int64_t output_index = threadIdx.x + blockIdx.x * blockDim.x; size_type warp_valid_count = 0; unsigned active_mask; @@ -175,7 +175,7 @@ __global__ void fused_concatenate_string_chars_kernel(column_device_view const* size_type const output_size, char* output_data) { - size_type output_index = threadIdx.x + blockIdx.x * blockDim.x; + int64_t output_index = threadIdx.x + blockIdx.x * blockDim.x; while (output_index < output_size) { // Lookup input index by searching for output index in offsets