如何獲取CUDA內核函數中數組的大小？

Question

我有一個這樣的CUDA內核函數：

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

和功能主要我把它叫做：

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

我怎樣才能得到共享存儲陣列的大小？ 這是我第一次在StackOverflow上有一個問題:)謝謝！

Answer 1

將它作爲內核參數傳遞？

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

Answer 2

您可以使用：sizeof(array)獲取陣列的大小（以字節爲單位）。