fix a bug

8 years ago · f23e800f67
parent e22aa12b52
commit f23e800f67
1 changed files with 1 additions and 0 deletions
--- a/paddle/math/MathFunctions.h
+++ b/paddle/math/MathFunctions.h
@ -36,6 +36,7 @@ extern "C" {
 extern "C" {
 #include <cblas.h>    
 int LAPACKE_sgetrf(int matrix_layout, int m, int n,
+                   float* a, int lda, int* ipiv);
 int LAPACKE_dgetrf(int matrix_layout, int m, int n,
                   double* a, int lda, int* ipiv);
 int LAPACKE_sgetri(int matrix_layout, int n, float* a,