Fixed malloc bug in lap_collect_traces

spcl · Nov 20, 2023 · d875e4f · d875e4f
1 parent c04fa8c
commit d875e4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/liballprof2/tracer_main.c b/src/liballprof2/tracer_main.c
@@ -71,7 +71,7 @@ static void lap_collect_traces(void) {
     PMPI_Comm_size(MPI_COMM_WORLD, &comm_size);
     int trace_size = ftell(lap_fptr);
     fseek(lap_fptr, 0, SEEK_SET);
-    int* trace_sizes = malloc(comm_size);
+    int* trace_sizes = malloc(comm_size * sizeof(int));
     if (trace_sizes == NULL) {
       fprintf(stderr, "lap2 ran out of memory when collecting traces :(\n");
       return;
@@ -130,4 +130,4 @@ int main(int argc, char** argv) {
   }
   lap_collect_traces();
   MPI_Finalize();
-}
+}