From 9635cffdc8ce998f63df99e2d32584e4869fc526 Mon Sep 17 00:00:00 2001
From: Cheng <zcbenz@gmail.com>
Date: Sun, 8 Dec 2024 11:26:06 +0900
Subject: [PATCH] Include io.h in MSVC for IO functions (#1661)

---
 mlx/io/load.h | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/mlx/io/load.h b/mlx/io/load.h
index e75057f68..3a58aff59 100644
--- 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 {