[FEA] Enable NN Descent to return distances

rapidsai / cuvs

cuVS - a library for vector search and clustering on the GPU

https://rapids.ai

Apache License 2.0

132 stars 48 forks source link

[FEA] Enable NN Descent to return distances #170

Open divyegala opened 1 month ago

divyegala commented 1 month ago

This feature is going to be merged in RAFT https://github.com/rapidsai/raft/pull/2345, this issue is to track progress and ensure that the feature gets forward to cuVS as well.

tfeher commented 2 weeks ago

Also we need to handle distance epilogue for NN Descent #2364