CUDA 核函数参数大小获取:使用 sizeof() 函数
在 CUDA 中,可以使用 'sizeof()' 函数来获取参数的大小。例如,在核函数中,如果要获取第一个参数的大小,可以使用以下代码:
__global__ void myKernel(int *data, size_t size)
{
size_t paramSize = sizeof(data);
// ...
}
在此示例中,'paramSize' 将包含 'data' 参数的大小(即指向整数数组的指针的大小)。您还可以使用 'sizeof()' 来获取其他类型的参数的大小,例如:
__global__ void myKernel(float *data, int count)
{
size_t param1Size = sizeof(data);
size_t param2Size = sizeof(count);
// ...
}
在此示例中,'param1Size' 将包含 'data' 参数的大小,'param2Size' 将包含 'count' 参数的大小。
原文地址: https://www.cveoy.top/t/topic/nGoB 著作权归作者所有。请勿转载和采集!