Skip to content

Commit 3bc39f2

Browse files
committed
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 <[email protected]>
1 parent 155ea45 commit 3bc39f2

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

src/utils/ucx/ucx_utils.cpp

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -450,18 +450,13 @@ nixlUcxContext::nixlUcxContext(std::vector<std::string> devs,
450450
unsigned major_version, minor_version, release_number;
451451
ucp_get_version(&major_version, &minor_version, &release_number);
452452

453-
config.modify ("ADDRESS_VERSION", "v2");
454-
config.modify ("RNDV_THRESH", "inf");
453+
config.modify("ADDRESS_VERSION", "v2");
454+
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)) {
458-
config.modify ("MAX_COMPONENT_MDS", "32");
459-
}
460-
461-
if (ucp_version >= UCP_VERSION(1, 20)) {
462-
config.modify ("MAX_RMA_RAILS", "4");
463-
} else {
464-
config.modify ("MAX_RMA_RAILS", "2");
459+
config.modify("MAX_COMPONENT_MDS", "32");
465460
}
466461

467462
const auto status = ucp_init (&ucp_params, config.getUcpConfig(), &ctx);

0 commit comments

Comments
 (0)