faiss icon indicating copy to clipboard operation
faiss copied to clipboard

Published 20 hours ago • verified publisher icon facebookresearch

faiss.knn_gpu vs torch.topk

Open algoriddle opened this issue 7 months ago • 3 comments

Summary

torch.topk outperforms faiss.knn_gpu - both classic and raft

Platform

OS: Linux, Tesla V100-SXM2-16GB

Faiss version:

faiss-gpu-raft 1.8.0 py3.12_h4c7d538_114_cuda12.1.1_nightly pytorch/label/nightly libfaiss 1.8.0 hb0f4bcb_114_cuda12.1.1_raft_nightly pytorch/label/nightly

Installed from: conda

Faiss compilation options: OPTIMIZE AVX2 GPU NVIDIA_RAFT

Running on:

  • [ ] CPU
  • [x] GPU

Interface:

  • [ ] C++
  • [x] Python

Reproduction instructions

See notebook: https://gist.github.com/algoriddle/bae7ebaf4cee6a63b218ce24f0558cf0

@cjnolet, can you tag the appropriate people on your side, please?

algoriddle avatar Jul 09 '24 16:07 algoriddle