cudaModuleLoadData 失败，错误代码为 201

Question

我有一个要在 GPU 上执行的 ptx 代码。我正在使用以下代码：

CUmodule cudaModule;

//the variable that stores the error associated with cuda API calls.
CUresult cudaErrorVariable;

//variable representing any cuda kernel function.
CUfunction CUDAPipelineKernel;

//initializing cuda driver
cudaErrorVariable = cuInit(0);

//checking for error while loading ptx code in CUmodule.
if(cudaErrorVariable != CUDA_SUCCESS){
    myLogger->error("Unable to initialize CUDA driver");
    return 1;
}

//loading the ptx code into the module.
cudaErrorVariable = cuModuleLoadData(&cudaModule, PTXCode);

//checking for error while loading ptx code in CUmodule.
if(cudaErrorVariable != CUDA_SUCCESS){
    cuGetErrorString(cudaErrorVariable, (const char **)&errorString);
    myLogger->error("Unable load ptx file into the module : CUDA Error {}", cudaErrorVariable);
    return 1;
}

cuModuleLoadData 函数返回错误代码 201。我不知道这个错误代码是什么意思。有人可以帮我找出错误吗？

Answer 1

这里是 cuInit 的链接，这是在任何 cuda 驱动程序 API 调用之前调用的第一个也是最重要的函数，如文档中所述。

为了完整起见，这里是上下文创建的链接：cuCtxCreate。

您也可以使用 Primary context，其灵感来自 cuda 示例目录中的 6_Advanced/ptxjit 示例，该示例使用 cudaMalloc 进行延迟初始化。

主要上下文对于每个设备都是唯一的，并与 CUDA 共享 运行时 API。这些功能允许与其他库集成 使用 CUDA。

Answer 2

正如您在相关的 documentation 中看到的，错误 201 是 CUDA_ERROR_INVALID_CONTEXT，这意味着您在尝试加载模块之前没有正确设置上下文。