From 60c74f63faf700c4e1d871a48afd7ba4274121bd Mon Sep 17 00:00:00 2001
From: gilbertlee-amd <44450918+gilbertlee-amd@users.noreply.github.com>
Date: Mon, 1 Feb 2021 09:48:09 -0700
Subject: [PATCH] [TransferBench] Restore some previous fixes - memory leak,
 PCIe address (#314)

[ROCm/rccl commit: 62e0447e9a50553985f4f08549b9b146652daf52]
---
 projects/rccl/tools/TransferBench/TransferBench.cpp | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/projects/rccl/tools/TransferBench/TransferBench.cpp b/projects/rccl/tools/TransferBench/TransferBench.cpp
index faed8dae44..a6ab78c46d 100644
--- a/projects/rccl/tools/TransferBench/TransferBench.cpp
+++ b/projects/rccl/tools/TransferBench/TransferBench.cpp
@@ -540,11 +540,12 @@ void DisplayTopology()
   printf("        |");
   for (int j = 0; j < numGpuDevices; j++)
     printf(" GPU %02d |", j);
-  printf("\n");
+  printf(" PCIe Bus ID\n");
   for (int j = 0; j <= numGpuDevices; j++)
     printf("--------+");
-  printf("\n");
+  printf("-------------\n");
 
+  char pciBusId[20];
   for (int i = 0; i < numGpuDevices; i++)
   {
     printf(" GPU %02d |", i);
@@ -565,7 +566,8 @@ void DisplayTopology()
                hopCount);
       }
     }
-    printf("\n");
+    HIP_CALL(hipDeviceGetPCIBusId(pciBusId, 20, i));
+    printf(" %s\n", pciBusId);
   }
 }
 
@@ -865,6 +867,7 @@ void CheckOrFill(ModeType mode, int N, bool isMemset, bool isHipCall, float* ptr
         exit(1);
       }
     }
+    free(hostBuffer);
   }
 
   free(refBuffer);