Include io.h in MSVC for IO functions (#1661)

ml-explore · Dec 8, 2024 · 9635cff · 9635cff
1 parent 96986fb
commit 9635cff
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/mlx/io/load.h b/mlx/io/load.h
@@ -2,11 +2,16 @@
 
 #pragma once
 
+#include <memory>
+#include <sstream>
+
 #include <fcntl.h>
+#ifdef _MSC_VER
+#include <io.h>
+#else
 #include <sys/stat.h>
 #include <unistd.h>
-#include <memory>
-#include <sstream>
+#endif
 
 #include "mlx/io/threadpool.h"