Skip to content

[FEA]: Provide Env-based API for cub::DeviceReduce::Min/ArgMin/Max/ArgMax #5121

Description

@gonidelis

Is this a duplicate?

Area

CUB

Is your feature request related to a problem? Please describe.

Continuing the work for #3855 provide env-based APIs for the cub::DeviceReduce::Min/ArgMin/Max/ArgMax primitives. Make sure:

  • new API works with default/non-default environment
  • new API accepting either deterministic requirements and stream argument.
  • new API accepts tuning environment

Describe the solution you'd like

Complete cub::DeviceReduce::Min/ArgMin/Max/ArgMax interface accepting environment env argument.

Describe alternatives you've considered

No response

Additional context

No response

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

Status
Done

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions