Merge pull request #795 from wenkaidu/nccl_sync

Sync latest update from NCCL
This commit is contained in:
Wenkai Du
2023-07-05 09:51:48 -07:00
committed by GitHub
+1 -1
View File
@@ -1098,7 +1098,7 @@ ncclResult_t ncclIbIsend(void* sendComm, void* data, int size, int tag, void* mh
int slot = (comm->fifoHead)%MAX_REQUESTS;
struct ncclIbRequest** reqs = comm->fifoReqs[slot];
slots = comm->fifo[slot];
int idx = comm->fifoHead+1;
uint64_t idx = comm->fifoHead+1;
if (slots[0].idx != idx) { *request = NULL; return ncclSuccess; }
nreqs = slots[0].nreqs;
// Wait until all data has arrived