aboutsummaryrefslogtreecommitdiff
path: root/Source/Utils/WarpXrocfftUtil.cpp
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--Source/Utils/WarpXrocfftUtil.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/Source/Utils/WarpXrocfftUtil.cpp b/Source/Utils/WarpXrocfftUtil.cpp
index d613ba8c6..ada12764d 100644
--- a/Source/Utils/WarpXrocfftUtil.cpp
+++ b/Source/Utils/WarpXrocfftUtil.cpp
@@ -13,7 +13,11 @@
// cstddef: work-around for ROCm/rocFFT <=4.3.0
// https://github.com/ROCmSoftwarePlatform/rocFFT/blob/rocm-4.3.0/library/include/rocfft.h#L36-L42
# include <cstddef>
-# include <rocfft/rocfft.h>
+# if __has_include(<rocfft/rocfft.h>) // ROCm 5.3+
+# include <rocfft/rocfft.h>
+# else
+# include <rocfft.h>
+# endif
#endif
void