rapidsai / raft

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.
https://docs.rapids.ai/api/raft/stable/
Apache License 2.0
680 stars 180 forks source link

Fix 0 recall issue in `raft_cagra_hnswlib` ANN benchmark #2369

Closed divyegala closed 1 week ago

divyegala commented 1 week ago

raft_cagra wrapper stopped including the dataset in the index to save memory, but this adversely affected raft_cagra_hnswlib wrapper because the dataset needed to be included in the index. The need for inclusion of the dataset is because we need the dataset to be serialized when writing to the hnswlib format.

cjnolet commented 1 week ago

/merge