Skip to content

Commit

Permalink
fix annotation for kokkos_malloc
Browse files Browse the repository at this point in the history
  • Loading branch information
Yuuichi Asahi committed Jan 8, 2025
1 parent 109e502 commit 7c30847
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions fft/src/KokkosFFT_ROCM_types.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -111,8 +111,8 @@ struct ScopedRocfftExecutionInfo {

// Set work buffer
if (workbuffersize > 0) {
m_workbuffer = Kokkos::kokkos_malloc<Kokkos::HIP>(
"kokkos_malloc workbuffer", workbuffersize);
m_workbuffer =
Kokkos::kokkos_malloc<Kokkos::HIP>("workbuffer", workbuffersize);

status = rocfft_execution_info_set_work_buffer(
m_execution_info, m_workbuffer, workbuffersize);
Expand Down

0 comments on commit 7c30847

Please sign in to comment.