FIX add cpu targets to CUDA 12 faiss exception

rapidsai · Oct 10, 2023 · 8638410 · 8638410
1 parent daffaf4
commit 8638410
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cpp/bench/ann/CMakeLists.txt b/cpp/bench/ann/CMakeLists.txt
@@ -63,15 +63,18 @@ else()
     set(RAFT_ANN_BENCH_USE_FAISS_GPU_FLAT OFF)
     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)
   else()
     set(RAFT_FAISS_ENABLE_GPU ON)
   endif()
 endif()
 
 set(RAFT_ANN_BENCH_USE_FAISS OFF)
 if(RAFT_ANN_BENCH_USE_FAISS_GPU_FLAT
-   OR RAFT_ANN_BENCH_USE_FAISS_GPU_IVFPQ
-   OR RAFT_ANN_BENCH_USE_FAISS_IFFLAT
+   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