CUDA：注入我自己的 PTX 函数？

Question

我希望能够使用 PTX 1.3 中尚未通过 C 接口实现的功能。有没有办法在 PTX 中编写我自己的函数并注入到现有的二进制文件中？

我正在寻找的功能是获取 %smid 的值

Answer 1

答案：

__noinline__ __device__ uint get_smid(void)
{
    uint ret;
    asm("mov.u32 %0, %smid;" : "=r"(ret) );
    return ret;
}