revert scan_by_key tile_idx type to int, static_cast on multiplicatio…

…n instead.
NVIDIA · May 10, 2021 · 62e1891 · 62e1891
1 parent f537e42
commit 62e1891
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,8 +512,8 @@ namespace __scan_by_key {
             inequality_op(equality_op_),
             scan_op(scan_op_)
       {
-        Size tile_idx      = blockIdx.x;
-        Size tile_base     = ITEMS_PER_TILE * tile_idx;
+        int  tile_idx       = blockIdx.x;
+        Size tile_base     = ITEMS_PER_TILE * static_cast<Size>(tile_idx);
         Size num_remaining = num_items - tile_base;
 
         if (num_remaining > ITEMS_PER_TILE)