testing seg type

NVIDIA · Jan 11, 2025 · 87efa98 · 87efa98
1 parent afeda78
commit 87efa98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cub/cub/device/dispatch/dispatch_segmented_sort.cuh b/cub/cub/device/dispatch/dispatch_segmented_sort.cuh
@@ -76,7 +76,7 @@ namespace detail
 namespace segmented_sort
 {
 // Type used to index within segments within a single invocation
-using per_invocation_segment_offset_t = unsigned int;
+using per_invocation_segment_offset_t =  int;
 // Type used for total number of segments and to index within segments globally
 using global_segment_offset_t = ::cuda::std::int64_t;
 
@@ -576,7 +576,7 @@ template <typename LargeSegmentPolicyT,
 CUB_RUNTIME_FUNCTION _CCCL_VISIBILITY_HIDDEN cudaError_t DeviceSegmentedSortContinuation(
   LargeKernelT large_kernel,
   SmallKernelT small_kernel,
-  int num_segments,
+  unsigned int num_segments,
   KeyT* d_current_keys,
   KeyT* d_final_keys,
   detail::device_double_buffer<KeyT> d_keys_double_buffer,