Bug fix in MG coarsen_grph (#1596)

- [x] Bug fix in MG coarsen_graph Authors: - Seunghwa Kang (https://github.com/seunghwak) Approvers: - Chuck Hastings (https://github.com/ChuckHastings) URL: #1596
rapidsai · May 12, 2021 · d97638d · d97638d
1 parent 85ceac3
commit d97638d
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/cpp/src/experimental/coarsen_graph.cu b/cpp/src/experimental/coarsen_graph.cu
@@ -284,12 +284,14 @@ coarsen_graph(
       store_transposed ? graph_view.get_number_of_local_adj_matrix_partition_cols(i)
                        : graph_view.get_number_of_local_adj_matrix_partition_rows(i),
       handle.get_stream());
-    // FIXME: this copy is unnecessary, beter fix RAFT comm's bcast to take const iterators for
-    // input
-    thrust::copy(rmm::exec_policy(handle.get_stream())->on(handle.get_stream()),
-                 labels,
-                 labels + major_labels.size(),
-                 major_labels.begin());
+    if (col_comm_rank == i) {
+      // FIXME: this copy is unnecessary, beter fix RAFT comm's bcast to take const iterators for
+      // input
+      thrust::copy(rmm::exec_policy(handle.get_stream())->on(handle.get_stream()),
+                   labels,
+                   labels + major_labels.size(),
+                   major_labels.begin());
+    }
     device_bcast(col_comm,
                  major_labels.data(),
                  major_labels.data(),
@@ -455,7 +457,7 @@ coarsen_graph(
                               cur_size;
         thrust::copy(rmm::exec_policy(handle.get_stream())->on(handle.get_stream()),
                      src_edge_first,
-                     src_edge_first + edgelist_major_vertices.size(),
+                     src_edge_first + number_of_partition_edges,
                      dst_edge_first);
       }
     }