Skip to content

Add an option to guarantee a static and runtime max_total_num_items to DeviceBatchedTopK #9279

@elstehle

Description

@elstehle

Description

max_total_num_items is needed to communicate an upper bound on the total number of items processed (e.g. the combined size of all segments in cub::DeviceBatchedTopK), which an algorithm can use to choose offset types (e.g., tile offset size) as well as sizing temporary storage buffers. Since this bound-information may not be attachable to a specific parameter (e.g., on a DeviceBatchedTopK and similarly for some other segmented algorithms that do not have a number of items parameter), we decided that for these cases it should go into the guarantees API.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type
No fields configured for issues without a type.

Projects

Status
In Review

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions