PML/UCX: improved error processing in MPI_Recv

- added OPAL_LIKELY macro Signed-off-by: Sergey Oblomov <sergeyo@nvidia.com>
open-mpi · Nov 2, 2020 · 4ef5fc2 · 4ef5fc2
1 parent 4ddfab2
commit 4ef5fc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ompi/mca/pml/ucx/pml_ucx_request.h b/ompi/mca/pml/ucx/pml_ucx_request.h
@@ -196,7 +196,7 @@ static inline int mca_pml_ucx_set_recv_status_safe(ompi_status_public_t* mpi_sta
 {
     if (mpi_status != MPI_STATUS_IGNORE) {
         return mca_pml_ucx_set_recv_status(mpi_status, ucp_status, info);
-    } else if ((ucp_status == UCS_OK) || (ucp_status == UCS_ERR_CANCELED)) {
+    } else if (OPAL_LIKELY(ucp_status == UCS_OK) || (ucp_status == UCS_ERR_CANCELED)) {
         return UCS_OK;
     } else if (ucp_status == UCS_ERR_MESSAGE_TRUNCATED) {
         return MPI_ERR_TRUNCATE;