From de0c586bad105456c845fec203ba7f2babd2cf60 Mon Sep 17 00:00:00 2001 From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com> Date: Tue, 3 Aug 2021 15:32:41 -0700 Subject: [PATCH] Sort IB devices based on device name (#413) [ROCm/rccl commit: 2d0ed8dff64adf9120a3f54adae792914e4e34ae] --- projects/rccl/src/transport/net_ib.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/projects/rccl/src/transport/net_ib.cc b/projects/rccl/src/transport/net_ib.cc index 6009b07cd7..2ca0b51b78 100644 --- a/projects/rccl/src/transport/net_ib.cc +++ b/projects/rccl/src/transport/net_ib.cc @@ -199,6 +199,8 @@ ncclResult_t ncclIbInit(ncclDebugLogger_t logFunction) { if (ncclNIbDevs == 0) { INFO(NCCL_INIT|NCCL_NET, "NET/IB : No device found."); } else { + auto cmpIbDevs = [](const void* n1, const void* n2) { return strcmp(((struct ncclIbDev*)n1)->devName, ((struct ncclIbDev*)n2)->devName); }; + qsort(ncclIbDevs, ncclNIbDevs, sizeof(struct ncclIbDev), cmpIbDevs); char line[1024]; line[0] = '\0'; for (int d=0; d