CUFFT库实践（1）

[adagio_chen]

好吧，最近经常要使用快速傅立叶变换来解决一些问题。下面就给出一个1维FFT的例子。特别需要注意的是，一组信号做一次FFT再做一次逆FFT，并不等于原信号，而需要做一次归一化（代码中的Scale函数）。

// 实数与复数之间的互相转换。楼猪死脑子，只用C2C......
__global__ 
void RToC(int size, float* input, float2* output)
{
	int i = blockIdx.x * blockDim.x + threadIdx.x;
	if (i < size)
	{
		output[i].x = input[i];
		output[i].y = 0;
	}
}

__global__ 
void CToR(int size, float2* input, float* output)
{
	int i = blockIdx.x * blockDim.x + threadIdx.x;
	if (i < size)
	{
		output[i] = input[i].x;
	}
}

template <class T>
__global__
void Scale(int size, T* input, float scale)
{
	int i = blockIdx.x * blockDim.x + threadIdx.x;
	if (i < size)
	{
		input[i] *= scale;
	}
}

// 正向 FFT
void FFT1D(int nx, float* input, float2* output)
{
	float* input_d;
	checkCudaErrors( cudaMalloc((void**)&input_d, nx * sizeof(float)) );
	checkCudaErrors( cudaMemcpy(input_d, input, nx * sizeof(float), cudaMemcpyHostToDevice) );

	float2* input_c;
	checkCudaErrors( cudaMalloc((void**)&input_c, nx * sizeof(float2)) );

	int thread = 128;
	int block = nx / 128 + 1;
	RToC<<<block, thread>>>(nx, input_d, input_c);

	cufftHandle plan;
	cufftPlan1d(&plan, nx, CUFFT_C2C, 1);
	cufftExecC2C(plan, input_c, input_c, CUFFT_FORWARD);

	checkCudaErrors( cudaMemcpy(output, input_c, nx * sizeof(float2), cudaMemcpyDeviceToHost) );

	cudaFree(input_c);
	cudaFree(input_d);
}

// 逆向 FFT
void IFFT1D(int nx, float2* input, float* output)
{
	float2* input_d;
	checkCudaErrors( cudaMalloc((void**)&input_d, nx * sizeof(float2)) );
	checkCudaErrors( cudaMemcpy(input_d, input, nx * sizeof(float2), cudaMemcpyHostToDevice) );

	cufftHandle plan;
	cufftPlan1d(&plan, nx, CUFFT_C2C, 1);
	cufftExecC2C(plan, input_d, input_d, CUFFT_INVERSE);

	float* output_d;
	checkCudaErrors( cudaMalloc((void**)&output_d, nx * sizeof(float)) );
	
	int thread = 128;
	int block = nx / thread + 1;
	CToR<<<block, thread>>>(nx, input_d, output_d);

	Scale<<<block, thread>>>(nx, output_d, 1.0f / nx);

	checkCudaErrors( cudaMemcpy(output, output_d, nx * sizeof(float), cudaMemcpyDeviceToHost) );
	
	cudaFree(input_d);
	cudaFree(output_d);
}

[austin9972]

[cattpon]

进来学习一下~

[hugh_z]

learning

[云满笔记]

FFT？ 记得大学的时候被它坑苦了 。。。 支持支持

[伊顺鸣]

是这样的，，，，，，，，，，

[xiaofeng6896]

好，赞一个。。。。。。。

[nudtzzt]

楼主你好，小弟最近刚刚开始学习CUDA，想用cufft对多帧图像进行傅里叶变换，能讲解一下cufft2D和cufft3D的区别么？

[Mary100860]

楼主你好，我按照你提供的代码实际运行操作了一遍，发现几个问题，想向你请教一下： 1、在运行的过程中，分配input_c显存失败，请问楼主也遇到这个问题了吗，是怎样解决的呢？ 2、在进行傅里叶变换的时候，输入数据是float2类型，为什么不用cufftComplex类型呢？ ps：本人小白，学习CUFFT库的使用，谢谢楼主！

[Xiaoweiwei902]

请问楼主，我在TK1板子上运行关于cufft进行图像处理时怎么比在我自己电脑上还慢呢？而且还慢很多