From 5d52f0285cc5e44f9e1558b0a766f29fddd6af40 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()

[ROCm/rccl-tests commit: c6afef0b6f76ffc55d4172d971be6cf5a08a73a4]
---
 projects/rccl-tests/src/common.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/projects/rccl-tests/src/common.cu b/projects/rccl-tests/src/common.cu
index 8588047d78..4ac00fb3d7 100644
--- a/projects/rccl-tests/src/common.cu
+++ b/projects/rccl-tests/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