如何从我编写的 __global__ 函数中获取 cuFunction？

Question

假设我想在我编写的某些源代码上使用 CUDA 的较低级别驱动程序 API。我知道cuLaunchKernel，但我似乎无法在文档中找到如何cuFunction从函数传递给它的确切解释__global__。

Answer 1

您使用cuModuleGetFunction。如果您不使用 C 链接，则传递的函数名称必须是重整的 C++ 名称。您可以在设备源的编译版本上使用 cuobjdump 来获取该信息。