zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 09:51:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Change the send message size

Browse Source
This commit is contained in:
Angelos Katharopoulos 2024-08-29 02:26:15 -07:00
parent e5354fcddb
commit e94f0028c3
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mlx/distributed/sockets/sockets.cpp
View File

@ -170,10 +170,17 @@ struct GroupImpl {
}
void send(const char* buf, size_t len, int dst) {
ssize_t r = sendto(
socket_fd_, buf, len, 0, peers_[dst].sockaddr(), peers_[dst].len);
if (r < 0) {
throw std::runtime_error("Send failed.");
while (len > 0) {
size_t l = std::min(len, 8192ul);
ssize_t r = sendto(
socket_fd_, buf, l, 0, peers_[dst].sockaddr(), peers_[dst].len);
if (r <= 0) {
std::ostringstream msg;
msg << "Send of " << l << " bytes failed (errno: " << errno << ")";
throw std::runtime_error(msg.str());
}
len -= l;
buf += l;
}
}