Set IB default settings to 2 QPs per connection, no split.

Mellanox · Jan 3, 2024 · 17e4f1e · 17e4f1e
1 parent 82400b2
commit 17e4f1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ib_plugin.c b/src/ib_plugin.c
@@ -247,7 +247,7 @@ ncclResult_t ncclIbInit(ncclDebugLogger_t logFunction) {
   return nccl_p2p_ib_init(&ncclNIbDevs, ncclIbDevs, ncclIbIfName, &ncclIbIfAddr, &ncclIbAsyncThread, logFunction);
 }
 
-NCCL_PARAM(IbQpsPerConn, "IB_QPS_PER_CONNECTION", 1);
+NCCL_PARAM(IbQpsPerConn, "IB_QPS_PER_CONNECTION", 2);
 
 ncclResult_t ncclIbInitVerbs(int dev, struct ibv_context* ctx, struct ncclIbVerbs* verbs) {
   verbs->dev = dev;
@@ -746,7 +746,7 @@ ncclResult_t ncclIbDeregMr(void* comm, void* mhandle) {
   return res;
 }
 
-NCCL_PARAM(IbSplitDataOnQps, "IB_SPLIT_DATA_ON_QPS", 1);
+NCCL_PARAM(IbSplitDataOnQps, "IB_SPLIT_DATA_ON_QPS", 0);
 
 ncclResult_t ncclIbMultiSend(struct ncclIbSendComm* comm, int slot) {
   struct ncclIbRequest** reqs = comm->fifoReqs[slot];