Test gpu build, isolating CMake steps

facebookresearch · May 2, 2024 · ad7141a · ad7141a
1 parent e123323
commit ad7141a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -57,10 +57,10 @@ option(FAISS_ENABLE_RAFT "Enable RAFT for GPU indexes." OFF)
 option(FAISS_ENABLE_PYTHON "Build Python extension." ON)
 option(FAISS_ENABLE_C_API "Build C API." OFF)
 
-#if(FAISS_ENABLE_GPU)
-#  set(CMAKE_CUDA_HOST_COMPILER ${CMAKE_CXX_COMPILER})
-#  enable_language(CUDA)
-#endif()
+if(FAISS_ENABLE_GPU)
+  set(CMAKE_CUDA_HOST_COMPILER ${CMAKE_CXX_COMPILER})
+  enable_language(CUDA)
+endif()
 
 if(FAISS_ENABLE_RAFT AND NOT TARGET raft::raft)
    find_package(raft COMPONENTS compiled distributed)