From d1c5b53a7e92ba987fed58a85a27a5d2e0feb5d7 Mon Sep 17 00:00:00 2001
From: Aleksa Gordic <gordicaleksa@gmail.com>
Date: Wed, 26 Jun 2024 10:43:39 +0000
Subject: [PATCH] Move socket header to utils.h

---
 llmc/utils.h  | 1 +
 llmc/zero.cuh | 6 ------
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/llmc/utils.h b/llmc/utils.h
index 6f33c40c6..4223e6bd7 100644
--- a/llmc/utils.h
+++ b/llmc/utils.h
@@ -15,6 +15,7 @@
 // implementation of dirent for Windows is in dev/unistd.h
 #ifndef _WIN32
 #include <dirent.h>
+#include <arpa/inet.h>
 #else
 #include <winsock2.h>
 #endif
diff --git a/llmc/zero.cuh b/llmc/zero.cuh
index fb0821a1c..2b19b1b81 100644
--- a/llmc/zero.cuh
+++ b/llmc/zero.cuh
@@ -5,12 +5,6 @@ Utilities for ZeRO sharding
 #ifndef LLMC_ZERO_CUH
 #define LLMC_ZERO_CUH
 
-#ifdef _WIN32
-#include <winsock2.h>
-#else
-#include <arpa/inet.h>
-#endif
-
 #include <cuda_runtime_api.h>
 #include <stdint.h>
 #include <stdlib.h>