2

次のコードは、cufftPlan1d を使用して単一の 1D 変換に適用するために、ここから変更されています。最終的にはバッチ処理された R2C 変換を実行したいのですが、以下のコードは個別の入力配列と出力配列を使用して単一の変換を実行します。

このコードを適応させて変換をインプレースで実行し、デバイスに割り当てられるメモリの量を減らすにはどうすればよいでしょうか?

Cuda 6.5 に感謝
- 注: MATLAB 2015a で mexFunction からコードを実行しています

コード：

#include <stdlib.h>
#include <stdio.h>
#include <cuda_runtime.h>
#include <cufft.h>
#define DATASIZE 8
#define BATCH 1
#define gpuErrchk(ans) { gpuAssert((ans), __FILE__, __LINE__); }
inline void gpuAssert(cudaError_t code, const char *file, int line, bool  abort=true)
{
   if (code != cudaSuccess) 
   {
        fprintf(stderr,"GPUassert: %s %s %d\n", cudaGetErrorString(code), file, line);            
        if (abort) exit(code);
   }
}

void main(int argc, char **argv)
{   

// --- Host side input data allocation and initialization
cufftReal *hostInputData = (cufftReal*)malloc(DATASIZE*sizeof(cufftReal));
for (int j=0; j<DATASIZE; j++) hostInputData[j] = (cufftReal)(j + 1);

// --- Device side input data allocation and initialization
cufftReal *deviceInputData; 
gpuErrchk(cudaMalloc((void**)&deviceInputData, DATASIZE * sizeof(cufftReal)));
cudaMemcpy(deviceInputData, hostInputData, DATASIZE * sizeof(cufftReal), cudaMemcpyHostToDevice);

// --- Host side output data allocation
cufftComplex *hostOutputData = (cufftComplex*)malloc((DATASIZE / 2 + 1) * BATCH * sizeof(cufftComplex));

// --- Device side output data allocation
cufftComplex *deviceOutputData;   gpuErrchk(cudaMalloc((void**)&deviceOutputData, (DATASIZE / 2 + 1) * sizeof(cufftComplex)));

cufftResult cufftStatus;
cufftHandle handle;

cufftStatus = cufftPlan1d(&handle, DATASIZE, CUFFT_R2C, BATCH);
if (cufftStatus != cudaSuccess) { mexPrintf("cufftPlan1d failed!"); }       

cufftStatus = cufftExecR2C(handle,  deviceInputData, deviceOutputData);
if (cufftStatus != cudaSuccess) { mexPrintf("cufftExecR2C failed!"); }  

// --- Device->Host copy of the results
gpuErrchk(cudaMemcpy(hostOutputData, deviceOutputData, (DATASIZE / 2 + 1) * sizeof(cufftComplex), cudaMemcpyDeviceToHost));

for (int j=0; j<(DATASIZE / 2 + 1); j++)
        printf("%i %f %f\n", j, hostOutputData[j].x, hostOutputData[j].y);

cufftDestroy(handle);
gpuErrchk(cudaFree(deviceOutputData));
gpuErrchk(cudaFree(deviceInputData));

}
4

2 に答える 2

0

解決策は、別の回答で既に提供されています: https://stackoverflow.com/a/19208070/678093

あなたの例では、これは次のことを意味します。

入力を cufftComplex として割り当てます。

cufftComplex *deviceInputData;
gpuErrchk(cudaMalloc((void**)&deviceInputData, DATASIZE * sizeof(cufftComplex)));
cudaMemcpy(deviceInputData, hostInputData, DATASIZE * sizeof(cufftReal), cudaMemcpyHostToDevice);

インプレース変換:

cufftStatus = cufftExecR2C(handle,  (cufftReal *)deviceInputData, deviceInputData);
gpuErrchk(cudaMemcpy(hostOutputData, deviceInputData, (DATASIZE / 2 + 1) * sizeof(cufftComplex), cudaMemcpyDeviceToHost));

ところで: MATLAB にはfft () の GPU アクセラレーション バージョンも含まれています。gpu.html#btjw5gk

于 2015-03-25T17:19:31.707 に答える