From 3ceb341a75716146c9cc0a7e7c52572b9413a219 Mon Sep 17 00:00:00 2001
From: Cheng <zcbenz@gmail.com>
Date: Sun, 8 Dec 2024 11:25:22 +0900
Subject: [PATCH] Use correct complex type for MSVC (#1660)

---
 mlx/backend/common/lapack.h | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/mlx/backend/common/lapack.h b/mlx/backend/common/lapack.h
index b3bb7ebf0..42d937785 100644
--- a/mlx/backend/common/lapack.h
+++ b/mlx/backend/common/lapack.h
@@ -2,6 +2,15 @@
 
 #pragma once
 
+// Required for Visual Studio.
+// https://github.com/OpenMathLib/OpenBLAS/blob/develop/docs/install.md
+#ifdef _MSC_VER
+#include <complex>
+#define LAPACK_COMPLEX_CUSTOM
+#define lapack_complex_float std::complex<float>
+#define lapack_complex_double std::complex<double>
+#endif
+
 #ifdef ACCELERATE_NEW_LAPACK
 #include <Accelerate/Accelerate.h>
 #else