From d51e33cd5056c84606c45b1f7c72b1a0bcba9326 Mon Sep 17 00:00:00 2001 From: Yuuichi Asahi Date: Wed, 30 Oct 2024 19:53:15 +0900 Subject: [PATCH] Make in_tmp Managed --- fft/src/KokkosFFT_Plans.hpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/fft/src/KokkosFFT_Plans.hpp b/fft/src/KokkosFFT_Plans.hpp index dbbb0d38..05f8687f 100644 --- a/fft/src/KokkosFFT_Plans.hpp +++ b/fft/src/KokkosFFT_Plans.hpp @@ -315,7 +315,10 @@ class Plan { using ManagableOutViewType = typename KokkosFFT::Impl::manageable_view_type::type; - InViewType in_tmp; + // FIXME in_tmp should be in InViewType, + // We will merge crop_or_pad and transpose, and remove the need for in_tmp + // InViewType in_tmp; + ManagableInViewType in_tmp; if (m_is_crop_or_pad_needed) { using LayoutType = typename ManagableInViewType::array_layout; ManagableInViewType const in_s(