#ifndef TH_GENERIC_FILE #define TH_GENERIC_FILE "generic/THLapack.h" #else /* AX=B */ TH_API void THLapack_(gesv)(int n, int nrhs, real *a, int lda, int *ipiv, real *b, int ldb, int* info); /* Solve a triangular system of the form A * X = B or A^T * X = B */ TH_API void THLapack_(trtrs)(char uplo, char trans, char diag, int n, int nrhs, real *a, int lda, real *b, int ldb, int* info); /* ||AX-B|| */ TH_API void THLapack_(gels)(char trans, int m, int n, int nrhs, real *a, int lda, real *b, int ldb, real *work, int lwork, int *info); /* Eigenvals */ TH_API void THLapack_(syev)(char jobz, char uplo, int n, real *a, int lda, real *w, real *work, int lwork, int *info); /* Non-sym eigenvals */ TH_API void THLapack_(geev)(char jobvl, char jobvr, int n, real *a, int lda, real *wr, real *wi, real* vl, int ldvl, real *vr, int ldvr, real *work, int lwork, int *info); /* svd */ TH_API void THLapack_(gesvd)(char jobu, char jobvt, int m, int n, real *a, int lda, real *s, real *u, int ldu, real *vt, int ldvt, real *work, int lwork, int *info); /* LU decomposition */ TH_API void THLapack_(getrf)(int m, int n, real *a, int lda, int *ipiv, int *info); TH_API void THLapack_(getrs)(char trans, int n, int nrhs, real *a, int lda, int *ipiv, real *b, int ldb, int *info); /* Matrix Inverse */ TH_API void THLapack_(getri)(int n, real *a, int lda, int *ipiv, real *work, int lwork, int* info); /* Positive Definite matrices */ /* Cholesky factorization */ void THLapack_(potrf)(char uplo, int n, real *a, int lda, int *info); /* Matrix inverse based on Cholesky factorization */ void THLapack_(potri)(char uplo, int n, real *a, int lda, int *info); /* Solve A*X = B with a symmetric positive definite matrix A using the Cholesky factorization */ void THLapack_(potrs)(char uplo, int n, int nrhs, real *a, int lda, real *b, int ldb, int *info); /* Cholesky factorization with complete pivoting. */ void THLapack_(pstrf)(char uplo, int n, real *a, int lda, int *piv, int *rank, real tol, real *work, int *info); /* QR decomposition */ void THLapack_(geqrf)(int m, int n, real *a, int lda, real *tau, real *work, int lwork, int *info); /* Build Q from output of geqrf */ void THLapack_(orgqr)(int m, int n, int k, real *a, int lda, real *tau, real *work, int lwork, int *info); /* Multiply Q with a matrix from output of geqrf */ void THLapack_(ormqr)(char side, char trans, int m, int n, int k, real *a, int lda, real *tau, real *c, int ldc, real *work, int lwork, int *info); #endif