在cuda中分配内存的时候,都是使用void**
进行内存的分配。
对于一般情况:
void memory(int* p) {
p = (int*)malloc(10);
}
int main() {
int* a = 0x0001;
memory(a);
return 0;
}
此时a指向的地址是0x0001
,而且p指向的地址也是0x0001
,但是a和p自身的地址是不一致的,因为在进行传递的过程中,会产生一个副本。因此采用上面的内存分配方式,当分配好新的内存之后,p指向的已经是新内存的首地址,而a依然是指向0x0001
。
因此cuda的实现方式:
void memory(void** p) {
*p = malloc(10);
}
int main() {
int* a = 0x0001;
memory((void**) &a);
return 0;
}
相当于p是指向a的地址的指针,因此*p = a
,给*p
分配内存,相当于让a指向新内存的首地址。