Skip to content

Commit 57c0c84

Browse files
authored
Merge pull request #2603 from xinzhao3/topic/revert-ucx-mt
Revert "PML/SPML/UCX: add UCX MT support to PML and SPML."
2 parents 5737a45 + 2d77912 commit 57c0c84

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

ompi/mca/pml/ucx/pml_ucx.c

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -134,14 +134,12 @@ int mca_pml_ucx_open(void)
134134
UCP_PARAM_FIELD_REQUEST_SIZE |
135135
UCP_PARAM_FIELD_REQUEST_INIT |
136136
UCP_PARAM_FIELD_REQUEST_CLEANUP |
137-
UCP_PARAM_FIELD_TAG_SENDER_MASK |
138-
UCP_PARAM_FIELD_MT_WORKERS_SHARED;
137+
UCP_PARAM_FIELD_TAG_SENDER_MASK;
139138
params.features = UCP_FEATURE_TAG;
140139
params.request_size = sizeof(ompi_request_t);
141140
params.request_init = mca_pml_ucx_request_init;
142141
params.request_cleanup = mca_pml_ucx_request_cleanup;
143142
params.tag_sender_mask = PML_UCX_SPECIFIC_SOURCE_MASK;
144-
params.mt_workers_shared = 0;
145143

146144
status = ucp_init(&params, config, &ompi_pml_ucx.ucp_context);
147145
ucp_config_release(config);
@@ -185,7 +183,7 @@ int mca_pml_ucx_init(void)
185183

186184
/* TODO check MPI thread mode */
187185
params.field_mask = UCP_WORKER_PARAM_FIELD_THREAD_MODE;
188-
params.thread_mode = UCS_THREAD_MODE_MULTI;
186+
params.thread_mode = UCS_THREAD_MODE_SINGLE;
189187

190188
status = ucp_worker_create(ompi_pml_ucx.ucp_context, &params,
191189
&ompi_pml_ucx.ucp_worker);

oshmem/mca/spml/ucx/spml_ucx_component.c

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,8 +127,7 @@ static int mca_spml_ucx_component_open(void)
127127

128128
memset(&params, 0, sizeof(params));
129129
params.field_mask = UCP_PARAM_FIELD_FEATURES;
130-
params.features = UCP_FEATURE_RMA|UCP_FEATURE_AMO32|UCP_FEATURE_AMO64|UCP_PARAM_FIELD_MT_WORKERS_SHARED;
131-
params.mt_workers_shared = 0;
130+
params.features = UCP_FEATURE_RMA|UCP_FEATURE_AMO32|UCP_FEATURE_AMO64;
132131

133132
err = ucp_init(&params, ucp_config, &mca_spml_ucx.ucp_context);
134133
ucp_config_release(ucp_config);
@@ -154,7 +153,7 @@ static int spml_ucx_init(void)
154153
ucs_status_t err;
155154

156155
params.field_mask = UCP_WORKER_PARAM_FIELD_THREAD_MODE;
157-
params.thread_mode = UCS_THREAD_MODE_MULTI;
156+
params.thread_mode = UCS_THREAD_MODE_SINGLE;
158157

159158
err = ucp_worker_create(mca_spml_ucx.ucp_context, &params,
160159
&mca_spml_ucx.ucp_worker);

0 commit comments

Comments
 (0)