zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-01 00:28:11 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Use correct complex type for MSVC (#1660)

Browse Source
This commit is contained in:
Cheng
2024-12-08 11:25:22 +09:00
committed by GitHub
parent 50fa705125
commit 3ceb341a75
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mlx/backend/common/lapack.h
View File

@@ -2,6 +2,15 @@
#pragma once #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 #ifdef ACCELERATE_NEW_LAPACK
#include <Accelerate/Accelerate.h> #include <Accelerate/Accelerate.h>
#else #else