我在visual C ++中创建了一个Dll文件来计算CUDA中复数数组的模数。该数组是cufftComplex的类型。然后我在LabVIEW中调用了Dll以检查结果的准确性。我收到的结果不正确。有人能告诉我下面的代码有什么问题吗?我认为我的内核函数应该有问题(我检索cufftComplex数据的方式应该是不正确的)。
#include <math.h>
#include <cstdlib>
#include <cuda_runtime.h>
#include <cufft.h>
extern "C" __declspec(dllexport) void Modulus(cufftComplex *digits,float *result);
__global__ void ModulusComputation(cufftComplex *a, int N, float *temp)
{
int idx = blockIdx.x*blockDim.x + threadIdx.x;
if (idx<N)
{
temp[idx] = sqrt((a[idx].x * a[idx].x) + (a[idx].y * a[idx].y));
}
}
void Modulus(cufftComplex *digits,float *result)
{
#define N 1024
cufftComplex *d_data;
float *temp;
size_t size = sizeof(cufftComplex)*N;
cudaMalloc((void**)&d_data, size);
cudaMalloc((void**)&temp, sizeof(float)*N);
cudaMemcpy(d_data, digits, size, cudaMemcpyHostToDevice);
int blockSize = 16;
int nBlocks = N/blockSize;
if( N % blockSize != 0 )
nBlocks++;
ModulusComputation <<< nBlocks, blockSize >>> (d_data, N,temp);
cudaMemcpy(result, temp, size, cudaMemcpyDeviceToHost);
cudaFree(d_data);
cudaFree(temp);
}
答案 0 :(得分:2)
在代码的最终cudaMemcpy中,您有:
cudaMemcpy(result, temp, size, cudaMemcpyDeviceToHost);
应该是:
cudaMemcpy(result, temp, sizeof(float)*N, cudaMemcpyDeviceToHost);
如果您为cuda调用添加了错误检查,您会看到此cuda调用(如最初编写的那样)会抛出错误。
还可以提出其他意见。例如,块大小(16)应该是32的整数倍。但这并不妨碍正常运行。
答案 1 :(得分:1)
在内核调用之后,在复制结果时,您使用size
作为内存大小。 cudaMemcpy
的第三个参数应为N * sizeof(float)
。