(gluten) branch main updated: [VL] Fix gpu shuffle writer test (#12357)

[marong]

This is an automated email from the ASF dual-hosted git repository.

marin-ma pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/gluten.git


The following commit(s) were added to refs/heads/main by this push:
     new fa83280d27 [VL] Fix gpu shuffle writer test (#12357)
fa83280d27 is described below

commit fa83280d27121a74ad1dd62796847049ecb7142d
Author: Rong Ma <mar...@apache.org>
AuthorDate: Fri Jun 26 13:17:23 2026 +0100

    [VL] Fix gpu shuffle writer test (#12357)
---
 cpp/velox/utils/GpuBufferBatchResizer.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/cpp/velox/utils/GpuBufferBatchResizer.cc 
b/cpp/velox/utils/GpuBufferBatchResizer.cc
index 9a6adc3960..e896442dde 100644
--- a/cpp/velox/utils/GpuBufferBatchResizer.cc
+++ b/cpp/velox/utils/GpuBufferBatchResizer.cc
@@ -68,7 +68,7 @@ struct DispatchColumn {
     auto values = buffers[bufferIdx++];
 
     // === Step 2: allocate GPU device buffers and copy ===
-    rmm::device_buffer dataBuf(values->size(), stream);
+    rmm::device_buffer dataBuf(values->size(), stream, mr);
     CUDF_CUDA_TRY(
         cudaMemcpyAsync(dataBuf.data(), values->data(), values->size(), 
cudaMemcpyHostToDevice, stream.value()));
 
@@ -78,7 +78,7 @@ struct DispatchColumn {
     cudf::data_type cudfType{typeId};
     size_t nullCount = nulls == nullptr || nulls->size() == 0
         ? 0
-        : cudf::null_count(reinterpret_cast<const 
cudf::bitmask_type*>(nulls->data()), 0, numRows, stream);
+        : cudf::null_count(static_cast<const 
cudf::bitmask_type*>(nullBuf->data()), 0, numRows, stream);
     return std::make_unique<cudf::column>(cudfType, numRows, 
std::move(dataBuf), std::move(*nullBuf), nullCount);
   }
 
@@ -118,7 +118,7 @@ struct DispatchColumn {
     // === Step 3: create cudf::column ===
     size_t nullCount = nulls == nullptr || nulls->size() == 0
         ? 0
-        : cudf::null_count(reinterpret_cast<const 
cudf::bitmask_type*>(nulls->data()), 0, numRows, stream);
+        : cudf::null_count(static_cast<const 
cudf::bitmask_type*>(mask->data()), 0, numRows, stream);
 
     auto offsetColumn = getOffsetsColumn(offsets);
 


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@gluten.apache.org
For additional commands, e-mail: commits-h...@gluten.apache.org