diff --git a/paddle/math/MathFunctions.h b/paddle/math/MathFunctions.h index 8c5f9dd479c9613075fdafff1b2e64a7cce93401..a941c548d43b12c87b3d53ec77d1819f17747744 100644 --- a/paddle/math/MathFunctions.h +++ b/paddle/math/MathFunctions.h @@ -36,6 +36,7 @@ extern "C" { extern "C" { #include 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,