Fix issue about 64-bit array length

dmlc · Nov 20, 2023 · 7e276c3 · 7e276c3
1 parent ba2ca4b
commit 7e276c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/array/cuda/utils.h b/src/array/cuda/utils.h
@@ -46,7 +46,7 @@ int _NumberOfBits(const T& range) {
   }
 
   int bits = 1;
-  while (bits < static_cast<int>(sizeof(T) * 8) && (1 << bits) < range) {
+  while (bits < static_cast<int>(sizeof(T) * 8) && (1ll << bits) < range) {
     ++bits;
   }