Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[FEA] Support generic reduction lambdas for reduce_cols/rows_by_key APIs #860

Open
divyegala opened this issue Sep 30, 2022 · 0 comments
Open

[FEA] Support generic reduction lambdas for reduce_cols/rows_by_key APIs #860

divyegala opened this issue Sep 30, 2022 · 0 comments
Labels
feature request New feature or request

Comments

@divyegala
Copy link
Member

reduce_cols_by_key and reduce_rows_by_key in raft::linalg currently only support sum-reduction. It should be trivial to support generic lambdas for these APIs.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
feature request New feature or request
Projects
VS/ML/DM Primitives Release Board
Status: No status
Milestone
No milestone
Development

No branches or pull requests
1 participant
@divyegala