Prevent WR index truncation in the InfiniBand transport plugin

Ta zmiana jest zawarta w:
Dmitrii Gabor authored and Sylvain Jeaugey committed 2023-06-28 11:39:19 +02:00
1 rodzic ea38312273
złożenie 6e24ef4e1f
Plik 1 zmieniony
+1 -1
+1 -1
Wyświetl plik
@@ -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