Properly resize edge_weights vector when using CSR representation. (#48)

* Properly resize edge_weights vector when using CSR representation. * Also fix it for vertex weights.
KarlsruheGraphGeneration · Sep 12, 2024 · 6cdeb60 · 6cdeb60
1 parent 7f8a9f8
commit 6cdeb60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kagen/generators/generator.cpp b/kagen/generators/generator.cpp
@@ -79,7 +79,7 @@ void Generator::GenerateEdgeWeights(EdgeWeightConfig weight_config, MPI_Comm com
             edge_weight_generator->GenerateEdgeWeights(graph_.edges, graph_.edge_weights);
             break;
         case GraphRepresentation::CSR:
-            if (!graph_.xadj.empty()) {
+            if (!graph_.edges.empty()) {
                 edge_weight_generator->GenerateEdgeWeights(graph_.edges, graph_.edge_weights);
             } else {
                 // for generated graph edgelist format is used for construction and then transformed to CSR only in the
@@ -113,7 +113,7 @@ void Generator::GenerateVertexWeights(VertexWeightConfig weight_config, MPI_Comm
             vertex_weight_generator->GenerateVertexWeights(graph_.vertex_range, graph_.edges, graph_.vertex_weights);
             break;
         case GraphRepresentation::CSR:
-            if (!graph_.xadj.empty()) {
+            if (!graph_.edges.empty()) {
                 vertex_weight_generator->GenerateVertexWeights(
                     graph_.vertex_range, graph_.edges, graph_.vertex_weights);
             } else {