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 `ef` setting in HNSW wrapper #2367

Closed divyegala closed 1 week ago

divyegala commented 1 week ago

Closes #2363

Bugs fixed:

  1. Setting ef in search, it was not being set at all before
  2. from_cagra used a hard coded filename to serialize CAGRA graph and deserialize to HNSW graph. The PR changes the hardcoded filename to a random string so that multiple graphs may be converted concurrently

cc @Presburger thank you for reporting these bugs

cjnolet commented 1 week ago

/merge