Skip to content

Commit

Permalink
Use rmm allocator in CAGRA prune (#1503)
Browse files Browse the repository at this point in the history
This PR modifies the CAGRA prune function to use rmm allocator instead of a custom memory allocator. And also removes the multi-GPU functionality that is not suitable for the [RAFT policy](https://docs.rapids.ai/api/raft/stable/developer_guide/#multi-gpu).

rel: #1461

Authors:
  - tsuki (https://github.com/enp1s0)
  - Tamas Bela Feher (https://github.com/tfeher)

Approvers:
  - Tamas Bela Feher (https://github.com/tfeher)

URL: #1503
  • Loading branch information
enp1s0 authored May 12, 2023
1 parent 56815fb commit b32cd60
Showing 1 changed file with 318 additions and 523 deletions.
Loading

0 comments on commit b32cd60

Please sign in to comment.