-2

我目前正在尝试循环运行多个 FFT，以克服 cuFFT 计划的最大 1.28 亿个元素。因此，例如，我将在一个循环中运行 1.28 亿个元素。

我的程序对于单个 FFT 调用工作得很好，但循环似乎不起作用。我想可能是因为我如何抵消 FFT。这是我如何做到的一个片段：

cufftComplex *d_signal;
checkCudaErrors(cudaMalloc((void **)&d_signal, mem_size));
cufftComplex *d_filter_kernel;
checkCudaErrors(cudaMalloc((void **)&d_filter_kernel, mem_size));

int rankSize = 2;       
int rank[2];
    rank[0] = TempSearchSizeY; rank[1] = TempSearchSizeX;       
int FFTPlanSize = 500;
cufftHandle planinitial;
cufftResult r;
r = cufftPlanMany(&planinitial, rankSize, rank, NULL, 1, 0, NULL, 1, 0, CUFFT_C2C, FFTPlanSize);
int NrOfFFTRuns = ceil(loadsize / FFTPlanSize);
int FFTOffset = 0;

    checkCudaErrors(cudaMemcpy(d_signal, imageNew, sizeof(Complex)*TempSearchArea*loadsize, cudaMemcpyHostToDevice));
    checkCudaErrors(cudaMemcpy(d_filter_kernel, tempNew, sizeof(Complex)*TempSearchArea*loadsize, cudaMemcpyHostToDevice));


    for (int a = 0; a < NrOfFFTRuns; a++){
                FFTOffset = FFTPlanSize*a;
                r = cufftExecC2C(planinitial, (cufftComplex *)&d_signal[FFTOffset], (cufftComplex *)&d_signal[FFTOffset], CUFFT_FORWARD);
                PrintFFTPlanStatus(r);
                r = cufftExecC2C(planinitial, (cufftComplex *)&d_filter_kernel[FFTOffset], (cufftComplex *)&d_filter_kernel[FFTOffset], CUFFT_FORWARD);
                PrintFFTPlanStatus(r);
                cout << "Run inital" << endl;
    {

上面的代码返回错误的结果。有人可以帮我解决问题吗？

4

1 回答 1

1

我自己想通了。

我忘记将每个批次的元素大小（TempSearchSizeY*TempSearchSizeX）乘以偏移值。它应该是

offset = a * element size * batch size.

本案仅包含

offset = a* batch size.
于 2016-06-09T09:28:26.310 回答