【CUDA】分配内存使用void**

版权声明:转载请注明出处 http://blog.csdn.net/TwT520Ly https://blog.csdn.net/TwT520Ly/article/details/81100301

在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指向新内存的首地址。

猜你喜欢

转载自blog.csdn.net/TwT520Ly/article/details/81100301