revert the call in rho_sym to fix SDFT failure

Author: maki49

source/module_elecstate/module_charge/symmetry_rho.cpp

@@ -127,8 +127,24 @@ void Symmetry_rho::psymm(double* rho_part,
     }
 
     // (3)
-Pgrid.bcast(rhotot.data(), rho_part);
-
+    const int ncxy = rho_basis->nx * rho_basis->ny;
+    std::vector<double> zpiece(ncxy);
+    for (int iz = 0; iz < rho_basis->nz; iz++)
+    {
+        ModuleBase::GlobalFunc::ZEROS(zpiece.data(), ncxy);
+        if (GlobalV::MY_RANK == 0)
+        {
+            for (int ix = 0; ix < rho_basis->nx; ix++)
+            {
+                for (int iy = 0; iy < rho_basis->ny; iy++)
+                {
+                    const int ir = ix * rho_basis->ny + iy;
+                    zpiece[ir] = rhotot[ix * rho_basis->ny * rho_basis->nz + iy * rho_basis->nz + iz];
+                }
+            }
+        }
+        Pgrid.zpiece_to_all(zpiece.data(), iz, rho_part);
+    }
 #endif
     return;
 }

source/module_hamilt_pw/hamilt_pwdft/parallel_grid.h

@@ -27,8 +27,8 @@ class Parallel_Grid
 		const int &nczp, const int &nrxx, const int &nbz, const int &bz); //LiuXh add 20180606
 
 #ifdef __MPI	
-    void zpiece_to_stogroup(double* zpiece, const int& iz, double* rho) const; //qainrui add for sto-dft 2021-7-21
     void zpiece_to_all(double* zpiece, const int& iz, double* rho) const;
+    void zpiece_to_stogroup(double* zpiece, const int& iz, double* rho) const; //qainrui add for sto-dft 2021-7-21
 
     /// @brief  Broadcast data from root to all processors. The index order is [x][y][z].
     void bcast(const double* const data_global, double* data_local)const;