Fix run without distributed

2025-07-23 10:02:12 +08:00 · 2024-09-06 11:02:29 -07:00 · 2024-09-06 11:02:29 -07:00 · 2e267bd6a8
commit 2e267bd6a8
parent a9746587f1
1 changed files with 1 additions and 1 deletions
--- a/mlx/distributed/sockets/sockets.cpp
+++ b/mlx/distributed/sockets/sockets.cpp
@ -168,7 +168,7 @@ struct GroupImpl {
    int success;

    // If we are expecting anyone to connect to us
-    if (rank_ < peers.size() - 1) {
+    if (rank_ + 1 < peers.size()) {
      // Create the socket to wait for connections from the peers
      int sock = socket(AF_INET, SOCK_STREAM, 0);
      if (sock < 0) {