Move socket header to utils.h

This commit is contained in:
Aleksa Gordic 2024-06-26 10:43:39 +00:00
parent 5645bd18b1
commit d27331cced
2 changed files with 1 additions and 6 deletions

View file

@ -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

View file

@ -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>