Skip to content

Commit 44490fb

Browse files
committed
change fft_cpu.h
1 parent 224e3ae commit 44490fb

File tree

6 files changed

+8
-10
lines changed

6 files changed

+8
-10
lines changed

source/module_basis/module_pw/module_fft/fft_base.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#ifndef FFT_BASE_H
22
#define FFT_BASE_H
3+
34
#include <complex>
4-
#include <string>
55
namespace ModulePW
66
{
77
template <typename FPTYPE>

source/module_basis/module_pw/module_fft/fft_bundle.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#include <cassert>
22
#include "fft_bundle.h"
3-
#include "fft_cpu.h"
3+
44
#include "module_base/module_device/device.h"
55
#include "module_base/module_device/memory_op.h"
66
#if defined(__CUDA)

source/module_basis/module_pw/module_fft/fft_bundle.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
#include <memory>
55
#include "fft_base.h"
6-
6+
#include "fft_cpu.h"
77
namespace ModulePW
88
{
99
class FFT_Bundle

source/module_basis/module_pw/module_fft/fft_cpu.h

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,8 @@
1-
#include "fft_base.h"
2-
#include "fftw3.h"
3-
4-
// #ifdef __ENABLE_FLOAT_FFTW
5-
6-
// #endif
7-
// #endif
81
#ifndef FFT_CPU_H
92
#define FFT_CPU_H
3+
4+
#include "fft_base.h"
5+
#include "fftw3.h"
106
namespace ModulePW
117
{
128
template <typename FPTYPE>

source/module_basis/module_pw/module_fft/fft_cuda.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
#ifndef FFT_CUDA_H
22
#define FFT_CUDA_H
3+
34
#include "fft_base.h"
45
#include "cufft.h"
56
#include "cuda_runtime.h"

source/module_basis/module_pw/module_fft/fft_rcom.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11

22
#ifndef FFT_ROCM_H
33
#define FFT_ROCM_H
4+
45
#include "fft_base.h"
56
#include <hipfft/hipfft.h>
67
#include <hip/hip_runtime.h>

0 commit comments

Comments
 (0)