Add headers for gcc

mlx/distributed/ibv/ibv.cpp

@@ -1079,6 +1079,8 @@ class IBVGroup : public GroupImpl {
 bool is_available() {
   if (__builtin_available(macOS 26.2, *)) {
     return true;
+  } else {
+    return false;
   }
 }
 

mlx/distributed/utils.cpp

@@ -2,6 +2,7 @@
 
 #include <netdb.h>
 #include <unistd.h>
+#include <cstring>
 #include <sstream>
 #include <thread>
 
@@ -14,7 +15,7 @@ namespace mlx::core::distributed::detail {
  */
 address_t parse_address(const std::string& ip, const std::string& port) {
   struct addrinfo hints, *res;
-  memset(&hints, 0, sizeof(hints));
+  std::memset(&hints, 0, sizeof(hints));
   hints.ai_family = AF_UNSPEC;
   hints.ai_socktype = SOCK_STREAM;
 

mlx/distributed/utils.h

@@ -3,6 +3,8 @@
 #pragma once
 
 #include <sys/socket.h>
+#include <functional>
+#include <string>
 
 namespace mlx::core::distributed::detail {