c++ - CUDA 驱动程序 CUmodule

Question

我试图了解 CUmodule 在 CUDA 驱动程序 API 函数中实际上代表什么。

许多 CUDA 驱动程序 API 函数都有一个 CUmodule 句柄，但它是什么？它是引导驱动程序调用过程的 dll 的句柄，并且我可以像 GetProcAddress 一样使用 CUDA 函数的名称调用它还是什么？

请把它简单化（最好是简单的英语）

Answer 1

它是您加载的 cuda 编译二进制文件的句柄（简单的谷歌查找和逻辑：P）

来源示例：

#include <stdio.h>
#include <cuda.h>
#include <cuda_runtime_api.h>

int main() {
    CUresult error;
    CUdevice cuDevice;
    CUcontext cuContext;
    CUmodule cuModule;
    CUfunction testkernel;
    // Initialize
    error = cuInit(0);
    if (error != CUDA_SUCCESS) printf("ERROR: cuInit, %i\n", error);
    error = cuDeviceGet(&cuDevice, 0);
    if (error != CUDA_SUCCESS) printf("ERROR: cuInit, %i\n", error);
    error = cuCtxCreate(&cuContext, 0, cuDevice);
    if (error != CUDA_SUCCESS) printf("ERROR: cuCtxCreate, %i\n", error);
    error = cuModuleLoad(&cuModule, "test.cubin");//see here!
    if (error != CUDA_SUCCESS) printf("ERROR: cuModuleLoad, %i\n", error);
    error = cuModuleGetFunction(&testkernel, cuModule, "a");
    if (error != CUDA_SUCCESS) printf("ERROR: cuModuleGetFunction, %i\n", error);
    return 0;
}