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.

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

Conversation

viclafargue
Copy link
Contributor

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 the raft::mr namespace would only be possible once all references to it are removed from RAFT and other libraries using it.

@viclafargue viclafargue requested review from a team as code owners June 29, 2021 12:32
@github-actions github-actions bot added the cpp label Jun 29, 2021
@viclafargue viclafargue requested a review from divyegala July 1, 2021 10:19
@divyegala divyegala added breaking Breaking change improvement Improvement / enhancement to an existing function labels Jul 1, 2021
Copy link
Member

@divyegala divyegala left a 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

cpp/include/raft/comms/test.hpp Outdated Show resolved Hide resolved
cpp/include/raft/comms/test.hpp Outdated Show resolved Hide resolved
cpp/include/raft/comms/test.hpp Outdated Show resolved Hide resolved
cpp/include/raft/label/classlabels.cuh Outdated Show resolved Hide resolved
cpp/include/raft/linalg/eig.cuh Outdated Show resolved Hide resolved
cpp/include/raft/matrix/math.cuh Outdated Show resolved Hide resolved
cpp/include/raft/sparse/csr.cuh Outdated Show resolved Hide resolved
cpp/include/raft/sparse/csr.cuh Outdated Show resolved Hide resolved
cpp/test/linalg/cholesky_r1.cu Outdated Show resolved Hide resolved
cpp/test/sparse/symmetrize.cu Outdated Show resolved Hide resolved
Copy link
Member

@divyegala divyegala left a 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!

cpp/include/raft/label/classlabels.cuh Outdated Show resolved Hide resolved
cpp/include/raft/linalg/eig.cuh Outdated Show resolved Hide resolved
@viclafargue viclafargue changed the base branch from branch-21.08 to branch-21.10 July 19, 2021 16:30
viclafargue added a commit to viclafargue/raft that referenced this pull request Aug 16, 2021
@viclafargue
Copy link
Contributor Author

Unnecessary following the merge of #286, that includes the changes.

rapids-bot bot pushed a commit to rapidsai/cugraph that referenced this pull request Aug 27, 2021
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
rapids-bot bot pushed a commit to rapidsai/cuml that referenced this pull request Aug 30, 2021
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
vimarsh6739 pushed a commit to vimarsh6739/cuml that referenced this pull request Oct 9, 2023
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
breaking Breaking change cpp improvement Improvement / enhancement to an existing function
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants