使用宽度(以字节为单位)的cudaMemcpy3D中的参数无效?

时间:2012-02-22 17:03:49

标签: cuda

我做了一个简单的texture3D测试,并在将数据复制到设备时发现了一种奇怪的行为。函数cudaMemcpy3D返回'无效参数'。

我发现问题与cudaExtent有关。 根据CUDA工具包参考手册4.0 ,cudaExtent参数如下:

  • w - 字节宽度
  • h - 元素高度
  • d - 元素深度

所以,我按如下方式准备了纹理:

// prepare texture 
cudaChannelFormatDesc t_desc = cudaCreateChannelDesc<baseType>();
// CUDA extent parameters w - Width in bytes, h - Height in elements, d - Depth in elements
cudaExtent t_extent = make_cudaExtent(NCOLS*sizeof(baseType), NROWS, DEPTH);
// CUDA arrays are opaque memory layouts optimized for texture fetching
cudaArray *i_ArrayPtr = NULL;
// allocate 3D
status = cudaMalloc3DArray(&i_ArrayPtr, &t_desc, t_extent);

并配置3D参数如下:

// prepare input data
cudaMemcpy3DParms i_3DParms = { 0 };
i_3DParms.srcPtr   = make_cudaPitchedPtr( (void*)h_idata, NCOLS*sizeof(baseType), NCOLS, NROWS);
i_3DParms.dstArray = i_ArrayPtr;
i_3DParms.extent   = t_extent;
i_3DParms.kind     = cudaMemcpyHostToDevice;

最后将数据复制到设备内存:

// copy input data from host to device
status = cudaMemcpy3D( &i_3DParms );

如果我仅指定x维度中的元素数量,问题就解决了:

cudaExtent t_extent = make_cudaExtent(NCOLS, NROWS, DEPTH);

不会产生任何错误,并且测试按预期工作。

我想知道我是否错过了cudaExtent函数或其他东西。 为什么width参数不需要用字节表示

1 个答案:

答案 0 :(得分:6)

对于CUDA数组,使用数组元素中给定的宽度指定范围。对于分配线性内存,使用以字节为单位指定的宽度指定范围。因为要使用cudaMalloc3DArray分配数组,所以请使用元素中的宽度。如果您使用cudaMalloc3D,则范围将具有以字节为单位的宽度。