mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-09-27 21:46:57 +02:00
fix flush for apple rdma
This commit is contained in:
committed by
Georgi Gerganov
parent
2543aa55a2
commit
1dd2a1cf90
@@ -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;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user