bmcv_fft
FFT运算。完整的使用步骤包括创建、执行、销毁三步。
创建
支持一维或者两维的FFT计算,其区别在于创建过程中,后面的执行和销毁使用相同的接口。
对于一维的FFT,支持多batch的运算,接口形式如下:
bm_status_t bmcv_fft_1d_create_plan( bm_handle_t handle, int batch, int len, bool forward, void *&plan);
处理器型号支持:
该接口仅支持BM1684。
输入参数说明:
bm_handle_t handle
输入参数。bm_handle 句柄
int batch
输入参数。batch的数量。
int int
输入参数。每个batch的长度。
bool forward
输入参数。是否为正向变换,false表示逆向变换。
void *&plan
输出参数。执行阶段需要使用的句柄。
返回值说明:
BM_SUCCESS: 成功
其他:失败
对于两维M*N的FFT运算,接口形式如下:
bm_status_t bmcv_fft_2d_create_plan( bm_handle_t handle, int M, int N, bool forward, void *&plan);
输入参数说明:
bm_handle_t handle
输入参数。bm_handle 句柄
int M
输入参数。第一个维度的大小。
int N
输入参数。第二个维度的大小。
bool forward
输入参数。是否为正向变换,false表示逆向变换。
void *&plan
输出参数。执行阶段需要使用的句柄。
返回值说明:
BM_SUCCESS: 成功
其他:失败
执行
使用上述创建后的plan就可以开始真正的执行阶段了,支持复数输入和实数输入两种接口,其格式分别如下:
bm_status_t bmcv_fft_execute( bm_handle_t handle, bm_device_mem_t inputReal, bm_device_mem_t inputImag, bm_device_mem_t outputReal, bm_device_mem_t outputImag, const void *plan); bm_status_t bmcv_fft_execute_real_input( bm_handle_t handle, bm_device_mem_t inputReal, bm_device_mem_t outputReal, bm_device_mem_t outputImag, const void *plan);
输入参数说明:
bm_handle_t handle
输入参数。bm_handle 句柄
bm_device_mem_t inputReal
输入参数。存放输入数据实数部分的device memory空间,对于一维的FFT,其大小为batch*len*sizeof(float),对于两维FFT,其大小为M*N*sizeof(float)。
bm_device_mem_t inputImag
输入参数。存放输入数据虚数部分的device memory空间,对于一维的FFT,其大小为batch*len*sizeof(float),对于两维FFT,其大小为M*N*sizeof(float)。
bm_device_mem_t outputReal
输出参数。存放输出结果实数部分的device memory空间,对于一维的FFT,其大小为batch*len*sizeof(float),对于两维FFT,其大小为M*N*sizeof(float)。
bm_device_mem_t outputImag
输出参数。存放输出结果虚数部分的device memory空间,对于一维的FFT,其大小为batch*len*sizeof(float),对于两维FFT,其大小为M*N*sizeof(float)。
const void *plan
输入参数。创建阶段所得到的句柄。
返回值说明:
BM_SUCCESS: 成功
其他:失败
销毁
当执行完成后需要销毁所创建的句柄。
void bmcv_fft_destroy_plan(bm_handle_t handle, void *plan);
示例代码
bool realInput = false; float *XRHost = new float[M * N]; float *XIHost = new float[M * N]; float *YRHost = new float[M * N]; float *YIHost = new float[M * N]; for (int i = 0; i < M * N; ++i) { XRHost[i] = rand() % 5 - 2; XIHost[i] = realInput ? 0 : rand() % 5 - 2; } bm_handle_t handle = nullptr; bm_dev_request(&handle, 0); bm_device_mem_t XRDev, XIDev, YRDev, YIDev; bm_malloc_device_byte(handle, &XRDev, M * N * 4); bm_malloc_device_byte(handle, &XIDev, M * N * 4); bm_malloc_device_byte(handle, &YRDev, M * N * 4); bm_malloc_device_byte(handle, &YIDev, M * N * 4); bm_memcpy_s2d(handle, XRDev, XRHost); bm_memcpy_s2d(handle, XIDev, XIHost); void *plan = nullptr; bmcv_fft_2d_create_plan(handle, M, N, forward, plan); if (realInput) bmcv_fft_execute_real_input(handle, XRDev, YRDev, YIDev, plan); else bmcv_fft_execute(handle, XRDev, XIDev, YRDev, YIDev, plan); bmcv_fft_destroy_plan(handle, plan); bm_memcpy_d2s(handle, YRHost, YRDev); bm_memcpy_d2s(handle, YIHost, YIDev); bm_free_device(handle, XRDev); bm_free_device(handle, XIDev); bm_free_device(handle, YRDev); bm_free_device(handle, YIDev); bm_dev_free(handle);