diff options
Diffstat (limited to 'Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H')
| -rw-r--r-- | Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H b/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H index b89c0106e..0dff6da7d 100644 --- a/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H +++ b/Source/FieldSolver/SpectralSolver/SpectralFieldDataRZ.H @@ -24,8 +24,10 @@ class SpectralFieldDataRZ // Define the FFTplans type, which holds one fft plan per box // (plans are only initialized for the boxes that are owned by // the local MPI rank) -#ifdef AMREX_USE_GPU +#if defined(AMREX_USE_CUDA) using FFTplans = amrex::LayoutData<cufftHandle>; +#elif defined(AMREX_USE_HIP) + using FFTplans = amrex::LayoutData<rocfft_plan>; #else using FFTplans = amrex::LayoutData<fftw_plan>; #endif |