Bug fix in next power of 2

ca6c2760 · Marek Kolodziej · 603e17a5 · ca6c2760
Commit ca6c2760 authored Mar 03, 2019 by Marek Kolodziej
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

csrc/welford.cu csrc/welford.cu +2 -2

No files found.
--- a/csrc/welford.cu
+++ b/csrc/welford.cu
@@ -20,13 +20,13 @@ __device__ __forceinline__ int lastpow2(int n)
 }

 __host__ __forceinline__ int h_next_pow2(unsigned int n) {
-    unsigned int old = n;
+    n--;
    n |= (n >>  1);
    n |= (n >>  2);
    n |= (n >>  4);
    n |= (n >>  8);
    n |= (n >> 16);
-    return n == old? n : n + 1;
+    return ++n;
 }

 __host__ __forceinline__ int h_last_pow2(unsigned int n) {