将内核链接到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。