From 5f5beba3920131a01fccc01d4e67bb604188401d Mon Sep 17 00:00:00 2001 From: Sergey Oblomov Date: Mon, 6 Dec 2021 17:50:02 +0200 Subject: [PATCH] UCP/TAG/SEND: fixed datatype issue - added slow path where datatype is processed (cherry picked from commit fc4e270cbe355dd6c30c611d2247adab3ae2609f) --- src/ucp/tag/tag_send.c | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/ucp/tag/tag_send.c b/src/ucp/tag/tag_send.c index 35752a9bea9..7054cac077d 100644 --- a/src/ucp/tag/tag_send.c +++ b/src/ucp/tag/tag_send.c @@ -264,9 +264,15 @@ UCS_PROFILE_FUNC(ucs_status_ptr_t, ucp_tag_send_nbx, contig_length, tag); ucp_request_send_check_status(status, ret, goto out); } - } else { + } else if (attr_mask == UCP_OP_ATTR_FLAG_NO_IMM_CMPL) { datatype = ucp_dt_make_contig(1); contig_length = count; + } else { + /* UCP_OP_ATTR_FIELD_DATATYPE | UCP_OP_ATTR_FLAG_NO_IMM_CMPL */ + datatype = param->datatype; + if (UCP_DT_IS_CONTIG(datatype)) { + contig_length = ucp_contig_dt_length(datatype, count); + } } if (ucs_unlikely(param->op_attr_mask & UCP_OP_ATTR_FLAG_FORCE_IMM_CMPL)) {