FIX ivf_flat and pq cmake variable underscores

rapidsai · Oct 10, 2023 · bdc8d9a · bdc8d9a
1 parent 8638410
commit bdc8d9a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cpp/bench/ann/CMakeLists.txt b/cpp/bench/ann/CMakeLists.txt
@@ -64,8 +64,8 @@ else()
     set(RAFT_ANN_BENCH_USE_FAISS_GPU_IVF_FLAT OFF)
     set(RAFT_ANN_BENCH_USE_FAISS_GPU_IVF_PQ OFF)
     set(RAFT_ANN_BENCH_USE_FAISS_CPU_FLAT OFF)
-    set(RAFT_ANN_BENCH_USE_FAISS_CPU_IVFPQ OFF)
-    set(RAFT_ANN_BENCH_USE_FAISS_CPU_IFFLAT OFF)
+    set(RAFT_ANN_BENCH_USE_FAISS_CPU_IVF_PQ OFF)
+    set(RAFT_ANN_BENCH_USE_FAISS_CPU_IVF_FLAT OFF)
   else()
     set(RAFT_FAISS_ENABLE_GPU ON)
   endif()
@@ -76,8 +76,8 @@ if(RAFT_ANN_BENCH_USE_FAISS_GPU_FLAT
    OR RAFT_ANN_BENCH_USE_FAISS_GPU_IVF_PQ
    OR RAFT_ANN_BENCH_USE_FAISS_GPU_IVF_FLAT
    OR RAFT_ANN_BENCH_USE_FAISS_CPU_FLAT
-   OR RAFT_ANN_BENCH_USE_FAISS_CPU_IVFPQ
-   OR RAFT_ANN_BENCH_USE_FAISS_CPU_IFFLAT
+   OR RAFT_ANN_BENCH_USE_FAISS_CPU_IVF_PQ
+   OR RAFT_ANN_BENCH_USE_FAISS_CPU_IVF_FLAT
 )
   set(RAFT_ANN_BENCH_USE_FAISS ON)
   set(RAFT_USE_FAISS_STATIC ON)