fix bug in GetDataByColumnPointers

microsoft · Oct 25, 2024 · 950199d · 950199d
2 parents b43f88b + 483e521
commit 950199d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/LightGBM/cuda/cuda_column_data.hpp b/include/LightGBM/cuda/cuda_column_data.hpp
@@ -101,7 +101,7 @@ class CUDAColumnData {
   std::vector<void*> GetDataByColumnPointers(const std::vector<std::unique_ptr<CUDAVector<void>>>& data_by_column) const {
     std::vector<void*> data_by_column_pointers(data_by_column.size(), nullptr);
     for (size_t i = 0; i < data_by_column.size(); ++i) {
-      data_by_column_pointers[i] = data_by_column[i].get();
+      data_by_column_pointers[i] = reinterpret_cast<void*>(data_by_column[i]->RawData());
     }
     return data_by_column_pointers;
   }