From 6e24ef4e1f1eac9f104d115ef65429f179924ee7 Mon Sep 17 00:00:00 2001
From: Dmitrii Gabor <dmitryg1709@gmail.com>
Date: Thu, 22 Jun 2023 13:42:33 +0300
Subject: [PATCH] Prevent WR index truncation in the InfiniBand transport
 plugin

---
 src/transport/net_ib.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/transport/net_ib.cc b/src/transport/net_ib.cc
index 83c1d6d..861fa57 100644
--- a/src/transport/net_ib.cc
+++ b/src/transport/net_ib.cc
@@ -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