bug fix in MG prim tests

rapidsai · Nov 19, 2024 · a622d97 · a622d97
1 parent 27a5ffe
commit a622d97
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/cpp/tests/CMakeLists.txt b/cpp/tests/CMakeLists.txt
@@ -698,8 +698,8 @@ if(BUILD_CUGRAPH_MG_TESTS)
     ConfigureTestMG(MG_COUNT_IF_V_TEST prims/mg_count_if_v.cu)
 
     ###############################################################################################
-    # - MG PRIMS TRANSFORM_REDUCE_V_FRONTIER_OUTGOING_E_BY_SRC_DST tests --------------------------
-    ConfigureTestMG(MG_TRANSFORM_REDUCE_V_FRONTIER_OUTGOING_E_BY_SRC_DST_TEST
+    # - MG PRIMS TRANSFORM_REDUCE_V_FRONTIER_OUTGOING_E_BY_DST tests ------------------------------
+    ConfigureTestMG(MG_TRANSFORM_REDUCE_V_FRONTIER_OUTGOING_E_BY_DST_TEST
                     prims/mg_transform_reduce_v_frontier_outgoing_e_by_dst.cu)
 
     ###############################################################################################

diff --git a/cpp/tests/prims/mg_transform_reduce_v_frontier_outgoing_e_by_dst.cu b/cpp/tests/prims/mg_transform_reduce_v_frontier_outgoing_e_by_dst.cu
@@ -248,11 +248,24 @@ class Tests_MGTransformReduceVFrontierOutgoingEBySrcDst
       auto mg_reduce_by_dst_aggregate_new_frontier_key_buffer =
         cugraph::allocate_dataframe_buffer<key_t>(0, handle_->get_stream());
       if constexpr (std::is_same_v<key_t, vertex_t>) {
+        cugraph::unrenumber_local_int_vertices(*handle_,
+                                               mg_reduce_by_dst_new_frontier_key_buffer.data(),
+                                               mg_reduce_by_dst_new_frontier_key_buffer.size(),
+                                               (*mg_renumber_map).data(),
+                                               mg_graph_view.local_vertex_partition_range_first(),
+                                               mg_graph_view.local_vertex_partition_range_last());
         mg_reduce_by_dst_aggregate_new_frontier_key_buffer =
           cugraph::test::device_gatherv(*handle_,
                                         mg_reduce_by_dst_new_frontier_key_buffer.data(),
                                         mg_reduce_by_dst_new_frontier_key_buffer.size());
       } else {
+        cugraph::unrenumber_local_int_vertices(
+          *handle_,
+          std::get<0>(mg_reduce_by_dst_new_frontier_key_buffer).data(),
+          std::get<0>(mg_reduce_by_dst_new_frontier_key_buffer).size(),
+          (*mg_renumber_map).data(),
+          mg_graph_view.local_vertex_partition_range_first(),
+          mg_graph_view.local_vertex_partition_range_last());
         std::get<0>(mg_reduce_by_dst_aggregate_new_frontier_key_buffer) =
           cugraph::test::device_gatherv(
             *handle_,