-
Notifications
You must be signed in to change notification settings - Fork 194
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.
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
[REVIEW] Allow generic reductions for the map then reduce op #94
[REVIEW] Allow generic reductions for the map then reduce op #94
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just a small suggestion, otherwise LGTM!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Adding requested change
Thanks @divyegala for the review, I have updated the ops to allow different output type, and updated the tests. |
a30c273
to
006b03b
Compare
@divyegala could you take a look and re-review? Thanks! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pre-approving in case my suggestion below looks invalid. If it is valid, then great! Thanks, Tamas
All issues are resolved as requested. |
Authors: - Divye Gala (https://github.com/divyegala) - Micka (https://github.com/lowener) - Corey J. Nolet (https://github.com/cjnolet) Approvers: - Corey J. Nolet (https://github.com/cjnolet) URL: rapidsai/cuvs#94
This PR adds a mapThenReduce prim, analogous to mapThenSumReduce.