[Misc] Fix signed unsigned comparison warning (#6602)

dmlc · Nov 23, 2023 · 5e78e07 · 5e78e07
1 parent 1b3f14b
commit 5e78e07
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/array/cuda/utils.h b/src/array/cuda/utils.h
@@ -11,6 +11,8 @@
 #include <dgl/runtime/ndarray.h>
 #include <dmlc/logging.h>
 
+#include <type_traits>
+
 #include "../../runtime/cuda/cuda_common.h"
 #include "dgl_cub.cuh"
 
@@ -46,7 +48,8 @@ int _NumberOfBits(const T& range) {
   }
 
   int bits = 1;
-  while (bits < static_cast<int>(sizeof(T) * 8) && (1ull << bits) < range) {
+  const auto urange = static_cast<std::make_unsigned_t<T>>(range);
+  while (bits < static_cast<int>(sizeof(T) * 8) && (1ull << bits) < urange) {
     ++bits;
   }