我試圖在cuda中實現減少,在那裏我找到了數組中的最大元素。我已經使用這個內核來尋找最小值,並且它可以工作,但是當我試圖找到最大值時它不起作用。我已經重複了這個算法,找不到錯誤。任何幫助真的會被讚賞。 (另外,我得到不同的結果,當我取消打印語句出現,這也是頗爲頭疼....)在CUDA中實現減少的問題
__global__
void findMaxAndMin(const float* const d_logLuminance, float* reduceCopy, int length, float* min_logLum, float* max_logLum){
int idx = threadIdx.x + blockDim.x*blockIdx.x;
if(idx >= length){
return;
}
reduceCopy[idx] = d_logLuminance[idx];
__syncthreads();
//do a reduction with max
for(int offset = 1;offset < length;offset = offset*2){
if(idx % (offset*2) == 0){
int compIdx = idx + offset;
if(compIdx < length){
float newVal = a_max(reduceCopy[idx], reduceCopy[compIdx]);
if(idx == 0){
//printf("val %f \n", newVal);
}
__syncthreads();
reduceCopy[idx] = newVal;
__syncthreads();
}
}
__syncthreads();
}
__syncthreads();
if(idx == 0){
*max_logLum = reduceCopy[0];
}
}
非常感謝你,我不知道一些這種cuda的具體行爲...我會盡力重新實現這一點。謝謝! –