Make in_tmp Managed

yasahi-hpc · Oct 30, 2024 · d51e33c · d51e33c
1 parent 4d1d582
commit d51e33c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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<OutViewType>::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(