2
我有一个内核,它计算总和。如果我通过内核计算所声明的变量数,我会假设每个内核共有5个寄存器*。但是,在分析内核时,会使用34个寄存器。我需要下到30个寄存器来允许执行1024个线程。CUDA - 内核使用比预期更多的寄存器?
任何人都可以看到什么是错的?
__global__ void sum_kernel(float* values, float bk_size, int start_idx, int end_idx, int resolution, float* avgs){
// Allocate shared memory (assuming a maximum of 1024 threads).
__shared__ float sums[1024];
// Boundary check.
if(blockIdx.x == 0){
avgs[blockIdx.x] = values[start_idx];
return;
}
else if(blockIdx.x == resolution-1) {
avgs[blockIdx.x] = values[start_idx+(end_idx-start_idx)-1];
return;
}
else if(blockIdx.x > resolution -2){
return;
}
// Iteration index calculation.
unsigned int idx_prev = floor((blockIdx.x + 0) * bk_size) + 1;
unsigned int from = idx_prev + threadIdx.x*(bk_size/blockDim.x);
unsigned int to = from + (bk_size/blockDim.x);
to = (to < (end_idx-start_idx))? to : (end_idx-start_idx);
// Partial average calculation using shared memory.
sums[threadIdx.x] = 0;
for (from; from < to; from++)
{
sums[threadIdx.x] += values[from+start_idx];
}
__syncthreads();
// Addition of partial sums.
if(threadIdx.x != 0) return;
from = 1;
for(from; from < 1024; from++)
{
sum += sums[from];
}
avgs[blockIdx.x] = sum;
}
- 假设每个指针2个寄存器,每个无符号整型1个寄存器,存储在常数存储参数。
小修正:谓词不存储在R寄存器中,但是谓词寄存器(在这种情况下为P0)。 – njuffa
@njuffa非常感谢。我立即修复了答案的最后一句。 – JackOLantern