Merge pull request #23 from yosefe/topic/ucp-stream-short-set-req-sen…

…d-lane UCP/STREAM: Set correct req->send.lane in stream/short progress
yosefe · Sep 1, 2020 · 6f67308 · 6f67308
2 parents 480cb44 + 1640941
commit 6f67308
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ucp/stream/stream_send.c b/src/ucp/stream/stream_send.c
@@ -148,6 +148,8 @@ static ucs_status_t ucp_stream_contig_am_short(uct_pending_req_t *self)
     if (ucs_likely(status == UCS_OK)) {
         ucp_request_complete_send(req, UCS_OK);
     }
+
+    req->send.lane = ucp_ep_get_am_lane(req->send.ep);
     return status;
 }