Skip to content

Add option to enable "sve" optimization level on armv9 #1121

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
LizYou wants to merge 10 commits into
base: main
Choose a base branch
from
Open

Add option to enable "sve" optimization level on armv9 #1121

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
99bd964
Add option to enable "sve" optimization level on armv9
LizYou Jul 16, 2025
fb9fe63
Merge branch 'branch-25.08' into faiss_sve_new
cjnolet Jul 22, 2025
123642e
Fix binary error
cjnolet Jul 22, 2025
59d0bfd
Merge branch 'faiss_sve_new' of github.com:LizYou/cuvs into faiss_sve…
cjnolet Jul 22, 2025
0c525d5
Merge branch 'branch-25.08' into faiss_sve_new
cjnolet Jul 31, 2025
13ae755
Update binary.cuh
cjnolet Aug 6, 2025
c82d54e
Merge branch 'branch-25.10' into faiss_sve_new
cjnolet Aug 6, 2025
09d8323
Merge branch 'branch-25.10' into faiss_sve_new
lowener Sep 23, 2025
81ffd6b
Merge branch 'branch-25.10' into faiss_sve_new
cjnolet Jan 30, 2026
cf803ed
Merge branch 'main' into faiss_sve_new
cjnolet Jan 30, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 20 additions & 3 deletions cpp/cmake/thirdparty/get_faiss.cmake
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
#=============================================================================

function(find_and_configure_faiss)
option(CUVS_CPU_ONLY_FAISS_ENABLE_SVE "Enable SVE support for CPU ONLY FAISS" OFF)
set(oneValueArgs VERSION REPOSITORY PINNED_TAG BUILD_STATIC_LIBS EXCLUDE_FROM_ALL ENABLE_GPU)
cmake_parse_arguments(PKG "${options}" "${oneValueArgs}"
"${multiValueArgs}" ${ARGN} )
Expand Down Expand Up @@ -33,16 +34,24 @@ function(find_and_configure_faiss)
endif()

include(${CMAKE_CURRENT_FUNCTION_LIST_DIR}/../modules/FindAVX.cmake)
include(${CMAKE_CURRENT_FUNCTION_LIST_DIR}/../modules/FindSVE.cmake)
# Link against AVX CPU lib if it exists
set(CUVS_FAISS_OPT_LEVEL "generic")
if(CXX_AVX2_FOUND)
set(CUVS_FAISS_OPT_LEVEL "avx2")
endif()
if(CXX_SVE_FOUND AND CUVS_CPU_ONLY_FAISS_ENABLE_SVE)
set(CUVS_FAISS_OPT_LEVEL "sve")
endif()


rapids_cpm_find(faiss ${version} ${find_args}
GLOBAL_TARGETS faiss faiss_avx2 faiss_gpu_objs faiss::faiss faiss::faiss_avx2
CPM_ARGS ${cpm_args}
rapids_cpm_find(faiss ${version} ${build_patch_only}
GLOBAL_TARGETS faiss faiss_avx2 faiss_sve faiss_gpu_objs faiss::faiss faiss::faiss_avx2 faiss::faiss_sve
CPM_ARGS
GIT_REPOSITORY ${repository}
GIT_TAG ${tag}
GIT_SHALLOW ${shallow} ${patch_command}
EXCLUDE_FROM_ALL ${exclude}
OPTIONS
"FAISS_ENABLE_GPU ${PKG_ENABLE_GPU}"
"FAISS_ENABLE_CUVS ${PKG_ENABLE_GPU}"
Expand All @@ -69,6 +78,12 @@ function(find_and_configure_faiss)
# faiss will have the conda includes/link dirs
target_link_libraries(faiss_avx2 PRIVATE $<TARGET_NAME_IF_EXISTS:conda_env>)
endif()
if(TARGET faiss_sve AND NOT TARGET faiss::faiss_sve)
add_library(faiss::faiss_sve ALIAS faiss_sve)
# We need to ensure that faiss has all the conda information. So we use this approach so that
# faiss will have the conda includes/link dirs
target_link_libraries(faiss_sve PRIVATE $<TARGET_NAME_IF_EXISTS:conda_env>)
endif()
if(TARGET faiss_gpu_objs AND NOT TARGET faiss::faiss_gpu_objs)
add_library(faiss::faiss_gpu_objs ALIAS faiss_gpu_objs)
# We need to ensure that faiss has all the conda information. So we use this approach so that
Expand Down Expand Up @@ -97,6 +112,8 @@ function(find_and_configure_faiss)
set(CUVS_FAISS_TARGETS "$<LINK_GROUP:RESCAN,$<LINK_LIBRARY:WHOLE_ARCHIVE,faiss_gpu_objs>,faiss::faiss>" PARENT_SCOPE)
elseif(CXX_AVX2_FOUND)
set(CUVS_FAISS_TARGETS faiss::faiss_avx2 PARENT_SCOPE)
elseif(CXX_SVE_FOUND AND CUVS_CPU_ONLY_FAISS_ENABLE_SVE)
set(CUVS_FAISS_TARGETS faiss::faiss_sve PARENT_SCOPE)
else()
set(CUVS_FAISS_TARGETS faiss::faiss PARENT_SCOPE)
endif()
Expand Down
Loading