diff --git a/lib/nodes/infiniband.c b/lib/nodes/infiniband.c index 91e0b1fbf..c5f6d76e0 100644 --- a/lib/nodes/infiniband.c +++ b/lib/nodes/infiniband.c @@ -711,7 +711,7 @@ int ib_stop(struct node *n) // Call RDMA disconnect function // Will flush all outstanding WRs to the Completion Queue and // will call RDMA_CM_EVENT_DISCONNECTED if that is done. - if (n->state == STATE_CONNECTED && ib->conn.port_space == RDMA_PS_TCP) { + if (n->state == STATE_CONNECTED && ib->conn.port_space != RDMA_PS_UDP) { ret = rdma_disconnect(ib->ctx.id); if (ret)