6

在下面我发布的字符中，我正在比较 FFTW 和 CUFFT 中运行的 IFFT 的结果。

出现这种不同的可能原因是什么？真的有那么多舍入错误吗？

这是相关的代码片段：

cufftHandle plan;
cufftComplex *d_data;
cufftComplex *h_data;
cudaMalloc((void**)&d_data, sizeof(cufftComplex)*W);

complex<float> *temp = (complex<float>*)fftwf_malloc(sizeof(fftwf_complex) * W);
h_data = (cufftComplex *)malloc(sizeof(cufftComplex)*W);
memset(h_data, 0, W*sizeof(cufftComplex));

/* Create a 1D FFT plan. */
cufftPlan1d(&plan, W, CUFFT_C2C, 1);

if (!reader->getData(rowBuff, row))    
    return 0;

// copy from read buffer to our FFT input buffer    
memcpy(indata, rowBuff, fCols * sizeof(complex<float>));

for(int c = 0; c < W; c++)
    h_data[c] = make_cuComplex(indata[c].real(), indata[c].imag());

cutilSafeCall(cudaMemcpy(d_data, h_data, W* sizeof(cufftComplex), cudaMemcpyHostToDevice));
cufftExecC2C(plan, d_data, d_data, CUFFT_INVERSE);
cutilSafeCall(cudaMemcpy(h_data, d_data,W * sizeof(cufftComplex), cudaMemcpyDeviceToHost));

for(int c = 0; c < W; c++)
    temp[c] =(cuCrealf(h_data[c]), cuCimagf(h_data[c]));

//execute ifft plan on "indata"
fftwf_execute(ifft);
 ...
 //dump out abs() values of the first 50 temp and outdata values. Had to convert h_data back to a normal complex

ifft 的定义如下：

ifft = fftwf_plan_dft_1d(freqCols, reinterpret_cast<fftwf_complex*>(indata),
                         reinterpret_cast<fftwf_complex*>(outdata), 
                         FFTW_BACKWARD, FFTW_ESTIMATE);

为了生成图表，我在 fftw_execute W 之后转储了 h_data 和 outdata，W 是我正在处理的图像行的宽度。

看到什么明显的东西了吗？

在此处输入图像描述

4

1 回答 1

8

所以看起来 CUFFT 正在返回一个实部和虚部，而 FFTW 只返回实部。当我拥有复杂的两个部分时，CUFFT 复杂库附带的 cuCabsf() 函数会导致它给我一个 sqrt(2) 的倍数

顺便说一句 - 我从来没有能够在 FFTW 和 CUFFT 之间的中间步骤中获得完全匹配的结果。如果你同时做IFFT和FFT，你应该得到一些接近的东西。

于 2011-03-23T21:22:03.020 回答