提交 e22aa12b 编写于 作者: L liaogang

fix a bug

上级 e2644fff
...@@ -32,14 +32,9 @@ extern "C" { ...@@ -32,14 +32,9 @@ extern "C" {
#include <lapacke.h> #include <lapacke.h>
#endif #endif
#ifdef PADDLE_USE_REFERENCE_CBLAS
extern "C" {
#include <cblas.h>
}
#endif
#ifndef LAPACK_FOUND #ifndef LAPACK_FOUND
extern "C" { extern "C" {
#include <cblas.h>
int LAPACKE_sgetrf(int matrix_layout, int m, int n, int LAPACKE_sgetrf(int matrix_layout, int m, int n,
int LAPACKE_dgetrf(int matrix_layout, int m, int n, int LAPACKE_dgetrf(int matrix_layout, int m, int n,
double* a, int lda, int* ipiv); double* a, int lda, int* ipiv);
...@@ -47,7 +42,7 @@ int LAPACKE_sgetri(int matrix_layout, int n, float* a, ...@@ -47,7 +42,7 @@ int LAPACKE_sgetri(int matrix_layout, int n, float* a,
int lda, const int* ipiv); int lda, const int* ipiv);
int LAPACKE_dgetri(int matrix_layout, int n, double* a, int LAPACKE_dgetri(int matrix_layout, int n, double* a,
int lda, const int* ipiv); int lda, const int* ipiv);
}; }
#endif #endif
#include <cmath> #include <cmath>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册