diff --git a/cpp/src/sampling/detail/conversion_utilities_impl.cuh b/cpp/src/sampling/detail/conversion_utilities_impl.cuh index 6eff015458..1061b28c44 100644 --- a/cpp/src/sampling/detail/conversion_utilities_impl.cuh +++ b/cpp/src/sampling/detail/conversion_utilities_impl.cuh @@ -46,9 +46,9 @@ rmm::device_uvector flatten_label_map( thrust::fill(handle.get_thrust_policy(), label_map.begin(), label_map.end(), int32_t{0}); thrust::scatter(handle.get_thrust_policy(), - std::get<0>(label_to_output_comm_rank).begin(), - std::get<0>(label_to_output_comm_rank).end(), std::get<1>(label_to_output_comm_rank).begin(), + std::get<1>(label_to_output_comm_rank).end(), + std::get<0>(label_to_output_comm_rank).begin(), label_map.begin()); return label_map;