Skip to content

Commit c9b7b97

Browse files
committed
UCX: Always use 2 RMA rails - v0.8.0
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 <[email protected]>
1 parent 2860e56 commit c9b7b97

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

src/utils/ucx/ucx_utils.cpp

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -452,18 +452,13 @@ nixlUcxContext::nixlUcxContext(std::vector<std::string> devs,
452452

453453
config.modify ("ADDRESS_VERSION", "v2");
454454
config.modify ("RNDV_THRESH", "inf");
455+
config.modify ("MAX_RMA_RAILS", "2");
455456

456457
unsigned ucp_version = UCP_VERSION(major_version, minor_version);
457458
if (ucp_version >= UCP_VERSION(1, 19)) {
458459
config.modify ("MAX_COMPONENT_MDS", "32");
459460
}
460461

461-
if (ucp_version >= UCP_VERSION(1, 20)) {
462-
config.modify ("MAX_RMA_RAILS", "4");
463-
} else {
464-
config.modify ("MAX_RMA_RAILS", "2");
465-
}
466-
467462
const auto status = ucp_init (&ucp_params, config.getUcpConfig(), &ctx);
468463
if (status != UCS_OK) {
469464
throw std::runtime_error ("Failed to create UCX context: " +

0 commit comments

Comments
 (0)