Remove finalize code out

Author: Critsium-xy

source/source_esolver/esolver_ks_pw.cpp

@@ -71,21 +71,6 @@ ESolver_KS_PW<T, Device>::~ESolver_KS_PW()
         this->pelec = nullptr;
     }
 
-    if (this->device == base_device::GpuDevice)
-    {
-#if defined(__CUDA) || defined(__ROCM)
-        ModuleBase::destoryBLAShandle();
-        hsolver::destroyGpuSolverHandle();
-        container::kernels::destroyGpuBlasHandle();
-        container::kernels::destroyGpuSolverHandle();
-#endif
-    }
-
-#ifdef __DSP
-    std::cout << " ** Closing DSP Hardware..." << std::endl;
-    mtfunc::dspDestoryHandle(GlobalV::MY_RANK);
-#endif
-
     if (PARAM.inp.device == "gpu" || PARAM.inp.precision == "single")
     {
         delete this->kspw_psi;

source/source_main/driver.h

@@ -40,6 +40,7 @@ class Driver
 
     // Init harewares according to Input parameters
     void init_hardware();
+    void finalize_hardware();
 };
 
 #endif

source/source_main/driver_run.cpp

@@ -106,14 +106,16 @@ void Driver::driver_run()
     p_esolver->after_all_runners(ucell);
 
     ModuleESolver::clean_esolver(p_esolver);
+    this->finalize_hardware();
 
     //! 6: output the json file
     Json::create_Json(&ucell, PARAM);
 
     return;
 }
 
-void Driver::init_hardware(){
+void Driver::init_hardware()
+{
 #if ((defined __CUDA) || (defined __ROCM))
     if (PARAM.inp.device == "gpu")
     {
@@ -129,3 +131,21 @@ void Driver::init_hardware(){
     mtfunc::dspInitHandle(GlobalV::MY_RANK);
 #endif
 }
+
+void Driver::finalize_hardware()
+{
+#if defined(__CUDA) || defined(__ROCM)
+    if (PARAM.inp.device == "gpu")
+    {
+        ModuleBase::destoryBLAShandle();
+        hsolver::destroyGpuSolverHandle();
+        container::kernels::destroyGpuBlasHandle();
+        container::kernels::destroyGpuSolverHandle();
+    }
+#endif
+
+#ifdef __DSP
+    std::cout << " ** Closing DSP Hardware..." << std::endl;
+    mtfunc::dspDestoryHandle(GlobalV::MY_RANK);
+#endif
+}