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

osc/rdma: use extent of the appropriate datatype in ompi_osc_rdma_rget_accumulate_internal()

Просмотр исходного кода 
origin_datatype and target_datatype might be different and hence have different extent,
so use either origin_extent or target_extent when appropriate.

Refs open-mpi/ompi#3569

Signed-off-by: Gilles Gouaillardet <gilles@rist.or.jp>
Этот коммит содержится в:
Gilles Gouaillardet 2017-05-24 15:00:18 +09:00
родитель 10b103a37d
Коммит 0f79259b94
1 изменённых файлов: 9 добавлений и 7 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

16
ompi/mca/osc/rdma/osc_rdma_accumulate.c
Просмотреть файл

@ -1015,7 +1015,7 @@ int ompi_osc_rdma_rget_accumulate_internal (ompi_osc_rdma_sync_t *sync, const vo
ompi_osc_rdma_module_t *module = sync->module; ompi_osc_rdma_module_t *module = sync->module;
mca_btl_base_registration_handle_t *target_handle; mca_btl_base_registration_handle_t *target_handle;
uint64_t target_address; uint64_t target_address;
ptrdiff_t lb, extent; ptrdiff_t lb, origin_extent, target_extent;
int ret; int ret;
/* short-circuit case. note that origin_count may be 0 if op is MPI_NO_OP */ /* short-circuit case. note that origin_count may be 0 if op is MPI_NO_OP */
@ -1027,20 +1027,22 @@ int ompi_osc_rdma_rget_accumulate_internal (ompi_osc_rdma_sync_t *sync, const vo
return OMPI_SUCCESS; return OMPI_SUCCESS;
} }
(void) ompi_datatype_get_extent (origin_datatype, &lb, &extent); (void) ompi_datatype_get_extent (target_datatype, &lb, &target_extent);
ret = osc_rdma_get_remote_segment (module, peer, target_disp, extent * target_count, &target_address, &target_handle); ret = osc_rdma_get_remote_segment (module, peer, target_disp, target_extent * target_count, &target_address, &target_handle);
if (OPAL_UNLIKELY(OMPI_SUCCESS != ret)) { if (OPAL_UNLIKELY(OMPI_SUCCESS != ret)) {
return ret; return ret;
} }
if (module->acc_single_intrinsic && extent <= 8) { (void) ompi_datatype_get_extent (origin_datatype, &lb, &origin_extent);
if (module->acc_single_intrinsic && origin_extent <= 8) {
if (module->acc_use_amo && ompi_datatype_is_predefined (origin_datatype)) { if (module->acc_use_amo && ompi_datatype_is_predefined (origin_datatype)) {
if (NULL == result_addr) { if (NULL == result_addr) {
ret = ompi_osc_rdma_acc_single_atomic (sync, origin_addr, origin_datatype, extent, peer, target_address, ret = ompi_osc_rdma_acc_single_atomic (sync, origin_addr, origin_datatype, origin_extent, peer, target_address,
target_handle, op, request); target_handle, op, request);
} else { } else {
ret = ompi_osc_rdma_fetch_and_op_atomic (sync, origin_addr, result_addr, origin_datatype, extent, peer, target_address, ret = ompi_osc_rdma_fetch_and_op_atomic (sync, origin_addr, result_addr, origin_datatype, origin_extent, peer, target_address,
target_handle, op, request); target_handle, op, request);
} }
@ -1049,7 +1051,7 @@ int ompi_osc_rdma_rget_accumulate_internal (ompi_osc_rdma_sync_t *sync, const vo
} }
} }
ret = ompi_osc_rdma_fetch_and_op_cas (sync, origin_addr, result_addr, origin_datatype, extent, peer, target_address, ret = ompi_osc_rdma_fetch_and_op_cas (sync, origin_addr, result_addr, origin_datatype, origin_extent, peer, target_address,
target_handle, op, request); target_handle, op, request);
if (OMPI_SUCCESS == ret) { if (OMPI_SUCCESS == ret) {
return OMPI_SUCCESS; return OMPI_SUCCESS;