Merge pull request #258 from fkallen/radix_sort_num_items_fix

Use data type OffsetT for DispatchRadixSort correctly in InitPassConfig
NVIDIA · Jan 22, 2021 · 8128ac8 · 8128ac8
2 parents caa350b + 767dadb
commit 8128ac8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub/device/dispatch/dispatch_radix_sort.cuh b/cub/device/dispatch/dispatch_radix_sort.cuh
@@ -1216,7 +1216,7 @@ struct DispatchRadixSort :
             DownsweepKernelT    downsweep_kernel,
             int                 ptx_version,
             int                 sm_count,
-            int                 num_items)
+            OffsetT             num_items)
         {
             cudaError error = cudaSuccess;
             do