2.15.1-1

Add support for H100 (sm90). Make sure NCCL kernel honor user stream priorities. [ROCm/rccl commit: da8152e57a]
2022-09-27 02:31:13 -07:00
parent 37ccaf1f82
commit b4bac0d15a
22 changed files with 303 additions and 76 deletions
@@ -211,8 +211,8 @@ static void groupCleanup(struct ncclComm** groupCommHeadPtr, struct ncclComm** g
    for (int i = 0; i < comm->nRanks; i++) {
      comm->tasks.peers[i].sendSeen = false;
      comm->tasks.peers[i].recvSeen = false;
-      comm->connectSend[i] = 0;
-      comm->connectRecv[i] = 0;
+      comm->connectSend[i] = 0UL;
+      comm->connectRecv[i] = 0UL;
    }
    comm->unlaunchedPlansHead = nullptr;
    // Reclaim abandoned kernel plan memory. Note ncclWork structs were already