Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Commit

Permalink
Fix device-side reduce launch with large inputs
Browse files Browse the repository at this point in the history
  • Loading branch information
gevtushenko committed Nov 25, 2022
1 parent bf56378 commit 32a3094
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cub/device/dispatch/dispatch_reduce.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -783,7 +783,7 @@ struct DispatchReduce : SelectedPolicy
.doit(single_tile_kernel,
d_block_reductions,
d_out,
reduce_grid_size,
reduce_grid_size, // triple_chevron is not type safe, make sure to use int
reduction_op,
init);

Expand Down Expand Up @@ -841,7 +841,7 @@ struct DispatchReduce : SelectedPolicy
DeviceReduceSingleTileKernel<MaxPolicyT,
AccumT *,
OutputIteratorT,
OffsetT,
int, // Always used with int offsets
ReductionOpT,
InitT,
AccumT>);
Expand Down

0 comments on commit 32a3094

Please sign in to comment.