UCP/TAG/SEND: fixed datatype issue

- added slow path where datatype is processed (cherry picked from commit fc4e270)
openucx · Jan 10, 2022 · 5f5beba · 5f5beba
1 parent 94140fd
commit 5f5beba
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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)) {