Различия между выходом FFTW и CUFFT - PullRequest
Новая
       33

Различия между выходом FFTW и CUFFT

6 голосов
/ 09 марта 2011

На графике, который я разместил ниже, я сравниваю результаты IFFT в FFTW и CUFFT.

Каковы возможные причины, по которым это выходит по-другому? Это действительно большая ошибка округления?

Вот соответствующий фрагмент кода: 

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);

и для генерации графика я вывел h_data и outdata после fftw_execute W - ширина строки обрабатываемого изображения.

Видите что-нибудь явно очевидное?

enter image description here

1 Ответ

8 голосов
/ 24 марта 2011

Похоже, CUFFT возвращает реальную и мнимую часть, а FFTW - только реальную. Функция cuCabsf (), которая поставляется с комплексной библиотекой CUFFT, приводит к тому, что это дает мне кратное sqrt (2), когда у меня есть обе части комплекса

В качестве отступления - мне никогда не удавалось получить точно совпадающие результаты на промежуточных этапах между FFTW и CUFFT. Если вы выполняете IFFT и FFT, вы должны получить что-то близкое.

...