From c6afef0b6f76ffc55d4172d971be6cf5a08a73a4 Mon Sep 17 00:00:00 2001
From: David Addison <daddison@nvidia.com>
Date: Mon, 5 Feb 2024 08:53:54 -0800
Subject: [PATCH] Added missing MPI_Comm_free() call before MPI_Finalize()

---
 src/common.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/common.cu b/src/common.cu
index 8588047..4ac00fb 100644
--- a/src/common.cu
+++ b/src/common.cu
@@ -1056,6 +1056,7 @@ testResult_t run() {
   PRINT("# Avg bus bandwidth    : %g %s\n", bw[0], check_avg_bw == -1 ? "" : (bw[0] < check_avg_bw*(0.9) ? "FAILED" : "OK"));
   PRINT("#\n");
 #ifdef MPI_SUPPORT
+  MPI_Comm_free(&mpi_comm);
   MPI_Finalize();
 #endif