diff --git a/CMakeLists.txt b/CMakeLists.txt index 3d23cded5..9e06959a1 100644 --- 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)