From 3bc99e14394405b67f66bc580d78c0fd1c85b16a Mon Sep 17 00:00:00 2001 From: sakthipriyas <sakthips@gmail.com> Date: Mon, 31 Aug 2020 09:40:30 +0000 Subject: [PATCH] sk kernel version latest --- psrdada_cpp/effelsberg/edd/src/SpectralKurtosisCuda.cu | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/psrdada_cpp/effelsberg/edd/src/SpectralKurtosisCuda.cu b/psrdada_cpp/effelsberg/edd/src/SpectralKurtosisCuda.cu index eca2fa02..456e147f 100755 --- a/psrdada_cpp/effelsberg/edd/src/SpectralKurtosisCuda.cu +++ b/psrdada_cpp/effelsberg/edd/src/SpectralKurtosisCuda.cu @@ -54,11 +54,10 @@ __global__ void compute_sk_kernel(thrust::complex<float> *data, std::size_t samp } __syncthreads(); - for(int s = 1; s < blockDim.x; s *= 2){ - int index = 2 * s * l_index; - if(index < blockDim.x){ - s1[index] += s1[index + s]; - s2[index] += s2[index + s]; + for(int s = blockDim.x / 2; s > 0; s >>= 1){ + if(l_index < s){ + s1[l_index] += s1[l_index + s]; + s2[l_index] += s2[l_index + s]; } __syncthreads(); } -- GitLab