如何创建和使用来自CUDA C的动态库“. so”，并在Linux环境（CentOS）下的C#代码中使用它？

Question

我试图创建一个动态库。所以用CUDA C内核在Linux环境(CentOS)下的C#代码内部使用。我寻找一种方法来做到这一点，但不幸的是没有找到一个完整清晰的解决方案。有些解决方案只是其中的一部分，比如在Linux上创建C共享库，或者使用nvcc在CUDA中创建库链，但是没有从CUDA C中创建动态库的方法。就像在这个解决方案中一样。

有没有办法创建这个动态库并在C#代码中成功使用它？

Answer 1

在搜索了多个不同的解决方案，并试图收集和测试可用的可能性之后，我终于找到了这个简单的方法。
原始的C库可以使用gcc一步完成，就像这个答案一样。

gcc -shared -o dll.so -fPIC dllmain.cpp

但请确保在. cpp文件中所需的函数之前添加extern"C"，如下所示：

#include <stdio.h>
extern "C" void func()
{
    // code
}

对于CUDA C，nvcc可以以同样的方式使用，类似于这个答案和这个答案的组合。确保使用. so而不是. dll并使用适当的设备架构，我在这里使用60，因为我正在使用“特斯拉P100-PCIE-16GB”。

nvcc -arch=sm_60 --compiler-options '-fPIC' -o dll.so --shared kernel.cu

.cu 文件与此类似。

#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdio.h>

extern "C" void myfunc(int a, int b, ...);
__global__ void kernel(int a, int b, ...);

__global__ void kernel(int a, int b, ...)
{
    int i = threadIdx.x;
    // kernel code
}

void myfunc(int a, int b, ...)
{
    // code
}

现在，动态库 .so 已创建，可以在 C# 代码中使用，如下所示。

using System;
using System.Runtime.InteropServices;
class Program
{
    [DllImport("dll.so")]
    static extern myfunc(int a, int b, ...);

    private void Method()
    {
        int a, b;
        // code
        myfunc(a, b, ...);
    }
}

然后使用 Mono 编译 C# 代码。

mcs Program.cs
mono Program.exe

但是可能有必要像这样设置所用库的路径。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/path/to/library/

这适用于简单的CUDAC代码，也可能适用于其他代码，但根据其复杂性，可能会出现一些问题。