CUDA 插件 dlopen

Question

我编写了一个 cuda 插件（动态库），并且我有一个用 C 编写的程序，它使用 dlopen() 来加载该插件。我正在使用 dlsym() 从该插件获取函数。对于我的应用程序来说，任何时候加载插件时，程序都会通过 dlopen() 调用获得新的句柄（库文件可能随后会被修改），这一点非常重要。 因此，在使用我的插件中的函数之后，我调用 dlclose()。调用 dlopen() - dlsym() - dlclose() 发生在我的程序执行期间（在循环中）。

如果我在使用 NVIDIA 驱动程序 256.35（CUDA 3.0 或 3.1）的计算机上工作，则会出现内存泄漏（我在插件 cudaMemGetInfo() 中使用调用诊断）。 如果我在使用 NVIDIA 驱动程序 195.36.15 (CUDA 3.0) 的计算机上工作，程序执行一段时间后会出现错误：“NVIDIA：无法打开设备文件 /dev/nvidia0（打开的文件太多）。”

如果我不使用 dlclose() 调用，程序运行正常，但在这种情况下，我无法在程序执行期间将插件替换为新插件。

有人遇到过这个问题吗？ 谢谢。

Answer 1

没人写CUDA插件吗？

我在 CUDA SDK 上找到了类似的示例：matrixMulDynlinkJIT。我对代码做了一些小修正。特别是，在文件 cuda_drvapi_dynlink.c 中，我更正了 cuInit() 函数：

CUDADRIVER CudaDrvLib = NULL;

CUresult CUDAAPI cuInit(unsigned int Flags)

{

    //CUDADRIVER CudaDrvLib;

    CUresult result;
    int driverVer;

    if (CudaDrvLib != NULL) {
      dlclose (CudaDrvLib);
      CudaDrvLib = NULL;
    }
     .......
}

并在文件 matrixMulDynlinkJIT.cpp 中添加了在 main() 函数中循环：

int main(int argc, char** argv)

{

   printf("[ %s ]\n", sSDKsample);


    while (1) {
       // initialize CUDA

       CUfunction matrixMul = NULL;
       cutilDrvSafeCallNoSync(initCUDA(&matrixMul, argc, argv));

        .....

    }//while (1)
    cutilExit();
}

所以，我遇到了与我的程序相同的问题（执行一段时间后）：“NVIDIA：无法打开设备文件 /dev/nvidia0（打开的设备文件太多）文件）”。
但是当我注释掉 cuda_drvapi_dynlink.c 文件中的 dlclose() 时 – 一切正常，

我无法理解这种行为......
有什么想法吗？