memory - 分配全局内存

Question

我有以下代码在 GPU 上分配全局内存。

__global__ void mallocTest()
{
    char* ptr = (char*)malloc(123);
    //....
    free(ptr);
}

每个线程都会为单独的 ptr 分配内存吗？
所以如果我有 2 个 10 个线程的块，那么分配 20 个数组(即每个线程分配内存供自己使用)？
如何只为每个块而不是每个线程分配内存？
即如果我有 2 个块和 10 个线程，则只分配 2 个数组。
这可能吗？

Answer 1

如果您在计算能力 2.0 或 2.1 设备上执行该代码，则每个线程都将从运行时全局内存堆执行分配。因此，如果您的执行网格有 20 个线程，您将获得 20 个分配:每个线程一个。

如果您希望每个块都有一个数组(并且您希望块中的每个线程都访问同一个数组)，那么逻辑方法是使用共享内存，特别是如果内存仅用于块的生命周期并且不是“t 打算再次使用。