File tree Expand file tree Collapse file tree 6 files changed +8
-10
lines changed
source/module_basis/module_pw/module_fft Expand file tree Collapse file tree 6 files changed +8
-10
lines changed Original file line number Diff line number Diff line change 11#ifndef FFT_BASE_H
22#define FFT_BASE_H
3+
34#include < complex>
4- #include < string>
55namespace ModulePW
66{
77template <typename FPTYPE>
Original file line number Diff line number Diff line change 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)
Original file line number Diff line number Diff line change 33
44#include < memory>
55#include " fft_base.h"
6-
6+ # include " fft_cpu.h "
77namespace ModulePW
88{
99class FFT_Bundle
Original file line number Diff line number Diff line change 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"
106namespace ModulePW
117{
128template <typename FPTYPE>
Original file line number Diff line number Diff line change 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"
Original file line number Diff line number Diff line change 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>
You can’t perform that action at this time.
0 commit comments