Move hardware initializer out from esolver

Author: Critsium-xy

source/source_esolver/esolver_ks_pw.cpp

@@ -58,18 +58,18 @@ ESolver_KS_PW<T, Device>::ESolver_KS_PW()
     this->device = base_device::get_device_type<Device>(this->ctx);
 
 #if ((defined __CUDA) || (defined __ROCM))
-    if (this->device == base_device::GpuDevice)
+    /*if (this->device == base_device::GpuDevice)
     {
         ModuleBase::createGpuBlasHandle();
         hsolver::createGpuSolverHandle();
         container::kernels::createGpuBlasHandle();
         container::kernels::createGpuSolverHandle();
-    }
+    }*/
 #endif
 
 #ifdef __DSP
-    std::cout << " ** Initializing DSP Hardware..." << std::endl;
-    mtfunc::dspInitHandle(GlobalV::MY_RANK);
+    // std::cout << " ** Initializing DSP Hardware..." << std::endl;
+    // mtfunc::dspInitHandle(GlobalV::MY_RANK);
 #endif
 }
 

source/source_main/driver.h

@@ -37,6 +37,9 @@ class Driver
 
     // the actual calculations
     void driver_run();
+
+    // Init harewares according to Input parameters
+    void init_hardware();
 };
 
 #endif

source/source_main/driver_run.cpp

@@ -6,6 +6,17 @@
 #include "source_io/para_json.h"
 #include "source_io/print_info.h"
 #include "source_md/run_md.h"
+#include "source_base/module_device/device.h"
+#include "source_base/module_device/memory_op.h"
+#include "source_base/kernels/math_kernel_op.h"
+#include "source_hsolver/kernels/dngvd_op.h"
+
+#include <ATen/kernels/blas.h>
+#include <ATen/kernels/lapack.h>
+
+#ifdef __DSP
+#include "source_base/kernels/dsp/dsp_connector.h"
+#endif
 
 /**
  * @brief This is the driver function which defines the workflow of ABACUS
@@ -47,6 +58,8 @@ void Driver::driver_run()
     unitcell::check_atomic_stru(ucell, PARAM.inp.min_dist_coef);
 
     //! 2: initialize the ESolver (depends on a set-up ucell after `setup_cell`)
+    this->init_hardware();
+
     ModuleESolver::ESolver* p_esolver = ModuleESolver::init_esolver(PARAM.inp, ucell);
 
     //! 3: initialize Esolver and fill json-structure
@@ -99,3 +112,20 @@ void Driver::driver_run()
 
     return;
 }
+
+void Driver::init_hardware(){
+#if ((defined __CUDA) || (defined __ROCM))
+    if (PARAM.inp.device == "gpu")
+    {
+        ModuleBase::createGpuBlasHandle();
+        hsolver::createGpuSolverHandle();
+        container::kernels::createGpuBlasHandle();
+        container::kernels::createGpuSolverHandle();
+    }
+#endif
+
+#ifdef __DSP
+    std::cout << " ** Initializing DSP Hardware..." << std::endl;
+    mtfunc::dspInitHandle(GlobalV::MY_RANK);
+#endif
+}