update the header file

Author: A-006

source/module_basis/module_pw/module_fft/fft_bundle.cpp

@@ -1,6 +1,14 @@
 #include <cassert>
 #include "fft_bundle.h"
-
+#include "fft_cpu.h"
+#include "module_base/module_device/device.h"
+#include "module_base/module_device/memory_op.h"
+#if defined(__CUDA)
+#include "fft_cuda.h"
+#endif
+#if defined(__ROCM)
+#include "fft_rcom.h"
+#endif
 
 template<typename FFT_BASE, typename... Args>
 std::unique_ptr<FFT_BASE> make_unique(Args &&... args)

source/module_basis/module_pw/module_fft/fft_bundle.h

@@ -3,15 +3,6 @@
 
 #include <memory>
 #include "fft_base.h"
-#include "fft_cpu.h"
-#include "module_base/module_device/device.h"
-#include "module_base/module_device/memory_op.h"
-#if defined(__CUDA)
-#include "fft_cuda.h"
-#endif
-#if defined(__ROCM)
-#include "fft_rcom.h"
-#endif
 
 namespace ModulePW
 {

source/module_hamilt_lcao/module_gint/kernels/cuda/gemm_selector.cuh

@@ -2,7 +2,7 @@
 #define GEMM_SELECTOR_H
 
 #include "module_cell/unitcell.h"
-
+#include "cuda_runtime.h"
 typedef std::function<void(int,
                            int,
                            int*,

source/module_hamilt_pw/hamilt_pwdft/global.h

@@ -15,6 +15,8 @@
 #include "module_hamilt_general/module_xc/xc_functional.h"
 #ifdef __CUDA
 #include "cublas_v2.h"
+#include "cufft.h"
+
 static const char* _cublasGetErrorString(cublasStatus_t error)
 {
     switch (error)