cuda - CUFFT:ピッチポインタのfftを計算するには?

Question

CUFFT を使用して画像の fft を計算しようとしています。CUFFT は、で割り当てられた単純なデバイス ポインターの fft のみを提供しているようですcudaMalloc。

私の入力画像は を使用して割り当てられcudaMallocPitchますが、画像ポインターのピッチを処理するオプションはありません。

現在、行の配置を削除してから fft を実行し、結果をピッチ付きポインターにコピーする必要があります。私の現在のコードは次のとおりです。

void fft_device(float* src, cufftComplex* dst, int width, int height, int srcPitch, int dstPitch)
{
    //src and dst are device pointers allocated with cudaMallocPitch

    //Convert them to plain pointers. No padding of rows.
    float *plainSrc;
    cufftComplex *plainDst;

    cudaMalloc<float>(&plainSrc,width * height * sizeof(float));
    cudaMalloc<cufftComplex>(&plainDst,width * height * sizeof(cufftComplex));

    cudaMemcpy2D(plainSrc,width * sizeof(float),src,srcPitch,width * sizeof(float),height,cudaMemcpyDeviceToDevice);

    cufftHandle handle;
    cufftPlan2d(&handle,width,height,CUFFT_R2C);

    cufftSetCompatibilityMode(handle,CUFFT_COMPATIBILITY_NATIVE);

    cufftExecR2C(handle,plainSrc,plainDst);

    cufftDestroy(handle);

    cudaMemcpy2D(dst,dstPitch,plainDst,width * sizeof(cufftComplex),width * sizeof(cufftComplex),height,cudaMemcpyDeviceToDevice);

    cudaFree(plainSrc);
    cudaFree(plainDst);
} 

正しい結果が得られますが、関数内で2つの余分なメモリ割り当てとコピーを行いたくありません。私はこのようなことをしたい:

void fft_device(float* src, cufftComplex* dst, int width, int height, int srcPitch, int dstPitch)
{
    //src and dst are device pointers allocated with cudaMallocPitch
    //Don't know how to handle pitch here???
    cufftHandle handle;
    cufftPlan2d(&handle,width,height,CUFFT_R2C);

    cufftSetCompatibilityMode(handle,CUFFT_COMPATIBILITY_NATIVE);

    cufftExecR2C(handle,src,dst);

    cufftDestroy(handle);
}

質問：

CUFFTを使用して直接ピッチポインタのfftを計算する方法は?