#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 | ) |
|