如何将数据传递给 CUDA 中的共享变量？

Question

我有一个传递3个数组的内核，第一个数组d_A1没有数据，仅用于写回数据，其他两个数组d_D1和d_ST1有数据。

第一个数组的大小为：

d_A1[13000000]

第二个数组的大小为：

d_D1[421]

第三个数组的大小为：

d_ST1[21]

N is 13000000

TestArray<<<n_blocks, block_size>>>(d_A1,N, d_D1, d_ST1);

现在我只想传递 d_D1[421] 和 d_ST1[21 的数据] 到共享数组，因此我将共享数组创建为：

__global__ void TestArray(int* A1, unsigned int N,  int* D1, unsigned int* ST1)
{

   unsigned int __align__(16) tid = threadIdx.x;
   unsigned int __align__(16) idx = __umul24(blockDim.x, blockIdx.x) + threadIdx.x;  
   __shared__ unsigned int __align__(16) s_D1[441];  //Shared array for d_D1
   __shared__ unsigned int __align__(16) s_ST1[21];  //Shared array for d_ST1

   if (idx < N)   //13000000

   {

问：如何将 d_D1[441] 和 d_ST1[21] 的数据传递到 s_D1[441] 和 s_ST1[21]？ 我尝试过：

      while (idx < 441)

        s_D1[tid] = d_D1[idx] 

      __syncthreads(); 


      while (idx < 21)

        s_ST1[tid] = d_ST1[idx] 


      __syncthreads();  

但是计算机死机了，我必须重新启动它。 我当时也尝试过一个，就是只有拳头 一会儿，然后只有第二一会儿，没有运气。

如果我使用全局内存，即 d_D1、​​d_ST1 一切正常。 所以问题是： 当数组的大小不是 N 时，如何将数据传递给共享变量/数组？

   }   //End of kernel processing



}

Answer 1

太好了，你是对的，它必须是 if (tid < 21) d_ST1[tid] = ST1[tid] 因为一个块足以满足这两个大小。确实非常感谢你。
但是，什么时候使用 if (idx < n) d_st1[tid] = st1[idx] ？