-
Notifications
You must be signed in to change notification settings - Fork 197
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] Replace RAFT device::buffer
adapter with RMM device_uvector
#283
[REVIEW] Replace RAFT device::buffer
adapter with RMM device_uvector
#283
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.
This may be a little pedantic, but I think we should use rmm::device_scalar
for 1-sized vectors. I also see a pattern of using cudaMemsetAsync
to set values for this scalars, and this wrapper has set_value_async
and set_value_to_zero_async
methods which will simplify the code a bit
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.
Very small changes, pre-approving. LGTM, thanks Victor!
Unnecessary following the merge of #286, that includes the changes. |
This PR apply modifications to the cuGraph codebase to account for changes in RAFT and RMM : - rapidsai/raft#283 - rapidsai/raft#285 - rapidsai/raft#286 - rapidsai/rmm#816 This PR requires some changes in the cuHornet dependency : rapidsai/cuhornet#52 Authors: - Victor Lafargue (https://github.com/viclafargue) Approvers: - Brad Rees (https://github.com/BradReesWork) - AJ Schmidt (https://github.com/ajschmidt8) - Seunghwa Kang (https://github.com/seunghwak) - Chuck Hastings (https://github.com/ChuckHastings) URL: #1707
This PR apply modifications to the cuML codebase to account for changes in RAFT and RMM : - rapidsai/raft#283 - rapidsai/raft#285 - rapidsai/raft#286 - rapidsai/rmm#816 Authors: - Victor Lafargue (https://github.com/viclafargue) - Dante Gama Dessavre (https://github.com/dantegd) Approvers: - William Hicks (https://github.com/wphicks) - Micka (https://github.com/lowener) - Dante Gama Dessavre (https://github.com/dantegd) - Divye Gala (https://github.com/divyegala) URL: #4077
This PR apply modifications to the cuML codebase to account for changes in RAFT and RMM : - rapidsai/raft#283 - rapidsai/raft#285 - rapidsai/raft#286 - rapidsai/rmm#816 Authors: - Victor Lafargue (https://github.com/viclafargue) - Dante Gama Dessavre (https://github.com/dantegd) Approvers: - William Hicks (https://github.com/wphicks) - Micka (https://github.com/lowener) - Dante Gama Dessavre (https://github.com/dantegd) - Divye Gala (https://github.com/divyegala) URL: rapidsai#4077
This PR contributes to planned changes described in #97.
It replaces references to RAFT's
raft::mr::device::buffer
into its RMM equivalent :rmm::device_uvector
in the RAFT codebase. Removing theraft::mr
namespace would only be possible once all references to it are removed from RAFT and other libraries using it.