diff --git a/src/trans/gpu/algor/hicfft.hip.cpp b/src/trans/gpu/algor/hicfft.hip.cpp index 22931c464..8278d9a4b 100644 --- a/src/trans/gpu/algor/hicfft.hip.cpp +++ b/src/trans/gpu/algor/hicfft.hip.cpp @@ -51,7 +51,7 @@ template class hicfft_plan { else if constexpr (Direction == HIPFFT_Z2D) fftSafeCall(hipfftExecZ2D(handle, data_complex_l, data_real_l)); } - void set_stream(cudaStream_t stream) { + void set_stream(hipStream_t stream) { fftSafeCall(hipfftSetStream(handle, stream)); } hicfft_plan(hipfftHandle handle_, int offset_)