c# - 如何将 C# 中的二维数组传递给 C++ DLL？

Question

我想将 C# 中的二维数组作为参数传递给 C++ DLL 中的函数(它必须是 C++，因为我使用的是 CUDA C++)。我尝试了很多东西，但未能直接使用数组或 vector 传递它。我唯一能做的就是将其转换为一维数组，将其连同其维度传递给函数，然后将其转换回二维 vector 。这是 C++ DLL 代码:

int Add(int* a, int m, int n)
{
    int i, j, ans;
    vector<vector<int>> A(m, vector<int>(n));
    for (i = 0; i < m; i++)
        for (j = 0; j < n; j++)
            A[i][j] = a[i * n + j];
    // Main calculations
    return ans;
}

这是传递数组的 C# 代码:

[DllImport("CUDALib.dll")]
static extern int Add(int[] a, int m, int n);
private void PassArray(int[,] A)
{
    int i, j, m, n, ans;
    int[] a;
    m = A.GetLength(0);
    n = A.GetLength(1);
    a = new int[m * n];
    for (i = 0; i < m; i++)
        for (j = 0; j < n; j++)
            a[i * n + j] = A[i, j];
    ans = Add(a, m, n);
}

有什么更快、更有效、更直接的方法吗？

Answer 1

C# 中的二维数组是 contiguous in memory ，因此不需要在任何一侧进行所有这些内存复制。您应该按原样在 C# 中传递数组指针:

[DllImport("CUDALib.dll", CallingConvention = CallingConvention.Cdecl)]
static extern int Add(int[,] a, int m, int n);
static void PassArray(int[,] A)
{
    int m = A.GetLength(0);
    int n = A.GetLength(1);
    int ans = Add(A, m, n);
}

然后访问 C++ 中的单个元素:

extern "C" __declspec(dllexport) int Add(int* a, int m, int n)
{
    int i, j;
    for (i = 0; i < m; i++)
        for (j = 0; j < n; j++)
            printf("%d %d: %d\n", i, j, a[i * n + j]);
    // Main calculations
    return 0;
}