Re: [問題] CUDA CUFFT對二維矩陣執行單x軸或y軸的 …

看板C_and_CPP (C/C++)作者lgen7604時間16年前 (2010/01/31 20:13)推噓1(1推 0噓 4→)

留言5則, 3人參與討論串1/2 (看更多)

其實你執行的結果並沒有錯如果你仔細看document的話會注意到在cufftExecR2C()的說明中有提到 This function stores the non-redundant Fourier coefficients in the odata array. 這句話指出執行cufftExecR2C()的結果是non-redundant 這就牽扯到實數DFT的特性問題當DFT的input為實數數列的時候第k項和第n-k項必定為共軛複數假設n=8 共軛情形如下 a[0]a[1]a[2]a[3]a[4]a[5]a[6]a[7] 如果n=9 共軛情形如下 a[0]a[1]a[2]a[3]a[4]a[5]a[6]a[7]a[8] 所以實際上DFT的結果只需要記錄n/2+1項就可以了你測試的範例為n=4 所以每次FFT的結果只要記錄3個複數就可以了正確的範例我改成如下 #include <stdio.h> #include <stdlib.h> #include <cuda_runtime.h> #include <cufft.h> #include <math.h> #define H 4 #define W 4 #define T H*W int main(){ cufftReal *a; a = (cufftReal*) malloc( sizeof(cufftReal)*(T) ); for(int i=0; i<T; i++) a[i] = 2.0; cufftHandle plan; cufftComplex *odata; cufftReal *idata; cudaMalloc( (void**)&idata, sizeof(cufftReal)*(T) ); cudaMalloc( (void**)&odata, sizeof(cufftComplex)*(H*(W/2+1)) ); cudaMemcpy(idata, a, sizeof(cufftReal) * T,cudaMemcpyHostToDevice); cufftPlan1d(&plan, W, CUFFT_R2C, H); cufftExecR2C(plan, (cufftReal*)idata, odata); cufftComplex *FFT_odata; FFT_odata = (cufftComplex*) malloc( sizeof(cufftComplex)*H*(W/2+1) ); cudaMemcpy( FFT_odata, odata, sizeof(cufftComplex)*H*(W/2+1), cudaMemcpyDeviceToHost ); for(int i=0; i <H; i++) { for(int j=0; j <(W/2+1); j++) { printf("(%d,%d)=%.3f%+.3fi ", i, j, FFT_odata[i*(W/2+1)+j].x, FFT_odata[i*(W/2+1)+j].y ); } for(int j=(W-1)/2; j >= 1; j--) { printf("(%d,%d)=%.3f%+.3fi ", i, W-j, FFT_odata[i*(W/2+1)+j].x, -FFT_odata[i*(W/2+1)+j].y ); } printf("\n"); } cufftDestroy(plan); cudaFree(idata); cudaFree(odata); system("PAUSE"); return 0; } 最後要注意的是MATLAB對2D的array做FFT的時候是對column執行上述範例則是對row執行FFT -- ※ 發信站: 批踢踢實業坊(ptt.cc) ◆ From: 122.120.32.11 ※ 編輯: lgen7604 來自: 122.120.32.11 (01/31 20:20)