use 'label_list' as a map form 'comm_rank' to 'label_map'

rapidsai · Nov 5, 2024 · 117a4ec · 117a4ec
1 parent 77303d5
commit 117a4ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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<int32_t> 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;