PaddlePaddle
diff --git a/‎paddle/phi/backends/dynload/cusolver.h
Copy file name to clipboard
+9-1Lines changed: 9 additions & 1 deletion b/‎paddle/phi/backends/dynload/cusolver.h
Copy file name to clipboard
+9-1Lines changed: 9 additions & 1 deletion
diff --git a/‎paddle/phi/backends/dynload/lapack.h
Copy file name to clipboard
+60-29Lines changed: 60 additions & 29 deletions b/‎paddle/phi/backends/dynload/lapack.h
Copy file name to clipboard
+60-29Lines changed: 60 additions & 29 deletions
diff --git a/‎paddle/phi/kernels/cpu/matrix_rank_kernel.cc
Copy file name to clipboard
+8-2Lines changed: 8 additions & 2 deletions b/‎paddle/phi/kernels/cpu/matrix_rank_kernel.cc
Copy file name to clipboard
+8-2Lines changed: 8 additions & 2 deletions
@@ -69,7 +69,6 @@ CUSOLVER_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUSOLVER_WRAP);
   __macro(cusolverDnDpotrfBatched);       \
   __macro(cusolverDnSpotrsBatched);       \
   __macro(cusolverDnDpotrsBatched);       \
-  __macro(cusolverDnSgesvdj_bufferSize);  \
   __macro(cusolverDnSgetrf_bufferSize);   \
   __macro(cusolverDnDgetrf_bufferSize);   \
   __macro(cusolverDnCgetrf_bufferSize);   \
@@ -86,9 +85,14 @@ CUSOLVER_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUSOLVER_WRAP);
   __macro(cusolverDnZungqr_bufferSize);   \
   __macro(cusolverDnDestroyGesvdjInfo);   \
   __macro(cusolverDnCreateGesvdjInfo);    \
+  __macro(cusolverDnSgesvdj_bufferSize);  \
   __macro(cusolverDnDgesvdj_bufferSize);  \
+  __macro(cusolverDnCgesvdj_bufferSize);  \
+  __macro(cusolverDnZgesvdj_bufferSize);  \
   __macro(cusolverDnSgesvdj);             \
   __macro(cusolverDnDgesvdj);             \
+  __macro(cusolverDnCgesvdj);             \
+  __macro(cusolverDnZgesvdj);             \
   __macro(cusolverDnSgetrf);              \
   __macro(cusolverDnSgetrs);              \
   __macro(cusolverDnDgetrs);              \
@@ -114,8 +118,12 @@ CUSOLVER_ROUTINE_EACH_R1(DECLARE_DYNAMIC_LOAD_CUSOLVER_WRAP)
   __macro(cusolverDnCreateSyevjInfo);          \
   __macro(cusolverDnSsyevj_bufferSize);        \
   __macro(cusolverDnDsyevj_bufferSize);        \
+  __macro(cusolverDnCheevj_bufferSize);        \
+  __macro(cusolverDnZheevj_bufferSize);        \
   __macro(cusolverDnSsyevj);                   \
   __macro(cusolverDnDsyevj);                   \
+  __macro(cusolverDnCheevj);                   \
+  __macro(cusolverDnZheevj);                   \
   __macro(cusolverDnDestroySyevjInfo);         \
   __macro(cusolverDnXsyevjSetSortEig);         \
   __macro(cusolverDnSsyevjBatched_bufferSize); \
 
@@ -299,35 +299,64 @@ extern "C" void spotrs_(char *uplo,
                         float *b,
                         int *ldb,
                         int *info);
-extern "C" void dgesdd_(char *,
-                        int *,
-                        int *,
-                        double *,
-                        int *,
-                        double *,
-                        double *,
-                        int *,
-                        double *,
-                        int *,
-                        double *,
-                        int *,
-                        int *,
-                        int *);
-extern "C" void sgesdd_(char *,
-                        int *,
-                        int *,
-                        float *,
-                        int *,
-                        float *,
-                        float *,
-                        int *,
-                        float *,
-                        int *,
-                        float *,
-                        int *,
-                        int *,
-                        int *);
-
+extern "C" void dgesdd_(char *jobz,
+                        int *m,
+                        int *n,
+                        double *a,
+                        int *lda,
+                        double *s,
+                        double *u,
+                        int *ldu,
+                        double *vt,
+                        int *ldvt,
+                        double *work,
+                        int *lwork,
+                        int *iwork,
+                        int *info);
+extern "C" void sgesdd_(char *jobz,
+                        int *m,
+                        int *n,
+                        float *a,
+                        int *lda,
+                        float *s,
+                        float *u,
+                        int *ldu,
+                        float *vt,
+                        int *ldvt,
+                        float *work,
+                        int *lwork,
+                        int *iwork,
+                        int *info);
+extern "C" void zgesdd_(char *jobz,
+                        int *m,
+                        int *n,
+                        std::complex<double> *a,
+                        int *lda,
+                        std::complex<double> *s,
+                        std::complex<double> *u,
+                        int *ldu,
+                        std::complex<double> *vt,
+                        int *ldvt,
+                        std::complex<double> *work,
+                        int *lwork,
+                        double *rwork,
+                        int *iwork,
+                        int *info);
+extern "C" void cgesdd_(char *jobz,
+                        int *m,
+                        int *n,
+                        std::complex<float> *a,
+                        int *lda,
+                        std::complex<float> *s,
+                        std::complex<float> *u,
+                        int *ldu,
+                        std::complex<float> *vt,
+                        int *ldvt,
+                        std::complex<float> *work,
+                        int *lwork,
+                        float *rwork,
+                        int *iwork,
+                        int *info);
 namespace phi {
 namespace dynload {
 
@@ -379,6 +408,8 @@ extern void *lapack_dso_handle;
   __macro(sgelss_);                  \
   __macro(sgesdd_);                  \
   __macro(dgesdd_);                  \
+  __macro(cgesdd_);                  \
+  __macro(zgesdd_);                  \
   __macro(zpotrs_);                  \
   __macro(cpotrs_);                  \
   __macro(dpotrs_);                  \
 
@@ -39,5 +39,11 @@ void MatrixRankKernel(const Context& dev_ctx,
 
 }  // namespace phi
 
-PD_REGISTER_KERNEL(
-    matrix_rank, CPU, ALL_LAYOUT, phi::MatrixRankKernel, float, double) {}
+PD_REGISTER_KERNEL(matrix_rank,
+                   CPU,
+                   ALL_LAYOUT,
+                   phi::MatrixRankKernel,
+                   float,
+                   double,
+                   phi::dtype::complex<float>,
+                   phi::dtype::complex<double>) {}