UCP RMA: Fix put-based plugin

Mellanox · Oct 3, 2024 · 0a245b7 · 0a245b7
1 parent b246b19
commit 0a245b7
Show file tree

Hide file tree

Showing 2 changed files with 1,066 additions and 1,031 deletions.
diff --git a/src/p2p_plugin.c b/src/p2p_plugin.c
@@ -247,7 +247,7 @@ ncclResult_t nccl_p2p_ib_get_properties(ncclIbDev *devs, int dev, ncclNetPropert
   props->maxComms = ibDev->maxQp;
 
   if (p2p_plugin == NCCL_P2P_IB || p2p_plugin == NCCL_P2P_UCX ||
-      nccl_p2p_is_uct_plugin(p2p_plugin)) {
+      p2p_plugin == NCCL_P2P_UCX_RMA || nccl_p2p_is_uct_plugin(p2p_plugin)) {
     props->maxRecvs = NCCL_NET_IB_MAX_RECVS;
   } else {
     props->maxRecvs = 1;