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 | 9 +++++++--
 1 file changed, 7 insertions(+), 2 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,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"