fix flush for apple rdma

This commit is contained in:
Aman Gupta
2026-09-01 20:49:22 +08:00
parent 28bd87b2cd
commit 7d04ce4149
+2 -2
View File
@@ -2209,13 +2209,13 @@ bool rpc_server::comm_allreduce(const rpc_msg_comm_allreduce_req & request) {
// rank 0 sends first, rank 1 receives first, so large payloads cannot deadlock // rank 0 sends first, rank 1 receives first, so large payloads cannot deadlock
if (state.rank == 0) { if (state.rank == 0) {
if (!state.peer->send_data(state.send_buf.data(), wire_bytes) || if (!state.peer->send_data(state.send_buf.data(), wire_bytes) || !state.peer->flush() ||
!state.peer->recv_data(state.recv_buf.data(), wire_bytes)) { !state.peer->recv_data(state.recv_buf.data(), wire_bytes)) {
return false; return false;
} }
} else { } else {
if (!state.peer->recv_data(state.recv_buf.data(), wire_bytes) || if (!state.peer->recv_data(state.recv_buf.data(), wire_bytes) ||
!state.peer->send_data(state.send_buf.data(), wire_bytes)) { !state.peer->send_data(state.send_buf.data(), wire_bytes) || !state.peer->flush()) {
return false; return false;
} }
} }