#include <cblas.h>
#include <lapack.h>
Go to the source code of this file.
|
| template<typename T, typename... Args> |
| void | geqrf (Args... args) |
| |
| template<typename T, typename... Args> |
| void | orgqr (Args... args) |
| |
| template<typename T, typename... Args> |
| void | syevd (Args... args) |
| |
| template<typename T, typename... Args> |
| void | potrf (Args... args) |
| |
| template<typename T, typename... Args> |
| void | gesvdx (Args... args) |
| |
| template<typename T, typename... Args> |
| void | getrf (Args... args) |
| |
| template<typename T, typename... Args> |
| void | getri (Args... args) |
| |
| template<typename T, typename... Args> |
| void | trtri (Args... args) |
| |
◆ INSTANTIATE_LAPACK_TYPES
| #define INSTANTIATE_LAPACK_TYPES |
( |
| FUNC | ) |
|
Value: template <typename T, typename... Args> \
void FUNC(Args... args) { \
if constexpr (std::is_same_v<T, float>) { \
MLX_LAPACK_FUNC(s##FUNC)(std::forward<Args>(args)...); \
} else if constexpr (std::is_same_v<T, double>) { \
MLX_LAPACK_FUNC(d##FUNC)(std::forward<Args>(args)...); \
} \
}
◆ MLX_LAPACK_FUNC
| #define MLX_LAPACK_FUNC |
( |
| f | ) |
|
◆ geqrf()
template<typename T, typename... Args>
| void geqrf |
( |
Args... | args | ) |
|
◆ gesvdx()
template<typename T, typename... Args>
| void gesvdx |
( |
Args... | args | ) |
|
◆ getrf()
template<typename T, typename... Args>
| void getrf |
( |
Args... | args | ) |
|
◆ getri()
template<typename T, typename... Args>
| void getri |
( |
Args... | args | ) |
|
◆ orgqr()
template<typename T, typename... Args>
| void orgqr |
( |
Args... | args | ) |
|
◆ potrf()
template<typename T, typename... Args>
| void potrf |
( |
Args... | args | ) |
|
◆ syevd()
template<typename T, typename... Args>
| void syevd |
( |
Args... | args | ) |
|
◆ trtri()
template<typename T, typename... Args>
| void trtri |
( |
Args... | args | ) |
|