UCP/RECV: Save prev request flags

yosefe · Sep 8, 2020 · 5a6d574 · 5a6d574
1 parent a0f02de
commit 5a6d574
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/ucp/core/ucp_request.h b/src/ucp/core/ucp_request.h
@@ -269,6 +269,7 @@ struct ucp_request {
             ucp_dt_state_t        state;
             ucp_worker_t          *worker;
             uct_tag_context_t     uct_ctx;  /* Transport offload context */
+            unsigned              prev_flags;
 
             union {
                 struct {

diff --git a/src/ucp/tag/tag_recv.c b/src/ucp/tag/tag_recv.c
@@ -63,6 +63,7 @@ ucp_tag_recv_common(ucp_worker_h worker, void *buffer, size_t count,
             ucp_tag_eager_sync_send_ack(worker, rdesc + 1, rdesc->flags);
         }
 
+        req->recv.prev_flags          = req->flags;
         req->flags                    = UCP_REQUEST_FLAG_RECV | req_flags;
         hdr_len                       = rdesc->payload_offset;
         recv_len                      = rdesc->length - hdr_len;