Use signed offsets

NVIDIA · Nov 13, 2022 · 6153c10 · 6153c10
1 parent 8b9f0b0
commit 6153c10
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cub/detail/choose_offset.cuh b/cub/detail/choose_offset.cuh
@@ -50,10 +50,10 @@ struct ChooseOffsetT
       !std::is_same<typename std::remove_cv<NumItemsT>::type, bool>::value,
     "NumItemsT must be an integral type, but not bool");
 
-  // Unsigned integer type for global offsets.
+  // Signed integer type for global offsets.
   using Type = typename std::conditional<sizeof(NumItemsT) <= 4,
-                                         std::uint32_t,
-                                         unsigned long long>::type;
+                                         std::int32_t,
+                                         std::int64_t>::type;
 };
 
 } // namespace detail