Added faiss gpu (disabled for normal runs).
@@ -31,6 +31,15 @@ float:
base:
args: [[5, 10, 20, 50, 100, 200, 400, 800, 1000]]
query-args: [[1, 2, 3, 4, 5, 8, 10, 20, 50, 100, 200]]
+ faiss-gpu:
+ disabled: true
+ docker-tag: ann-benchmarks-faiss
+ module: ann_benchmarks.algorithms.faiss_gpu
+ constructor: FaissGPU
+ run-groups:
+ base:
+ args: [[400, 1024, 4096, 8192, 16384],
+ [1, 10, 40, 100, 200]]
hnswlib:
docker-tag: ann-benchmarks-hnswlib
module: ann_benchmarks.algorithms.hnswlib
