Cuda实现快速傅里叶变换FFT操作

cuda快速傅里叶变换函数库

CUDA为开发人员提供了多种库，每一类库针对某一特定领域的应用，CUFFT库则是CUDA中专门用于进行傅里叶变换的函数库，可以通过调用cufft函数库中的相关函数直接实现FFT操作。

实现过程

int nlen = 2048,d=1;
float MD_PI =3.1415;
cufftComplex *d_pfFiltImpResp;
cufftComplex *d_pfFiltfft;
cudaMalloc((void **)&d_pfFiltImpResp, sizeof(cufftComplex) * nLen);
cudaMalloc((void **)&d_pfFiltfft, sizeof(cufftComplex) * nLen);
//对d_pfFiltImpResp进行赋值
for (int i=0,i<nlen/2+1,++i)
{
	if (i== 0)
	{
		d_pfFiltImpResp[i].x = 1 / (4 * d *d) ;
	}
	else if (i% 2 == 1)
	{
		float temp = -1.0 / ((MD_PI * i) * (MD_PI * i) * d * d);
		d_pfFiltImpResp[i].x = temp;
		d_pfFiltImpResp[nlen- i].x = temp;
	}
	else
	{
		d_pfFiltImpResp[i] = 0;
	}
}
//创建FFT操作句柄函数
cufftHandle plan;
cufftCreate(&plan);
//类型为复数到复数的一维FFT变换
cufftPlan1d(&plan, nLen, CUFFT_C2C, 1);
//CUFFT_FORWARD为正向FFT，CUFFT_INVERSE为逆向FFT，输出的d_pfFiltfft为FFT之后的结果
cufftResult result = cufftExecC2C(plan, d_pfFiltImpResp, d_pfFiltfft, CUFFT_FORWARD);
cufftDestroy(plan);
cudaFree(d_pfFiltImpResp);
cudaFree(d_pfFiltfft);