在统一内存CUDA C/C++中分配2D向量

#include <cuda_runtime.h>

int main() {
    int numRows = 10; // 向量的行数
    int numCols = 5; // 向量的列数

    // 分配2D向量
    float** d_vector;
    cudaMallocManaged(&d_vector, numRows * sizeof(float*));
    for (int i = 0; i < numRows; i++) {
        cudaMallocManaged(&(d_vector[i]), numCols * sizeof(float));
    }

    // 使用2D向量
    for (int i = 0; i < numRows; i++) {
        for (int j = 0; j < numCols; j++) {
            d_vector[i][j] = i + j;
        }
    }

    // 释放内存
    for (int i = 0; i < numRows; i++) {
        cudaFree(d_vector[i]);
    }
    cudaFree(d_vector);

    return 0;
}