MLX
 
Loading...
Searching...
No Matches
lapack.h File Reference
#include <cblas.h>
#include <lapack.h>

Go to the source code of this file.

Macros

#define MLX_LAPACK_FUNC(f)
 
#define INSTANTIATE_LAPACK_TYPES(FUNC)
 

Functions

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)
 

Macro Definition Documentation

◆ 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)
Value:
f##_

Function Documentation

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