fix illegal memory access

rapidsai · Nov 22, 2024 · d997656 · d997656
1 parent fc0426d
commit d997656
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/sampling/neighbor_sampling_impl.hpp b/cpp/src/sampling/neighbor_sampling_impl.hpp
@@ -408,7 +408,7 @@ neighbor_sample_impl(raft::handle_t const& handle,
 
     thrust::transform(handle.get_thrust_policy(),
                       thrust::make_counting_iterator<edge_t>(0),
-                      thrust::make_counting_iterator<edge_t>(result_offsets->size()),
+                      thrust::make_counting_iterator<edge_t>(result_offsets->size() - 1),
                       result_offsets->begin() + 1,
                       [result_labels = raft::device_span<label_t const>(
                          cp_result_labels->data(), cp_result_labels->size())] __device__(auto idx) {