From ad7141a38c7a94338137aab98fa221f131658e6f Mon Sep 17 00:00:00 2001 From: ramilbakhshyiev <7597197+ramilbakhshyiev@users.noreply.github.com> Date: Thu, 2 May 2024 04:03:41 +0000 Subject: [PATCH] Test gpu build, isolating CMake steps --- CMakeLists.txt | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 3d23cded50..9e06959a12 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)