fix: rocm types

yasahi-hpc · Dec 18, 2024 · 3fdd5bf · 3fdd5bf
1 parent 45c489b
commit 3fdd5bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fft/src/KokkosFFT_ROCM_types.hpp b/fft/src/KokkosFFT_ROCM_types.hpp
@@ -85,8 +85,6 @@ struct ScopedRocfftPlan {
                    const std::vector<int> &out_extents,
                    const std::vector<int> &fft_extents, int howmany,
                    Direction direction, bool is_inplace) {
-    ScopedRocfftPlanDescription scoped_description;
-
     auto [in_array_type, out_array_type, fft_direction] =
         get_in_out_array_type(transform_type, direction);
 
@@ -101,6 +99,8 @@ struct ScopedRocfftPlan {
     auto reversed_fft_extents =
         convert_int_type_and_reverse<int, std::size_t>(fft_extents);
 
+    // Create a plan description
+    ScopedRocfftPlanDescription scoped_description;
     rocfft_status status = rocfft_plan_description_set_data_layout(
         scoped_description.description(),  // description handle
         in_array_type,                     // input array type