fix flush for apple rdma

This commit is contained in:
Aman Gupta
2026-09-23 20:20:21 +03:00
committed by Georgi Gerganov
parent 2543aa55a2
commit 1dd2a1cf90
+2 -2
View File
@@ -2238,13 +2238,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
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)) {
return false;
}
} else {
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;
}
}