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] Adopt RMM as the backend for the handle's stream pool #141

Closed
afender opened this issue Feb 8, 2021 · 0 comments · Fixed by #148
Closed

[FEA] Adopt RMM as the backend for the handle's stream pool #141

afender opened this issue Feb 8, 2021 · 0 comments · Fixed by #148
Assignees
@afender
Labels
enhancement New feature or request

Comments

@afender
Copy link
Member

afender commented Feb 8, 2021

RMM now provides a CUDA stream pool for RAPIDS which makes our custom implementation in RAFT handle redundant.
A linear accessor was added in rapidsai/rmm#696 which makes it compatible with existing usage in cuML.
The next step is to adopt that in RAFT.
@afender afender added the enhancement New feature or request label Feb 8, 2021
@afender afender self-assigned this Feb 8, 2021
@afender afender mentioned this issue Feb 11, 2021
Merged
dantegd pushed a commit to dantegd/raft that referenced this issue Jul 23, 2024
@cjnolet
Changing RAFT_EXPLICT_* to CUVS_EXPLITI_* (rapidsai#141)
3515f44 
Changing this for now. We'll work on removing this macro in a future release.

Authors:
  - Corey J. Nolet (https://github.com/cjnolet)

Approvers:
  - Ben Frederickson (https://github.com/benfred)
  - Divye Gala (https://github.com/divyegala)

URL: rapidsai/cuvs#141
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@afender afender

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Streams upgrade in RAFT handle (RMM backend + create handle from parent's pool) afender/raft
1 participant
@afender