如何理解CUDA中的cudaMalloc()的参数

对指针和地址比较熟悉的童鞋可以比较容易的理解这个概念,为了更好的让更多的人理解cudaMalloc的参数,写这篇文章,以飨读者。

首先看下此运行时函数的原型:

[cpp] view plain copy

  1. cudaError_t cudaMalloc (void **devPtr, size_t  size );   


此函数返回值是CUDA中定义的一个错误代码。

主要的第一个参数。为什么是两个星星呢?用个例子来说明下。

[cpp] view plain copy

  1. float *device_data=NULL;  
  2. size_t size = 1024*sizeof(float);  
  3. cudaMalloc((void**)&device_data, size);  


上面这个例子中我在显存中申请了一个包含1024个单精度浮点数的一维数组。而device_data这个指针是存储在主存上的。之所以取device_data的地址,是为了将cudaMalloc在显存上获得的数组首地址赋值给device_data。在函数中为形参赋值是不会在实参中繁盛变化的,但是指针传递的是地址,我们操作了某个地址的数据,实际上是真的改变了指定地址的数据。像这个申请显存的函数,第一个参数传递的是device_data这个指针的地址,然后改变这个地址的内容就会带给实参真正的改变。

如果大家不明白,留言奥。

2015年3月13号16点42分追加:

“指针”也是一种变量,之所以叫他指针,完全是因为它的值是内存地址

既然指针是变量,那么变量也是需要存储空间的。

cudaMalloc的第一个参数传递的是存储在cpu内存中的指针变量的地址,cudaMalloc在执行完成后,向这个地址中写入了一个地址值(此地址值是GPU显存里的)。

既然cudaMalloc的第一个参数传递的是存储在cpu内存中的指针变量的地址,那直接&device_data好了,为什么前边还要写(void**)

Re: bendanban 2017-04-14 20:20发表 [回复]

回复甜糯米团子:因为cudamalloc的c接口是第一个参数是void**,c语言有些编译器不支持隐式类型转换,而大部分c++支持这种隐式转换。

猜你喜欢

转载自blog.csdn.net/curtern/article/details/88204334