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

osc/ucx: Fix data corruption with non-contiguous accumulates

Просмотр исходного кода 
Signed-off-by: Nysal Jan K.A <jnysal@in.ibm.com>
Этот коммит содержится в:
Nysal Jan K.A 2019-07-24 13:04:41 +05:30
родитель 94f26f5a51
Коммит 3529d44702
1 изменённых файлов: 6 добавлений и 4 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

10
ompi/mca/osc/ucx/osc_ucx_comm.c
Просмотреть файл

@ -510,12 +510,13 @@ int ompi_osc_ucx_accumulate(const void *origin_addr, int origin_count,
if ((op != &ompi_mpi_op_maxloc.op && op != &ompi_mpi_op_minloc.op) ||
ompi_datatype_is_contiguous_memory_layout(temp_dt, temp_count)) {
size_t temp_size;
char *curr_temp_addr = (char *)temp_addr;
ompi_datatype_type_size(temp_dt, &temp_size);
while (origin_ucx_iov_idx < origin_ucx_iov_count) {
int curr_count = origin_ucx_iov[origin_ucx_iov_idx].len / temp_size;
ompi_op_reduce(op, origin_ucx_iov[origin_ucx_iov_idx].addr,
temp_addr, curr_count, temp_dt);
temp_addr = (void *)((char *)temp_addr + curr_count * temp_size);
curr_temp_addr, curr_count, temp_dt);
curr_temp_addr += curr_count * temp_size;
origin_ucx_iov_idx++;
}
} else {
@ -730,12 +731,13 @@ int ompi_osc_ucx_get_accumulate(const void *origin_addr, int origin_count,
if ((op != &ompi_mpi_op_maxloc.op && op != &ompi_mpi_op_minloc.op) ||
ompi_datatype_is_contiguous_memory_layout(temp_dt, temp_count)) {
size_t temp_size;
char *curr_temp_addr = (char *)temp_addr;
ompi_datatype_type_size(temp_dt, &temp_size);
while (origin_ucx_iov_idx < origin_ucx_iov_count) {
int curr_count = origin_ucx_iov[origin_ucx_iov_idx].len / temp_size;
ompi_op_reduce(op, origin_ucx_iov[origin_ucx_iov_idx].addr,
temp_addr, curr_count, temp_dt);
temp_addr = (void *)((char *)temp_addr + curr_count * temp_size);
curr_temp_addr, curr_count, temp_dt);
curr_temp_addr += curr_count * temp_size;
origin_ucx_iov_idx++;
}
} else {