CUDA：注入我自己的PTX功能？

本文介绍了CUDA：注入我自己的PTX功能？的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！

问题描述

我想能够使用PTX 1.3中的一个功能，该功能尚未实现它的C接口。有没有办法在PTX中编写自己的函数并注入到现有的二进制文件中？

I would like to be able to use a feature in PTX 1.3 which is not yet implemented it the C interface. Is there a way to write my own function in PTX and inject into an existing binary?

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

The feature I'm looking for is getting the value of %smid

推荐答案

答案：

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

这篇关于CUDA：注入我自己的PTX功能？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！