Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
CAGRA graph optimizer: clamp rev_graph_count (#1987)
This PR fixes a potential out of bounds access. While building the reverse graph, we count the number of potential edges in `rev_graph_count` https://github.com/rapidsai/raft/blob/047bfb2a7a24a97a4bece0b553521533868e7889/cpp/include/raft/neighbors/detail/cagra/graph_core.cuh#L204-L205 While we store only max `output_graph_degree` neighbors, the `rev_graph_count` can be larger than `output_graph_degree`. The loop that uses `rev_graph` is updated to take this limit into account. Authors: - Tamas Bela Feher (https://github.com/tfeher) Approvers: - tsuki (https://github.com/enp1s0) - Corey J. Nolet (https://github.com/cjnolet) URL: #1987
- Loading branch information