Prevent WR index truncation in the InfiniBand transport plugin

Этот коммит содержится в:
Dmitrii Gabor
2023-06-22 13:42:33 +03:00
коммит произвёл Sylvain Jeaugey
родитель ea38312273
Коммит 6e24ef4e1f
+1 -1
Просмотреть файл
@@ -1064,7 +1064,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