Allocate a 2D vector in unified memory CUDA C/C++

Question

So I need to create a matrix with different row lengths, and this is how it looks like in normal C/C++

int** MpesosT = (int**)malloc(N * sizeof(int*));
    for (int i = 0; i < N; i++)
    {
        MpesosT[i] = (int*)malloc(vecinosT[i] * sizeof(int));
    }

However, I don't know how to do this using the CUDA function to allocate memory:

int* Vector;    cudaMallocManaged(&Vector, VectorSize* sizeof(int));

I can't just use a vector of size N*N or something, because every row has a different size, so how could I do that?

Answer 1

Took a couple of hours, but I found the way to do it. In case anyone has the same problem:

double** Matrix;
cudaMallocManaged((double***)&Matrix, N * sizeof(double*));
for (i = 0; i < N; i++)
{
    cudaMallocManaged((double**)&Matrix[i], rowlength[i] * sizeof(double));
}

This way, every row has a different length