From 2b36c0935572c7b2f91d3473b2b56f768471e4a2 Mon Sep 17 00:00:00 2001 From: Seunghwa Kang Date: Tue, 11 May 2021 09:45:57 -0400 Subject: [PATCH 1/2] bug fix --- cpp/src/experimental/coarsen_graph.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/src/experimental/coarsen_graph.cu b/cpp/src/experimental/coarsen_graph.cu index 6397f92e336..47ed6d88457 100644 --- a/cpp/src/experimental/coarsen_graph.cu +++ b/cpp/src/experimental/coarsen_graph.cu @@ -455,7 +455,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); } } From 2f2e475b26a69af0a553dbb7c8704f50f0d7eadb Mon Sep 17 00:00:00 2001 From: Seunghwa Kang Date: Tue, 11 May 2021 13:54:22 -0400 Subject: [PATCH 2/2] additional bug fix --- cpp/src/experimental/coarsen_graph.cu | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/cpp/src/experimental/coarsen_graph.cu b/cpp/src/experimental/coarsen_graph.cu index 47ed6d88457..f0864b8af63 100644 --- 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(),