facebookresearch/faiss

more than 2048 nearest neighbors

arnavmdas opened this issue · 1 comments

Is it possible to modify the source code such that it is possible to retrieve more than 2047 nearest neighbors on faiss-gpu?

The details of the k-value constraints can be found in the Billion-scale similarity search with GPUs paper section 4.2. Since the paper was published, the k-value constraint was increased from 1024 to 2048 in the 1.5.1 release. The implementation of the warp select algorithm can be found in faiss/gpu/utils/warpselect path.

Do you have a specific use case that requires more than 2048 neighbors?