@@ -814,9 +814,10 @@ int ompi_osc_ucx_rput(const void *origin_addr, int origin_count,
814
814
}
815
815
816
816
mca_osc_ucx_component .num_incomplete_req_ops ++ ;
817
+ /* TODO: investigate whether ucp_worker_flush_nb is a better choice here */
817
818
ret = opal_common_ucx_wpmem_fetch_nb (module -> mem , UCP_ATOMIC_FETCH_OP_FADD ,
818
819
0 , target , & (module -> req_result ),
819
- sizeof (uint64_t ), remote_addr ,
820
+ sizeof (uint64_t ), remote_addr & (~ 0x7 ) ,
820
821
req_completion , ucx_req );
821
822
if (ret != OMPI_SUCCESS ) {
822
823
return ret ;
@@ -865,9 +866,10 @@ int ompi_osc_ucx_rget(void *origin_addr, int origin_count,
865
866
}
866
867
867
868
mca_osc_ucx_component .num_incomplete_req_ops ++ ;
869
+ /* TODO: investigate whether ucp_worker_flush_nb is a better choice here */
868
870
ret = opal_common_ucx_wpmem_fetch_nb (module -> mem , UCP_ATOMIC_FETCH_OP_FADD ,
869
871
0 , target , & (module -> req_result ),
870
- sizeof (uint64_t ), remote_addr ,
872
+ sizeof (uint64_t ), remote_addr & (~ 0x7 ) ,
871
873
req_completion , ucx_req );
872
874
if (ret != OMPI_SUCCESS ) {
873
875
return ret ;
0 commit comments