fix > intmax num inputs for scan_by_key

NVIDIA · May 6, 2021 · f537e42 · f537e42
1 parent fa54f2c
commit f537e42
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thrust/system/cuda/detail/scan_by_key.h b/thrust/system/cuda/detail/scan_by_key.h
@@ -512,7 +512,7 @@ namespace __scan_by_key {
             inequality_op(equality_op_),
             scan_op(scan_op_)
       {
-        int  tile_idx      = blockIdx.x;
+        Size tile_idx      = blockIdx.x;
         Size tile_base     = ITEMS_PER_TILE * tile_idx;
         Size num_remaining = num_items - tile_base;
 
@@ -734,7 +734,7 @@ namespace __scan_by_key {
                              ScanOp                     scan_op,
                              AddInitToScan              add_init_to_scan)
   {
-    int          num_items    = static_cast<int>(thrust::distance(keys_first, keys_last));
+    Size         num_items    = static_cast<Size>(thrust::distance(keys_first, keys_last));
     size_t       storage_size = 0;
     cudaStream_t stream       = cuda_cub::stream(policy);
     bool         debug_sync   = THRUST_DEBUG_SYNC_FLAG;