From fd2bf5ef3d255e760a78f3f9843b5c5e4c7e368b Mon Sep 17 00:00:00 2001 From: fehrmann <henning.fehrmann@aei.mpg.de> Date: Fri, 29 Jan 2021 10:06:22 +0100 Subject: [PATCH] added some FFTW handling --- hardware_settings.h | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/hardware_settings.h b/hardware_settings.h index b311022..7486d0a 100644 --- a/hardware_settings.h +++ b/hardware_settings.h @@ -32,6 +32,8 @@ #define __PREFIX(c) hip##c #define __CREATE_HANDLE(h) rocblas_create_handle(h) #define __DESTROY_HANDLE(h) rocblas_destroy_handle(h) +#define __DESTROY_PLAN(p) rocfft_plan_destroy(p) +#define __FFTW_PLAN rocfft_plan #endif #ifdef CUDA @@ -49,4 +51,6 @@ #define __PREFIX(c) cuda##c #define __CREATE_HANDLE(h) cublasCreate(h) #define __DESTROY_HANDLE(h) cublasDestroy(h) +#define __DESTROY_PLAN(p) cufftDestroy(p) +#define __FFTW_PLAN cufftHandle #endif -- GitLab