Fix incorrect rocm-smi ID conversion (#648)

(cherry picked from commit 57764f8)
ROCm · Dec 1, 2022 · 31f4553 · 31f4553
1 parent f7aec7d
commit 31f4553
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/misc/rocm_smi_wrap.cc b/src/misc/rocm_smi_wrap.cc
@@ -59,7 +59,7 @@ ncclResult_t rocm_smi_getDevicePciBusIdString(uint32_t deviceIndex, char* busId,
    *  | Device   | [ 7: 3] |
    *  | Function | [ 2: 0] |
    **/
-  snprintf(busId, len, "%04lx:%02lx:%02lx.%01lx", (id) >> 32, (id & 0xff00) >> 8, (id & 0xf0) >> 4, (id & 0x3));
+  snprintf(busId, len, "%04lx:%02lx:%02lx.%01lx", (id) >> 32, (id & 0xff00) >> 8, (id & 0xf8) >> 3, (id & 0x7));
   return ncclSuccess;
 }