From 945355d707d60c2e507899f262e87eb00d8624d8 Mon Sep 17 00:00:00 2001
From: Divye Gala <divyegala@gmail.com>
Date: Fri, 20 Oct 2023 00:26:06 -0400
Subject: [PATCH] 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: https://github.com/rapidsai/raft/pull/1916
---
 python/raft-ann-bench/src/raft-ann-bench/plot/__main__.py | 4 +++-
 python/raft-ann-bench/src/raft-ann-bench/run/algos.yaml   | 6 +++---
 2 files changed, 6 insertions(+), 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
index 233607c281..6ec2cdaf22 100644
--- 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
index f4f928505f..7ea360e0c9 100644
--- 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: