update the header file

A-006 · A-006 · commit f920328c19a0 · 2024-11-20T11:26:11.000+08:00
diff --git a/source/module_basis/module_pw/module_fft/fft_base.h b/source/module_basis/module_pw/module_fft/fft_base.h
@@ -3,6 +3,7 @@
 
 #include <complex>
 #include <string>
+#include "module_base/module_device/memory_op.h"
 namespace ModulePW
 {
 template <typename FPTYPE>
diff --git a/source/module_basis/module_pw/module_fft/fft_bundle.cpp b/source/module_basis/module_pw/module_fft/fft_bundle.cpp
@@ -1,7 +1,5 @@
 #include <cassert>
 #include "fft_bundle.h"
-#include "fft_cpu.h"
-#include "module_base/module_device/device.h"
 #if defined(__CUDA)
 #include "fft_cuda.h"
 #endif
diff --git a/source/module_basis/module_pw/module_fft/fft_bundle.h b/source/module_basis/module_pw/module_fft/fft_bundle.h
@@ -2,14 +2,11 @@
 #define FFT_TEMP_H
 
 #include "fft_base.h"
-#include <memory>
 #include "fft_cpu.h"
-#ifdef __CUDA
-#include "fft_cuda.h"
-#endif
-#ifdef __ROCM
-#include "fft_rocm.h"
-#endif
+#include "module_base/module_device/device.h"
+#include "module_base/module_device/memory_op.h"
+#include <memory>
+
 namespace ModulePW
 {
 class FFT_Bundle

Original file line number	Diff line number	Diff line change
`@@ -3,6 +3,7 @@`
`3`	`3`
`4`	`4`	`#include <complex>`
`5`	`5`	`#include <string>`
	`6`	`+#include "module_base/module_device/memory_op.h"`
`6`	`7`	`namespace ModulePW`
`7`	`8`	`{`
`8`	`9`	`template <typename FPTYPE>`