ports/openmpi
ports/openmpi
1
1
Форкнуть 0
Код Релизы Активность

PML/SPML/UCX: add UCX MT support to PML and SPML.

Просмотр исходного кода 
Signed-off-by: Xin Zhao <xinz@mellanox.com>
Этот коммит содержится в:
Xin Zhao 2016-12-15 23:59:15 +02:00
родитель 00b87ea829
Коммит 0ecf3c951c
2 изменённых файлов: 7 добавлений и 4 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

6
ompi/mca/pml/ucx/pml_ucx.c
Просмотреть файл

@ -134,12 +134,14 @@ int mca_pml_ucx_open(void)
UCP_PARAM_FIELD_REQUEST_SIZE |
UCP_PARAM_FIELD_REQUEST_INIT |
UCP_PARAM_FIELD_REQUEST_CLEANUP |
UCP_PARAM_FIELD_TAG_SENDER_MASK;
UCP_PARAM_FIELD_TAG_SENDER_MASK |
UCP_PARAM_FIELD_MT_WORKERS_SHARED;
params.features = UCP_FEATURE_TAG;
params.request_size = sizeof(ompi_request_t);
params.request_init = mca_pml_ucx_request_init;
params.request_cleanup = mca_pml_ucx_request_cleanup;
params.tag_sender_mask = PML_UCX_SPECIFIC_SOURCE_MASK;
params.mt_workers_shared = 0;
status = ucp_init(&params, config, &ompi_pml_ucx.ucp_context);
ucp_config_release(config);
@ -183,7 +185,7 @@ int mca_pml_ucx_init(void)
/* TODO check MPI thread mode */
params.field_mask = UCP_WORKER_PARAM_FIELD_THREAD_MODE;
params.thread_mode = UCS_THREAD_MODE_SINGLE;
params.thread_mode = UCS_THREAD_MODE_MULTI;
status = ucp_worker_create(ompi_pml_ucx.ucp_context, &params,
&ompi_pml_ucx.ucp_worker);

5
oshmem/mca/spml/ucx/spml_ucx_component.c
Просмотреть файл

@ -127,7 +127,8 @@ static int mca_spml_ucx_component_open(void)
memset(&params, 0, sizeof(params));
params.field_mask = UCP_PARAM_FIELD_FEATURES;
params.features = UCP_FEATURE_RMA|UCP_FEATURE_AMO32|UCP_FEATURE_AMO64;
params.features = UCP_FEATURE_RMA|UCP_FEATURE_AMO32|UCP_FEATURE_AMO64|UCP_PARAM_FIELD_MT_WORKERS_SHARED;
params.mt_workers_shared = 0;
err = ucp_init(&params, ucp_config, &mca_spml_ucx.ucp_context);
ucp_config_release(ucp_config);
@ -153,7 +154,7 @@ static int spml_ucx_init(void)
ucs_status_t err;
params.field_mask = UCP_WORKER_PARAM_FIELD_THREAD_MODE;
params.thread_mode = UCS_THREAD_MODE_SINGLE;
params.thread_mode = UCS_THREAD_MODE_MULTI;
err = ucp_worker_create(mca_spml_ucx.ucp_context, &params,
&mca_spml_ucx.ucp_worker);