Prevent WR index truncation in the InfiniBand transport plugin

[ROCm/rccl commit: 6e24ef4e1f]
Цей коміт міститься в:
Dmitrii Gabor authored and Sylvain Jeaugey committed 2023-06-28 11:39:19 +02:00
1 джерело 2dc2c86ec1
коміт b1b47d2e18
1 file changed
+1 -1
+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