fix hipcc

correaa · Mar 8, 2025 · 1a53378 · 1a53378
1 parent 812110b
commit 1a53378
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/include/boost/multi/adaptors/fft.hpp b/include/boost/multi/adaptors/fft.hpp
@@ -25,21 +25,21 @@ namespace boost::multi::fft{
 	template<std::size_t I> struct priority : std::conditional_t<I==0, std::true_type, struct priority<I-1>>{};
 
 	template<class... Args> auto dft_aux(priority<0> /*unused*/, Args&&... args) BOOST_MULTI_DECLRETURN_(                  fftw::dft(std::forward<Args>(args)...))
-	#if defined(__CUDA__) || defined(__NVCC__)
+	#if defined(__CUDA__) || defined(__NVCC__) || defined(__HIPCC__)
 	template<class... Args> auto dft_aux(priority<1> /*unused*/, Args&&... args) BOOST_MULTI_DECLRETURN_(::boost::multi::cufft ::dft(std::forward<Args>(args)...))
 	#endif
 	template<          class... Args> auto dft(Args&&... args) BOOST_MULTI_DECLRETURN_(dft_aux_(priority<1>{}, std::forward<Args>(args)...))
 
 	template<class In, class... Args> auto dft(std::array<bool, std::decay_t<In>::dimensionality> which, In&& in, Args&&... args) BOOST_MULTI_DECLRETURN_(dft_aux(priority<1>{}, which, std::forward<In>(in), std::forward<Args>(args)...))
 
 	template<class... Args> auto dft_forward_aux(priority<0> /*unused*/, Args&&... args) BOOST_MULTI_DECLRETURN_(  fftw::dft_forward(std::forward<Args>(args)...))
-	#if defined(__CUDA__) || defined(__NVCC__)
+	#if defined(__CUDA__) || defined(__NVCC__) || defined(__HIPCC__)
 	template<class... Args> auto dft_forward_aux(priority<1> /*unused*/, Args&&... args) BOOST_MULTI_DECLRETURN_(cufft ::dft_forward(std::forward<Args>(args)...))
 	#endif
 	template<class In, class... Args> auto dft_forward(std::array<bool, std::decay_t<In>::dimensionality> which, In&& in, Args&&... args) BOOST_MULTI_DECLRETURN_(dft_forward_aux(priority<1>{}, which, std::forward<In>(in), std::forward<Args>(args)...))
 
 	template<class... Args> auto dft_backward_aux(priority<0> /*unused*/, Args&&... args) BOOST_MULTI_DECLRETURN_(  fftw::dft_backward(std::forward<Args>(args)...))
-	#if defined(__CUDA__) || defined(__NVCC__)
+	#if defined(__CUDA__) || defined(__NVCC__) || defined(__HIPCC__)
 	template<class... Args> auto dft_backward_aux(priority<1> /*unused*/, Args&&... args) BOOST_MULTI_DECLRETURN_(cufft ::dft_backward(std::forward<Args>(args)...))
 	#endif
 	template<class In, class... Args> auto dft_backward(std::array<bool, In::dimensionality> which, In const& in, Args&&... args) -> decltype(auto) {return dft_backward_aux(priority<1>{}, which, in, std::forward<Args>(args)...);}