package An official xmake package repository
https://xrepo.xmake.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.5 KiB
35 lines
1.5 KiB
4 years ago
|
package("faiss")
|
||
|
|
|
||
|
|
set_homepage("https://github.com/facebookresearch/faiss/")
|
||
|
|
set_description("A library for efficient similarity search and clustering of dense vectors.")
|
||
|
|
set_license("MIT")
|
||
|
|
|
||
|
|
add_urls("https://github.com/facebookresearch/faiss/archive/refs/tags/$(version).tar.gz",
|
||
|
|
"https://github.com/facebookresearch/faiss.git")
|
||
|
|
add_versions("v1.7.0", "f86d346ac9f409ee30abe37e52f6cce366b7f60d3924d65719f40aa07ceb4bec")
|
||
|
|
|
||
|
|
add_deps("cmake")
|
||
|
|
add_deps("cuda", {system = true, optional = true})
|
||
|
|
add_deps("mkl", "cuda", {system = true, optional = true})
|
||
|
|
if is_plat("linux") then
|
||
|
|
add_syslinks("pthread")
|
||
|
|
end
|
||
|
|
on_load("windows|x64", "linux", function (package)
|
||
|
|
if not find_package("mkl") then
|
||
|
|
package:add("deps", "openblas")
|
||
|
|
end
|
||
|
|
end)
|
||
|
|
|
||
|
|
on_install("windows|x64", "linux", function (package)
|
||
|
|
io.replace("CMakeLists.txt", "add_subdirectory(demos)", "", {plain = true})
|
||
|
|
io.replace("CMakeLists.txt", "add_subdirectory(tutorial/cpp)", "", {plain = true})
|
||
|
|
local configs = {"-DFAISS_ENABLE_PYTHON=OFF", "-DBUILD_TESTING=OFF"}
|
||
|
|
table.insert(configs, "-DCMAKE_BUILD_TYPE=" .. (package:debug() and "Debug" or "Release"))
|
||
|
|
table.insert(configs, "-DFAISS_ENABLE_GPU=" .. (package:dep("cuda"):exists() and "ON" or "OFF"))
|
||
|
|
import("package.tools.cmake").install(package, configs)
|
||
|
|
end)
|
||
|
|
|
||
|
|
on_test(function (package)
|
||
|
|
assert(package:has_cxxtypes("faiss::MultiIndexQuantizer", {configs = {languages = "c++11"}, includes = "faiss/IndexPQ.h"}))
|
||
|
|
end)
|