UCP/AM: Adjust max_short for UCP_AM_SEND_REPLY

openucx · Apr 29, 2021 · 05ac481 · 05ac481
1 parent 7456fe5
commit 05ac481
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/ucp/core/ucp_am.c b/src/ucp/core/ucp_am.c
@@ -23,9 +23,6 @@
 #include <ucs/datastruct/array.inl>
 
 
-#define UCP_AM_SHORT_REPLY_MAX_SIZE  (UCS_ALLOCA_MAX_SIZE - \
-                                      sizeof(ucs_ptr_map_key_t))
-
 UCS_ARRAY_IMPL(ucp_am_cbs, unsigned, ucp_am_entry_t, static)
 
 ucs_status_t ucp_am_init(ucp_worker_h worker)
@@ -902,9 +899,8 @@ ucp_am_send_req(ucp_request_t *req, size_t count,
 static UCS_F_ALWAYS_INLINE ssize_t
 ucp_am_get_short_max_reply(ucp_ep_h ep)
 {
-    return ucs_min(ucp_ep_config(ep)->am_u.max_eager_short -
-                   sizeof(ucs_ptr_map_key_t),
-                   UCP_AM_SHORT_REPLY_MAX_SIZE);
+    return ucs_min(ucp_ep_config(ep)->am_u.max_eager_short,
+                   UCS_ALLOCA_MAX_SIZE) - sizeof(ucs_ptr_map_key_t);
 }
 
 static UCS_F_ALWAYS_INLINE ucs_status_t