PERF: Assertion on final ack message with RNDV

Author: iyastreb

src/tools/perf/lib/ucp_tests.cc

@@ -275,11 +275,21 @@ class ucp_perf_test_runner {
                               const ucp_am_recv_param_t *rx_params)
     {
         ucs_assert(!(rx_params->recv_attr & UCP_AM_RECV_ATTR_FLAG_DATA));
-        ucs_assert(length == ucx_perf_get_message_size(&m_perf.params));
+        unsigned my_index = rte_call(&m_perf, group_index);
+        if (my_index == 0) {
+            /* Receiver may only receive payload */
+            size_t UCS_V_UNUSED expected_length =
+                                    ucx_perf_get_message_size(&m_perf.params);
+            ucs_assertv(length == expected_length, "length=%zu, expected=%zu,"
+                        " index=%u", length, expected_length, my_index);
+        } else if (my_index == 1) {
+            /* Sender may only receive final ack */
+            ucs_assertv(length == 1, "length=%zu, expected=1, index=%u", length,
+                        my_index);
+        }
 
         ucs_status_ptr_t sp = ucp_am_recv_data_nbx(m_perf.ucp.worker, data,
-                                                   m_am_rx_buffer,
-                                                   m_am_rx_length,
+                                                   m_am_rx_buffer, length,
                                                    &m_am_rx_params);
         ucs_assert(UCS_PTR_IS_PTR(sp));
         ucp_request_release(sp);