__global__ void add( int *c, const int* a, const int* b )
{
int x = blockIdx.x;
int y = blockIdx.y;
int offset = x + y * gridDim.x;
c[offset] = a[offset] + b[offset];
}
在上面的示例中,我猜x
,y
,offset
保存在寄存器中
nvcc -Xptxas -v 提供4 registers, 24+16 bytes smem
profiler 显示4个寄存器
以及 ptx 文件的负责人:
.reg .u16 %rh<4>;
.reg .u32 %r<9>;
.reg .u64 %rd<10>;
.loc 15 21 0
$LDWbegin__Z3addPiPKiS1_:
.loc 15 26 0
有人可以澄清寄存器的用法吗?在Fermi中,每个线程的最大寄存器数为63。在我的程序中,我想测试内核消耗太多寄存器的情况(因此变量可能必须自动存储在本地存储器中,从而导致性能下降)。然后在这一点上,我可以将一个内核分成两个,这样每个线程都有足够的寄存器。假设SM资源足以用于并发内核。
我不确定我是不对。
答案 0 :(得分:15)
PTX中的寄存器分配与内核的最终寄存器消耗完全无关。 PTX只是最终机器代码的中间表示,并使用static single assignment form,这意味着PTX中的每个寄存器仅使用一次。一块带有数百个寄存器的PTX可以编译成只有几个寄存器的内核。
寄存器分配由ptxas
完成,作为一个完全独立的编译传递(由驱动程序静态或即时,或两者),它可以对输入PTX执行大量代码重新排序和优化提高吞吐量和保存寄存器,这意味着原始C中的变量或PTX中的寄存器与组装内核的最终寄存器数之间几乎没有关系。
nvcc
确实提供了一些影响汇编程序的寄存器分配行为的方法。您有__launch_bounds__
为编译器提供可能影响寄存器分配的启发式提示,编译器/汇编器采用-maxrregcount
参数(寄存器溢出到本地存储器的潜在费用,这会降低性能) 。 volatile关键字用于对基于nvopen64的旧版本编译器产生影响,并可能影响本地内存溢出行为。但是你不能在原始C代码或PTX汇编语言代码中任意控制或引导寄存器分配。