From 3bc39f2f50ecc1efa01b2f722470f541d6b3f608 Mon Sep 17 00:00:00 2001 From: Mikhail Brinskii Date: Fri, 14 Nov 2025 15:02:49 +0200 Subject: [PATCH] UCX: Always use 2 RMA rails There is no need to use four rails with IB, as it can cause performance degradation in certain cases. This change was originally introduced for AWS/EFA support, but the preferred plugin for EFA is now libfabric. Signed-off-by: Mikhail Brinskii --- src/utils/ucx/ucx_utils.cpp | 13 ++++--------- 1 file changed, 4 insertions(+), 9 deletions(-) diff --git a/src/utils/ucx/ucx_utils.cpp b/src/utils/ucx/ucx_utils.cpp index d337e7247..7866395ff 100644 --- a/src/utils/ucx/ucx_utils.cpp +++ b/src/utils/ucx/ucx_utils.cpp @@ -450,18 +450,13 @@ nixlUcxContext::nixlUcxContext(std::vector devs, unsigned major_version, minor_version, release_number; ucp_get_version(&major_version, &minor_version, &release_number); - config.modify ("ADDRESS_VERSION", "v2"); - config.modify ("RNDV_THRESH", "inf"); + config.modify("ADDRESS_VERSION", "v2"); + config.modify("RNDV_THRESH", "inf"); + config.modify("MAX_RMA_RAILS", "2"); unsigned ucp_version = UCP_VERSION(major_version, minor_version); if (ucp_version >= UCP_VERSION(1, 19)) { - config.modify ("MAX_COMPONENT_MDS", "32"); - } - - if (ucp_version >= UCP_VERSION(1, 20)) { - config.modify ("MAX_RMA_RAILS", "4"); - } else { - config.modify ("MAX_RMA_RAILS", "2"); + config.modify("MAX_COMPONENT_MDS", "32"); } const auto status = ucp_init (&ucp_params, config.getUcpConfig(), &ctx);