代码之家 › 专栏 › 技术社区 › shoosh

注射我自己的PTX功能?

gpu cuda

4

shoosh · 技术社区 · 15 年前

我希望能够使用ptx 1.3中的一个特性,这个特性还没有在C接口中实现。有没有一种方法可以在ptx中编写自己的函数并注入到现有的二进制文件中?

我要找的功能是获得 %smid

1 回复 | 直到 12 年前

1

6

shoosh 15 年前

答案是:

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