diff --git a/dev/unistd.h b/dev/unistd.h index 411757f..6ff68be 100644 --- a/dev/unistd.h +++ b/dev/unistd.h @@ -4,6 +4,7 @@ #define _CRT_SECURE_NO_WARNINGS #define _USE_MATH_DEFINES +#define WIN32_LEAN_AND_MEAN #include #include diff --git a/llmc/utils.h b/llmc/utils.h index 74a06d2..e09bdce 100644 --- a/llmc/utils.h +++ b/llmc/utils.h @@ -7,6 +7,7 @@ #ifndef UTILS_H #define UTILS_H +#include #include #include #include diff --git a/train_gpt2.cu b/train_gpt2.cu index 301cbea..7060923 100644 --- a/train_gpt2.cu +++ b/train_gpt2.cu @@ -1,11 +1,6 @@ /* GPT-2 Transformer Neural Net training loop. See README.md for usage. */ - -#ifdef _WIN32 -#define WIN32_LEAN_AND_MEAN -#endif - #include #include #include