运行在什么设备上的cuda内核

Question

是否可以通过内核（GPU代码）找出运行代码的设备？ 您可以找到blockIdx，threadIdx，blockDim和gridDim，但是是否可以找到CUDA设备ID？

对于那些想知道的人，我计划在OptiX程序中使用它，以更好地控制OptiX缓冲区。

Answer 1

这可以在OptiX 3.0和3.5中使用，但我不能保证它总是可以使用。 我们应该在将来的版本中为此提供一个真正的API。

namespace rti_internal_register
{
    __device__ unsigned reg_device_id;
}

rtBuffer<uchar4, 2> output_buffer;

RT_PROGRAM void pinhole_camera()
{
    if( rti_internal_register::reg_device_id == 0 )
        output_buffer[launch_index] = make_color( make_float3( 0.5, 0.0, 0.0 ) );
    else
        output_buffer[launch_index] = make_color( make_float3( 0.0, 0.5, 0.0 ) );
}