@@ -1086,9 +1086,10 @@ int ompi_osc_ucx_rput(const void *origin_addr, int origin_count,
1086
1086
}
1087
1087
1088
1088
mca_osc_ucx_component .num_incomplete_req_ops ++ ;
1089
+ /* TODO: investigate whether ucp_worker_flush_nb is a better choice here */
1089
1090
ret = opal_common_ucx_wpmem_fetch_nb (module -> mem , UCP_ATOMIC_FETCH_OP_FADD ,
1090
1091
0 , target , & (module -> req_result ),
1091
- sizeof (uint64_t ), remote_addr ,
1092
+ sizeof (uint64_t ), remote_addr & (~ 0x7 ) ,
1092
1093
req_completion , ucx_req );
1093
1094
if (ret != OMPI_SUCCESS ) {
1094
1095
OMPI_OSC_UCX_REQUEST_RETURN (ucx_req );
@@ -1138,9 +1139,10 @@ int ompi_osc_ucx_rget(void *origin_addr, int origin_count,
1138
1139
}
1139
1140
1140
1141
mca_osc_ucx_component .num_incomplete_req_ops ++ ;
1142
+ /* TODO: investigate whether ucp_worker_flush_nb is a better choice here */
1141
1143
ret = opal_common_ucx_wpmem_fetch_nb (module -> mem , UCP_ATOMIC_FETCH_OP_FADD ,
1142
1144
0 , target , & (module -> req_result ),
1143
- sizeof (uint64_t ), remote_addr ,
1145
+ sizeof (uint64_t ), remote_addr & (~ 0x7 ) ,
1144
1146
req_completion , ucx_req );
1145
1147
if (ret != OMPI_SUCCESS ) {
1146
1148
OMPI_OSC_UCX_REQUEST_RETURN (ucx_req );
0 commit comments