From d516392facd219c4295e38de7bd7e0bf736bb036 Mon Sep 17 00:00:00 2001 From: David Addison Date: Fri, 28 Feb 2025 13:23:26 -0800 Subject: [PATCH] Add PCI domain and device ID for GPU device BDF display [ROCm/rccl-tests commit: b4300cc79d05dd35e26f13afcdb4938f29aa31a5] --- projects/rccl-tests/src/common.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/projects/rccl-tests/src/common.cu b/projects/rccl-tests/src/common.cu index 9277ea2b15..0d4dfc1944 100644 --- a/projects/rccl-tests/src/common.cu +++ b/projects/rccl-tests/src/common.cu @@ -984,8 +984,8 @@ testResult_t run() { int rank = proc*nThreads*nGpus+i; cudaDeviceProp prop; CUDACHECK(cudaGetDeviceProperties(&prop, cudaDev)); - len += snprintf(line+len, MAX_LINE-len, "# Rank %2d Group %2d Pid %6d on %10s device %2d [0x%02x] %s\n", - rank, color, getpid(), hostname, cudaDev, prop.pciBusID, prop.name); + len += snprintf(line+len, MAX_LINE-len, "# Rank %2d Group %2d Pid %6d on %10s device %2d [%04x:%02x:%02x] %s\n", + rank, color, getpid(), hostname, cudaDev, prop.pciDomainID, prop.pciBusID, prop.pciDeviceID, prop.name); maxMem = std::min(maxMem, prop.totalGlobalMem); }