fix: HIP plans

yasahi-hpc · Dec 3, 2024 · 9e2cf2f · 9e2cf2f
1 parent b89170a
commit 9e2cf2f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/fft/src/KokkosFFT_HIP_plans.hpp b/fft/src/KokkosFFT_HIP_plans.hpp
@@ -80,6 +80,8 @@ auto create_plan(const ExecutionSpace& exec_space,
   plan = std::make_unique<PlanType>();
   // hipfftResult hipfft_rt = hipfftCreate(&(*plan));
   // KOKKOSFFT_THROW_IF(hipfft_rt != HIPFFT_SUCCESS, "hipfftCreate failed");
+
+  hipStream_t stream     = exec_space.hip_stream();
   hipfftResult hipfft_rt = hipfftSetStream((*plan).plan(), stream);
   KOKKOSFFT_THROW_IF(hipfft_rt != HIPFFT_SUCCESS, "hipfftSetStream failed");
 

diff --git a/fft/src/KokkosFFT_HIP_types.hpp b/fft/src/KokkosFFT_HIP_types.hpp
@@ -31,7 +31,7 @@ struct ScopedHIPfftPlanType {
   ScopedHIPfftPlanType() { hipfftCreate(&m_plan); }
   ~ScopedHIPfftPlanType() { hipfftDestroy(m_plan); }
 
-  ScopedHIPfftPlanType &plan() { return m_plan; }
+  hipfftHandle &plan() { return m_plan; }
 };
 
 #if defined(ENABLE_HOST_AND_DEVICE)