Skip to content

Commit

Permalink
Improve MG PageRank scalability (#2038)
Browse files Browse the repository at this point in the history
Improve MG PageRank performance & scalability in multi-node many GPU systems

Authors:
  - Seunghwa Kang (https://github.com/seunghwak)

Approvers:
  - Chuck Hastings (https://github.com/ChuckHastings)
  - Andrei Schaffer (https://github.com/aschaffer)
  - Kumar Aatish (https://github.com/kaatish)

URL: #2038
  • Loading branch information
seunghwak authored Mar 14, 2022
1 parent 8dcb900 commit 5fe65f6
Show file tree
Hide file tree
Showing 6 changed files with 218 additions and 202 deletions.
1 change: 0 additions & 1 deletion cpp/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -230,7 +230,6 @@ add_library(cugraph SHARED
src/structure/create_graph_from_edgelist_mg.cu
src/structure/symmetrize_edgelist_sg.cu
src/structure/symmetrize_edgelist_mg.cu
src/utilities/host_barrier.cpp
src/visitors/graph_envelope.cpp
src/visitors/visitors_factory.cpp
src/visitors/bfs_visitor.cpp
Expand Down
6 changes: 3 additions & 3 deletions cpp/include/cugraph/prims/copy_to_adj_matrix_row_col.cuh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright (c) 2020-2021, NVIDIA CORPORATION.
* Copyright (c) 2020-2022, NVIDIA CORPORATION.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -170,7 +170,7 @@ void copy_to_matrix_major(raft::handle_t const& handle,
matrix_partition_device_view_t<vertex_t, edge_t, weight_t, GraphViewType::is_multi_gpu>(
graph_view.get_matrix_partition_view(i));

if (col_comm_rank == i) {
if (i == col_comm_rank) {
auto vertex_partition =
vertex_partition_device_view_t<vertex_t, GraphViewType::is_multi_gpu>(
graph_view.get_vertex_partition_view());
Expand Down Expand Up @@ -365,7 +365,7 @@ void copy_to_matrix_minor(raft::handle_t const& handle,
matrix_partition_device_view_t<vertex_t, edge_t, weight_t, GraphViewType::is_multi_gpu>(
graph_view.get_matrix_partition_view(size_t{0}));
for (int i = 0; i < row_comm_size; ++i) {
if (row_comm_rank == i) {
if (i == row_comm_rank) {
auto vertex_partition =
vertex_partition_device_view_t<vertex_t, GraphViewType::is_multi_gpu>(
graph_view.get_vertex_partition_view());
Expand Down
Loading

0 comments on commit 5fe65f6

Please sign in to comment.