c++ - 将内核链接到 PTX 函数

Question

我能否将 PTX 文件中包含的 PTX 函数用作外部设备函数，以将其链接到另一个应调用该函数的 .cu 文件？

这是CUDA - link kernels together的另一个问题其中函数本身不包含在 .cu 文件中，但我宁愿有一个 PTX 函数以某种方式链接。

Answer 1

您可以通过cuModuleLoad 和cuModuleGetFunction 从文件系统中加载您自己的代码中包含PTX 代码的文件，如下所示:

CUmodule module;
CUfunction function;

const char* module_file = "my_ptx_file.ptx";
const char* kernel_name = "my_kernel_name";

err = cuModuleLoad(&module, module_file);
err = cuModuleGetFunction(&function, module, kernel_name);

您还可以将 PTX 代码作为字符串直接传递给 CUDA 驱动程序，请参阅 Passing the PTX program to the CUDA driver directly .