cuda - 如何在CUDA内核函数中获取数组的大小？

Question

我有一个像这样的 CUDA 内核函数:

__global__ void kernel(int a, int b)
{
extern __shared__ unsigned char array[];
//int size = /*size of array*/; = 16
}

在 main 函数中我称之为:

sharedMemSize = 16;
kernel<<<gridSize, blockSize, sharedMemSize>>>(1, 2);

如何获取共享内存数组的大小？这是我第一次在 StackOverflow 上提问:) 谢谢!!!

Answer 1

您可以使用以下方法获取数组的大小(以字节为单位):sizeof(array)。