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

2025-09-01 04:24:36 +08:00 · 2024-12-08 11:26:06 +09:00
parent 96986fb362
commit 9635cffdc8
1 changed files with 8 additions and 3 deletions
--- a/mlx/io/load.h
+++ b/mlx/io/load.h
@@ -2,12 +2,17 @@

 #pragma once

-#include <fcntl.h>
-#include <sys/stat.h>
-#include <unistd.h>
 #include <memory>
 #include <sstream>

+#include <fcntl.h>
+#ifdef _MSC_VER
+#include <io.h>
+#else
+#include <sys/stat.h>
+#include <unistd.h>
+#endif
+
 #include "mlx/io/threadpool.h"

 namespace mlx::core {