Merge pull request #67 from evgeny-leksikov/int3_ka_fix

UCT/RC/MLX5: fix keep alive protocol
yosefe · Oct 22, 2020 · 4830be9 · 4830be9
2 parents 2761424 + 7d0c740
commit 4830be9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/uct/ib/rc/accel/rc_mlx5_ep.c b/src/uct/ib/rc/accel/rc_mlx5_ep.c
@@ -579,6 +579,11 @@ ucs_status_t uct_rc_mlx5_ep_fc_ctrl(uct_ep_t *tl_ep, unsigned op,
      * messages are bundled with AM. */
     ucs_assert(op == UCT_RC_EP_FC_PURE_GRANT);
 
+    if (ucs_unlikely(ep->tx.wq.super.verbs.qp->state != IBV_QPS_RTS)) {
+        return (ep->tx.wq.super.verbs.qp->state == IBV_QPS_INIT) ? UCS_OK :
+               UCS_ERR_CONNECTION_RESET;
+    }
+
     UCT_RC_CHECK_RES(&iface->super, &ep->super);
     uct_rc_mlx5_txqp_inline_post(iface, IBV_QPT_RC,
                                  &ep->super.txqp, &ep->tx.wq,