fix compile in sdft

Author: Qianruipku

source/module_base/math_chebyshev.cpp

@@ -575,6 +575,7 @@ bool Chebyshev<REAL>::checkconverge(
     std::function<void(std::complex<REAL>* in, std::complex<REAL>* out, const int)> funA,
     std::complex<REAL>* wavein,
     const int N,
+    const int LDA,
     REAL& tmax,
     REAL& tmin,
     REAL stept)
@@ -584,9 +585,9 @@ bool Chebyshev<REAL>::checkconverge(
     std::complex<REAL>* arrayn;
     std::complex<REAL>* arrayn_1;
 
-    arraynp1 = new std::complex<REAL>[N];
-    arrayn = new std::complex<REAL>[N];
-    arrayn_1 = new std::complex<REAL>[N];
+    arraynp1 = new std::complex<REAL>[LDA];
+    arrayn = new std::complex<REAL>[LDA];
+    arrayn_1 = new std::complex<REAL>[LDA];
 
     ModuleBase::GlobalFunc::DCOPY(wavein, arrayn_1, N);
 

source/module_base/math_chebyshev.h

@@ -191,6 +191,7 @@ class Chebyshev
     bool checkconverge(std::function<void(std::complex<REAL>* in, std::complex<REAL>* out, const int)> funA,
                        std::complex<REAL>* wavein,
                        const int N,
+                       const int LDA,
                        REAL& tmax,  // trial number for upper bound
                        REAL& tmin,  // trial number for lower bound
                        REAL stept); // tmax = max() + stept, tmin = min() - stept

source/module_base/test/math_chebyshev_test.cpp

@@ -346,15 +346,15 @@ TEST_F(MathChebyshevTest, checkconverge)
     double tmin = -1.1;
     double tmax = 1.1;
     bool converge;
-    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, tmax, tmin, 0.2);
+    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, 2, tmax, tmin, 0.2);
     EXPECT_TRUE(converge);
-    converge = p_chetest->checkconverge(fun_sigma_y, v + 2, 2, tmax, tmin, 0.2);
+    converge = p_chetest->checkconverge(fun_sigma_y, v + 2, 2, 2, tmax, tmin, 0.2);
     EXPECT_TRUE(converge);
     EXPECT_NEAR(tmin, -1.1, 1e-8);
     EXPECT_NEAR(tmax, 1.1, 1e-8);
 
     tmax = -1.1;
-    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, tmax, tmin, 2.2);
+    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, 2, tmax, tmin, 2.2);
     EXPECT_TRUE(converge);
     EXPECT_NEAR(tmin, -1.1, 1e-8);
     EXPECT_NEAR(tmax, 1.1, 1e-8);
@@ -363,12 +363,12 @@ TEST_F(MathChebyshevTest, checkconverge)
     v[0] = std::complex<double>(0, 1), v[1] = 1;
     fun.factor = 1.5;
     tmin = -1.1, tmax = 1.1;
-    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, tmax, tmin, 0.2);
+    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, 2, tmax, tmin, 0.2);
     EXPECT_FALSE(converge);
 
     fun.factor = -1.5;
     tmin = -1.1, tmax = 1.1;
-    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, tmax, tmin, 0.2);
+    converge = p_chetest->checkconverge(fun_sigma_y, v, 2, 2, tmax, tmin, 0.2);
     EXPECT_FALSE(converge);
     fun.factor = 1;
 
@@ -632,9 +632,9 @@ TEST_F(MathChebyshevTest, checkconverge_float)
 
     auto fun_sigma_yf
         = [&](std::complex<float>* in, std::complex<float>* out, const int m = 1) { fun.sigma_y(in, out, m); };
-    converge = p_fchetest->checkconverge(fun_sigma_yf, v, 2, tmax, tmin, 0.2);
+    converge = p_fchetest->checkconverge(fun_sigma_yf, v, 2, 2, tmax, tmin, 0.2);
     EXPECT_TRUE(converge);
-    converge = p_fchetest->checkconverge(fun_sigma_yf, v + 2, 2, tmax, tmin, 0.2);
+    converge = p_fchetest->checkconverge(fun_sigma_yf, v + 2, 2, 2, tmax, tmin, 0.2);
     EXPECT_TRUE(converge);
     EXPECT_NEAR(tmin, -1.1, 1e-6);
     EXPECT_NEAR(tmax, 1.1, 1e-6);

source/module_hamilt_pw/hamilt_stodft/sto_iter.cpp

@@ -140,7 +140,7 @@ void Stochastic_Iter::checkemm(const int& ik, const int istep, const int iter, S
                                        std::placeholders::_1,
                                        std::placeholders::_2,
                                        std::placeholders::_3);
-            converge = p_che->checkconverge(hchi_norm, pchi, npw, *p_hamilt_sto->emax, *p_hamilt_sto->emin, 5.0);
+            converge = p_che->checkconverge(hchi_norm, pchi, npw, stowf.npwx, *p_hamilt_sto->emax, *p_hamilt_sto->emin, 5.0);
 
             if (!converge)
             {

source/module_hamilt_pw/hamilt_stodft/sto_tool.cpp

@@ -74,7 +74,7 @@ void check_che(const int& nche_in,
                                            std::placeholders::_1,
                                            std::placeholders::_2,
                                            std::placeholders::_3);
-                converge = chetest.checkconverge(hchi_norm, pchi, npw, *p_hamilt_sto->emax, *p_hamilt_sto->emin, 2.0);
+                converge = chetest.checkconverge(hchi_norm, pchi, npw, p_stowf->npwx, *p_hamilt_sto->emax, *p_hamilt_sto->emin, 2.0);
 
                 if (!converge)
                 {