From a062c802984ade5bb9e838540e8aecfca4e7d1ee Mon Sep 17 00:00:00 2001
From: gilbertlee-amd <44450918+gilbertlee-amd@users.noreply.github.com>
Date: Wed, 21 Oct 2020 16:13:36 -0600
Subject: [PATCH] [TransferBench] Displaying PCIe Bus ID (#288)

* Adding PCIe BusID per GPU in topology display

[ROCm/rccl commit: 61e1a71d14ce7917dbb4a1cd199599dca4795656]
---
 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 e1d5f60777..1bca854f0d 100644
--- a/projects/rccl/tools/TransferBench/TransferBench.cpp
+++ b/projects/rccl/tools/TransferBench/TransferBench.cpp
@@ -522,10 +522,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++)
   {
@@ -547,7 +549,8 @@ void DisplayTopology()
                hopCount);
       }
     }
-    printf("\n");
+    HIP_CALL(hipDeviceGetPCIBusId(pciBusId, 20, i));
+    printf(" %s\n", pciBusId);
   }
 }