mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
80314d9f07
* add initial cmake support * update cmake, add cmake instructions to INSTALL * update findopenmp and INSTALL * change FindOpenBLAS.cmake to cater for macports - change cblas.h to openblas_config.h since macports does not ship cblas.h with openblas. * revise INSTALL for cmake
20 lines
675 B
CMake
20 lines
675 B
CMake
# configure cuda
|
|
|
|
find_package(CUDA QUIET REQUIRED)
|
|
if(CUDA_FOUND)
|
|
include_directories(SYSTEM ${CUDA_INCLUDE_DIRS})
|
|
list(APPEND CUDA_LINKER_LIBS ${CUDA_CUDART_LIBRARY} ${CUDA_curand_LIBRARY} ${CUDA_CUBLAS_LIBRARIES})
|
|
else(CUDA_FOUND)
|
|
message(STATUS "Could not locate cuda, disabling cuda support.")
|
|
set(BUILD_WITH_GPU OFF)
|
|
return()
|
|
endif(CUDA_FOUND)
|
|
|
|
# set cuda flags
|
|
if (CMAKE_BUILD_TYPE STREQUAL "Debug")
|
|
list(APPEND CUDA_NVCC_FLAGS "-arch=sm_35;-std=c++11;-DVERBOSE;-g;-lineinfo;-Xcompiler;-ggdb")
|
|
else()
|
|
list(APPEND CUDA_NVCC_FLAGS "-arch=sm_35;-std=c++11;-DVERBOSE;-O3;-DNDEBUG;-Xcompiler;-DNDEBU")
|
|
endif()
|
|
set(CUDA_PROPAGATE_HOST_FLAGS OFF)
|