Fix FAISS CPU algorithm names in raft-ann-bench (#1916)

Authors: - Divye Gala (https://github.com/divyegala) Approvers: - Corey J. Nolet (https://github.com/cjnolet) URL: #1916
rapidsai · Oct 20, 2023 · 945355d · 945355d
1 parent dbcd05d
commit 945355d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/python/raft-ann-bench/src/raft-ann-bench/plot/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/plot/__main__.py
@@ -352,7 +352,9 @@ def load_all_results(
     result_files = [
         result_filename
         for result_filename in result_files
-        if f"{k}-{batch_size}" in result_filename
+        if "csv" in result_filename
+        and f"{k}-{batch_size}"
+        == "-".join(result_filename.replace(".csv", "").split("-")[1:])
     ]
     if len(algorithms) > 0:
         result_files = [

diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/algos.yaml b/python/raft-ann-bench/src/raft-ann-bench/run/algos.yaml
@@ -10,13 +10,13 @@ faiss_gpu_ivf_pq:
 faiss_gpu_ivf_sq:
   executable: FAISS_GPU_IVF_PQ_ANN_BENCH
   requires_gpu: true
-faiss_flat:
+faiss_cpu_flat:
   executable: FAISS_CPU_FLAT_ANN_BENCH
   requires_gpu: false
-faiss_ivf_flat:
+faiss_cpu_ivf_flat:
   executable: FAISS_CPU_IVF_FLAT_ANN_BENCH
   requires_gpu: false
-faiss_ivf_pq:
+faiss_cpu_ivf_pq:
   executable: FAISS_CPU_IVF_PQ_ANN_BENCH
   requires_gpu: false
 raft_ivf_flat: