|
|
|
|
@ -32,14 +32,9 @@ extern "C" {
|
|
|
|
|
#include <lapacke.h>
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
#ifdef PADDLE_USE_REFERENCE_CBLAS
|
|
|
|
|
extern "C" {
|
|
|
|
|
#include <cblas.h>
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
#ifndef LAPACK_FOUND
|
|
|
|
|
extern "C" {
|
|
|
|
|
#include <cblas.h>
|
|
|
|
|
int LAPACKE_sgetrf(int matrix_layout, int m, int n,
|
|
|
|
|
int LAPACKE_dgetrf(int matrix_layout, int m, int n,
|
|
|
|
|
double* a, int lda, int* ipiv);
|
|
|
|
|
@ -47,7 +42,7 @@ int LAPACKE_sgetri(int matrix_layout, int n, float* a,
|
|
|
|
|
int lda, const int* ipiv);
|
|
|
|
|
int LAPACKE_dgetri(int matrix_layout, int n, double* a,
|
|
|
|
|
int lda, const int* ipiv);
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
#include <cmath>
|
|
|
|
|
|