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); } }