0

我正在编写必须在 GTX690（双芯片）板上运行相同内核的代码。由于计算是相当可分离的，我不需要设备之间的数据交换，我只是在 CPU 上合并结果。我了解如何在每个设备上运行代码，以及如何在每个设备的内存空间中提供 i/o 数据空间。

我在尝试设置在两个设备上运行的内核都使用的常量时遇到问题。我需要为每个设备制作一个阵列吗？如果是这样，我如何在内核中检查哪个设备正在运行，以便我可以从数组中访问正确的常量？

这是我正在使用的一些代码。当我仅使用 1 个芯片（通过设置numDev = 1）时，它可以正常工作，但不能同时使用。

__constant__ float d_cellSizeZ;  
std::vector<int*> d_cell;  
.................  
bool Init(int cellsN_, float size_){  
  bool res = true;
  if(cudaSuccess != cudaGetDeviceCount(&numDev))  
    return false;  
  //numDev = 1;  
  d_cl.resize(numDev);  
  for(int i = 0; i < numDev; ++i){  
    res &= (cudaSuccess == cudaSetDevice(i));  
    if(!res)  
      break;  
    res &= (cudaSuccess == cudaMalloc((void**)&d_cell[i], cellsN_*sizef(int)));  
  };  
  res &= (cudaSuccess == cudaMemcpyToSymbol(d_cellSizeZ, &size_, sizeof(float)));  
  if(!res)  
    Cleanup();  
  return res;  
}

在内核中，我只使用d_cellSizeZ常量。那么，我应该如何为每个设备创建一个常量，以便在同一个内核中很好地使用？

另一个问题是：如果我尝试在设备之间交换数据，它会通过 PCI 总线还是双芯片板上存在一些内部路径？

4

1 回答 1

2

您只是d_cellSizeZ在最后一个设备上进行初始化，因此在其他设备上它将是未定义的。您需要d_cellSizeZ在每个设备上进行初始化，最简单的方法是按照 Greg 在评论中建议的那样在循环内执行此操作：

for(int i = 0; i < numDev; ++i)
{  
    checkCudaErrors(cudaSetDevice(i));
    checkCudaErrors(cudaMalloc((void**)&d_cell[i], cellsN_*sizef(int)));
    checkCudaErrors(cudaMemcpyToSymbol(d_cellSizeZ, &size_, sizeof(float)));
};

d_cellSizeZ 符号的重用确实有点奇怪。幕后有一些聪明之处，但本质上该cudaMemcpyToSymbol()功能会查找当前活动设备上的符号，因此每次都会复制到正确的设备。

于 2012-09-10T16:25:29.737 回答