Skip to content

Commit 8b3b9a3

Browse files
authored
Merge pull request #10404 from MamziB/mamzi/get-accum
OSC/UCX: Fix the get/put race condition in get_accumulate
2 parents dda17d4 + ccd615d commit 8b3b9a3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

ompi/mca/osc/ucx/osc_ucx_comm.c

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -945,6 +945,11 @@ int get_accumulate_req(const void *origin_addr, int origin_count,
945945
return ret;
946946
}
947947

948+
ret = opal_common_ucx_wpmem_flush(module->mem, OPAL_COMMON_UCX_SCOPE_EP, target);
949+
if (ret != OMPI_SUCCESS) {
950+
return ret;
951+
}
952+
948953
if (op != &ompi_mpi_op_no_op.op) {
949954
if (op == &ompi_mpi_op_replace.op) {
950955
ret = ompi_osc_ucx_put(origin_addr, origin_count, origin_dt,

0 commit comments

Comments
 (0)