MS-484 1. update faiss version 2. idmap add search config

Former-commit-id: e9b2131870e0a011a94c6cd2fed8166d9ebd80a5
This commit is contained in:
xj.lin 2019-09-05 11:59:24 +08:00
parent ea0443c84e
commit 6b422840b4
2 changed files with 4 additions and 1 deletions

View File

@ -261,7 +261,7 @@ else()
message(STATUS ${FAISS_SOURCE_URL})
endif()
# set(FAISS_MD5 "a589663865a8558205533c8ac414278c")
set(FAISS_MD5 "31167ecbd1903fec600dc4ac00b9be9e")
set(FAISS_MD5 "57da9c4f599cc8fa4260488b1c96e1cc")
if(DEFINED ENV{KNOWHERE_ARROW_URL})
set(ARROW_SOURCE_URL "$ENV{KNOWHERE_ARROW_URL}")

View File

@ -39,6 +39,9 @@ DatasetPtr IDMAP::Search(const DatasetPtr &dataset, const Config &config) {
}
auto k = config["k"].as<size_t>();
auto metric_type = config["metric_type"].as_string() == "L2" ?
faiss::METRIC_L2 : faiss::METRIC_INNER_PRODUCT;
index_->metric_type = metric_type;
GETTENSOR(dataset)