maintain symmetry in weighted graph generation for testing

rapidsai · Dec 8, 2023 · bf33d09 · bf33d09
1 parent e291883
commit bf33d09
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/cpp/tests/utilities/test_graphs.hpp b/cpp/tests/utilities/test_graphs.hpp
@@ -633,12 +633,14 @@ construct_graph(raft::handle_t const& handle,
 
   if (drop_multi_edges) {
     std::tie(d_src_v, d_dst_v, d_weights_v, std::ignore, std::ignore) =
-      cugraph::remove_multi_edges<vertex_t, edge_t, weight_t, int32_t>(handle,
-                                                                       std::move(d_src_v),
-                                                                       std::move(d_dst_v),
-                                                                       std::move(d_weights_v),
-                                                                       std::nullopt,
-                                                                       std::nullopt);
+      cugraph::remove_multi_edges<vertex_t, edge_t, weight_t, int32_t>(
+        handle,
+        std::move(d_src_v),
+        std::move(d_dst_v),
+        std::move(d_weights_v),
+        std::nullopt,
+        std::nullopt,
+        is_symmetric ? true /* keep minimum weight edges to maintain symmetry */ : false);
   }
 
   graph_t<vertex_t, edge_t, store_transposed, multi_gpu> graph(handle);