mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
1ea134eb83
Summary: This diff is related to https://github.com/facebookresearch/faiss/issues/1722 File structure with `-DFAISS_ENABLE_GPU=OFF` ``` /usr/local/include/faiss/c_api ├── AutoTune_c.h ├── clone_index_c.h ├── Clustering_c.h ├── error_c.h ├── error_impl.h ├── faiss_c.h ├── impl │ └── AuxIndexStructures_c.h ├── Index_c.h ├── index_factory_c.h ├── IndexFlat_c.h ├── index_io_c.h ├── IndexIVF_c.h ├── IndexIVFFlat_c.h ├── IndexLSH_c.h ├── IndexPreTransform_c.h ├── IndexScalarQuantizer_c.h ├── IndexShards_c.h ├── macros_impl.h ├── MetaIndexes_c.h └── VectorTransform_c.h ``` File structure with `-DFAISS_ENABLE_GPU=ON` ``` /usr/local/include/faiss/c_api ├── AutoTune_c.h ├── clone_index_c.h ├── Clustering_c.h ├── error_c.h ├── error_impl.h ├── faiss_c.h ├── gpu │ ├── DeviceUtils_c.h │ ├── GpuAutoTune_c.h │ ├── GpuClonerOptions_c.h │ ├── GpuIndex_c.h │ ├── GpuIndicesOptions_c.h │ ├── GpuResources_c.h │ ├── macros_impl.h │ └── StandardGpuResources_c.h ├── impl │ └── AuxIndexStructures_c.h ├── Index_c.h ├── index_factory_c.h ├── IndexFlat_c.h ├── index_io_c.h ├── IndexIVF_c.h ├── IndexIVFFlat_c.h ├── IndexLSH_c.h ├── IndexPreTransform_c.h ├── IndexScalarQuantizer_c.h ├── IndexShards_c.h ├── macros_impl.h ├── MetaIndexes_c.h └── VectorTransform_c.h ``` Pull Request resolved: https://github.com/facebookresearch/faiss/pull/1841 Reviewed By: mdouze Differential Revision: D27992822 Pulled By: beauby fbshipit-source-id: 63fa9a39c77502d138453cb4b04c50652e732196
23 lines
701 B
CMake
23 lines
701 B
CMake
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
target_sources(faiss_c PRIVATE
|
|
DeviceUtils_c.cpp
|
|
GpuAutoTune_c.cpp
|
|
GpuClonerOptions_c.cpp
|
|
GpuIndex_c.cpp
|
|
GpuResources_c.cpp
|
|
StandardGpuResources_c.cpp
|
|
)
|
|
|
|
file(GLOB FAISS_C_API_GPU_HEADERS RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} "*.h")
|
|
faiss_install_headers("${FAISS_C_API_GPU_HEADERS}" c_api/gpu)
|
|
|
|
find_package(CUDAToolkit REQUIRED)
|
|
target_link_libraries(faiss_c PUBLIC CUDA::cudart CUDA::cublas)
|
|
|
|
add_executable(example_gpu_c EXCLUDE_FROM_ALL example_gpu_c.c)
|
|
target_link_libraries(example_gpu_c PRIVATE faiss_c)
|