Sort IB devices based on device name (#413)

(cherry picked from commit 2d0ed8d)
ROCm · Aug 6, 2021 · 6847913 · 6847913
1 parent d306fd3
commit 6847913
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transport/net_ib.cc b/src/transport/net_ib.cc
@@ -197,6 +197,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<ncclNIbDevs; d++) {