Use correct complex type for MSVC

ml-explore · Dec 7, 2024 · 16549d4 · 16549d4
1 parent 50fa705
commit 16549d4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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