add global num vertices/edges query function (#52)

KarlsruheGraphGeneration · Oct 31, 2024 · 4e51662 · 4e51662
1 parent d61c597
commit 4e51662
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/kagen/kagen.cpp b/kagen/kagen.cpp
@@ -361,10 +361,17 @@ std::ostream& operator<<(std::ostream& out, const GraphRepresentation representa
 
     return out << "<invalid>";
 }
+
 SInt Graph::NumberOfLocalVertices() const {
     return vertex_range.second - vertex_range.first;
 }
 
+SInt Graph::NumberOfGlobalVertices() const {
+    SInt number_vertices = NumberOfLocalVertices();
+    MPI_Allreduce(MPI_IN_PLACE, &number_vertices, 1, MPI_UNSIGNED_LONG_LONG, MPI_SUM, MPI_COMM_WORLD);
+    return number_vertices;
+}
+
 SInt Graph::NumberOfLocalEdges() const {
     switch (representation) {
         case GraphRepresentation::EDGE_LIST:
@@ -377,6 +384,12 @@ SInt Graph::NumberOfLocalEdges() const {
     __builtin_unreachable();
 }
 
+SInt Graph::NumberOfGlobalEdges() const {
+    SInt number_edges = NumberOfLocalEdges();
+    MPI_Allreduce(MPI_IN_PLACE, &number_edges, 1, MPI_UNSIGNED_LONG_LONG, MPI_SUM, MPI_COMM_WORLD);
+    return number_edges;
+}
+
 void Graph::Clear() {
     edges.clear();
     xadj.clear();

diff --git a/kagen/kagen.h b/kagen/kagen.h
@@ -208,8 +208,12 @@ struct Graph {
 
     SInt NumberOfLocalVertices() const;
 
+    SInt NumberOfGlobalVertices() const;
+
     SInt NumberOfLocalEdges() const;
 
+    SInt NumberOfGlobalEdges() const;
+
     void SortEdgelist();
 
     void FreeEdgelist();