Skip to content

Refactor and cleanup env-based CUB APIs and underlying implementation #5606

@gonidelis

Description

@gonidelis

The new cub env based APIs reuse a lot of template code that can be extracted to a common header so that we avoid code duplication as discussed here. Put the common lines of code under a single header and use it as common functionality under the APIs implementation.

Also try consolidating the two underlying env cub::DeviceReduce implementations by template away the Dispatch type.

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

Status

In Review

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions