btl/base_am_rdma: fix handling of btl_send returning 1

Active message RDMA uses btl_send to send the initial
request and RDMA response.

btl_send will return 0 when the descriptor has been
successfully queued for send, and will return 1 when
the desciptor has been successfully sent.

Currently, active message RDMA treats the return value
1 as an error, thus will either return the value
to caller, or retry the send.

This patch address the issue by correctly handling
the return value 1.

Signed-off-by: Wei Zhang <[email protected]>

Author: wzamazon

opal/mca/btl/base/btl_base_am_rdma.c

@@ -372,6 +372,7 @@ static inline int mca_btl_base_am_rdma_advance(mca_btl_base_module_t *btl,
                                                mca_btl_base_rdma_context_t *context,
                                                bool send_descriptor)
 {
+    int ret;
     const size_t remaining = context->total_size - context->sent;
 
     if (0 == remaining) {
@@ -409,7 +410,13 @@ static inline int mca_btl_base_am_rdma_advance(mca_btl_base_module_t *btl,
     }
 
     if (send_descriptor) {
-        return btl->btl_send(btl, endpoint, descriptor, mca_btl_base_rdma_tag(hdr->type));
+        /* This descriptor was created with MCA_BTL_DES_SEND_ALWAYS_CALLBACK set, so safe to treat
+         * OPAL_SUCCESS and 1 return codes the same */
+        assert(0 != (descriptor->des_flags && MCA_BTL_DES_SEND_ALWAYS_CALLBACK));
+        ret = btl->btl_send(btl, endpoint, descriptor, mca_btl_base_rdma_tag(hdr->type));
+        if (ret == 1)
+            ret = OPAL_SUCCESS;
+        return ret;
     }
 
     /* queue for later to avoid btl_send in callback */
@@ -614,7 +621,13 @@ static int mca_btl_base_am_rdma_respond(mca_btl_base_module_t *btl,
 
     send_descriptor->des_cbfunc = NULL;
 
+    /* There is no callback for the response descriptor, therefore it is
+     * safe to treat 0 and 1 return codes the same
+     */
     int ret = btl->btl_send(btl, endpoint, send_descriptor, mca_btl_base_rdma_resp_tag());
+    if (ret == 1)
+        ret = OPAL_SUCCESS;
+
     if (OPAL_UNLIKELY(OPAL_SUCCESS != ret)) {
         *descriptor = send_descriptor;
     }
@@ -779,6 +792,8 @@ static int mca_btl_base_am_rdma_progress(void)
 
     OPAL_THREAD_SCOPED_LOCK(&default_module.mutex, ACTION1);
 
+    /* The initiator descriptor was created with MCA_BTL_DES_SEND_ALWAYS_CALLBACK set, so safe to treat
+     * 0 and 1 return codes of btl_send the same */
 #define ACTION2                                                         \
     mca_btl_base_am_rdma_queued_descriptor_t *descriptor, *next;        \
     OPAL_LIST_FOREACH_SAFE (descriptor, next,                           \
@@ -787,11 +802,12 @@ static int mca_btl_base_am_rdma_progress(void)
         mca_btl_base_rdma_context_t *context =                          \
             (mca_btl_base_rdma_context_t *)                             \
             descriptor->descriptor->des_context;                        \
+        assert(0 != (descriptor->descriptor->des_flags && MCA_BTL_DES_SEND_ALWAYS_CALLBACK)); \
         int ret = descriptor->btl->btl_send(descriptor->btl,            \
                                             descriptor->endpoint,       \
                                             descriptor->descriptor,     \
                                             mca_btl_base_rdma_tag(context->type)); \
-        if (OPAL_SUCCESS == ret) {                                      \
+        if (OPAL_SUCCESS == ret || 1 == ret) {                                      \
             opal_list_remove_item(&default_module.queued_initiator_descriptors, \
                                   &descriptor->super);                  \
         }                                                               \