feature/design_of_v2_layer_converter
liaogang 9 years ago
parent e2644fffc5
commit e22aa12b52

@ -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>

Loading…
Cancel
Save