Prevent WR index truncation in the InfiniBand transport plugin
Этот коммит содержится в:
коммит произвёл
Sylvain Jeaugey
родитель
ea38312273
Коммит
6e24ef4e1f
@@ -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
|
||||
|
||||
Ссылка в новой задаче
Block a user