我链接到的库使用 cuda 运行时 API。因此,它在首次调用 cuda 函数时隐式创建一个 cuda 上下文。
我的代码(使用库)应使用驱动程序 API。现在,如何让两者(运行时和驱动程序 API)同时工作?
库在库初始化时调用 cudaSetDevice 函数。(我无法改变这一点)。
我能否以某种方式确定上下文并告诉驱动程序 API 使用该上下文?
cuCtxGetCurrent()
获取当前上下文(可能由运行时创建)