[TransferBench] Displaying PCIe Bus ID (#288)
* Adding PCIe BusID per GPU in topology display
[ROCm/rccl commit: 61e1a71d14]
Этот коммит содержится в:
1 родитель
0282595de5
Коммит
a062c80298
1 file changed
+6
-3
@@ -522,10 +522,12 @@ void DisplayTopology()
|
|||||||
printf(" |");
|
printf(" |");
|
||||||
for (int j = 0; j < numGpuDevices; j++)
|
for (int j = 0; j < numGpuDevices; j++)
|
||||||
printf(" GPU %02d |", j);
|
printf(" GPU %02d |", j);
|
||||||
printf("\n");
|
printf(" PCIe Bus ID\n");
|
||||||
for (int j = 0; j <= numGpuDevices; j++)
|
for (int j = 0; j <= numGpuDevices; j++)
|
||||||
printf("--------+");
|
printf("--------+");
|
||||||
printf("\n");
|
printf("-------------\n");
|
||||||
|
|
||||||
|
char pciBusId[20];
|
||||||
|
|
||||||
for (int i = 0; i < numGpuDevices; i++)
|
for (int i = 0; i < numGpuDevices; i++)
|
||||||
{
|
{
|
||||||
@@ -547,7 +549,8 @@ void DisplayTopology()
|
|||||||
hopCount);
|
hopCount);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
printf("\n");
|
HIP_CALL(hipDeviceGetPCIBusId(pciBusId, 20, i));
|
||||||
|
printf(" %s\n", pciBusId);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user