Merge pull request openucx#6307 from bureddy/put_zcopy-v1.10.x

UCP/RNDV: Set addr NULL in RTS if reg md_map is NULL. - v1.10.x
kunpengcompute · Feb 10, 2021 · abb8737 · abb8737
2 parents 364f7cf + d0f98a4
commit abb8737
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/ucp/rndv/rndv.c b/src/ucp/rndv/rndv.c
@@ -107,7 +107,9 @@ size_t ucp_rndv_rts_pack(ucp_request_t *sreq, ucp_rndv_rts_hdr_t *rndv_rts_hdr,
 
     /* Pack remote keys (which can be empty list) */
     if (UCP_DT_IS_CONTIG(sreq->send.datatype) &&
-        ucp_rndv_is_get_zcopy(sreq, worker->context)) {
+        ucp_rndv_is_get_zcopy(sreq, worker->context) &&
+        (UCP_MEM_IS_HOST(sreq->send.mem_type) ||
+         (sreq->send.state.dt.dt.contig.md_map != 0))) {
         /* pack rkey, ask target to do get_zcopy */
         rndv_rts_hdr->address = (uintptr_t)sreq->send.buffer;
         rkey_buf              = UCS_PTR_BYTE_OFFSET(rndv_rts_hdr,