mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
1dc992bf26
Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/2682 IndexShards normally sees the indexes as opaque, so there is no way to factrorize the coarse quantizer. This diff introduces IndexIVFShards that handles IVF indexes with a common quantizer so that the quantization is computed only once. Reviewed By: alexanderguzhva Differential Revision: D42781513 fbshipit-source-id: 441316eff4c1ba0468501c456af9194ea5f042d6
63 lines
1.6 KiB
C++
63 lines
1.6 KiB
C++
/**
|
|
* 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.
|
|
*/
|
|
|
|
#include "IndexReplicas_c.h"
|
|
#include <faiss/IndexReplicas.h>
|
|
#include "macros_impl.h"
|
|
|
|
using faiss::Index;
|
|
using faiss::IndexReplicas;
|
|
|
|
DEFINE_DESTRUCTOR(IndexReplicas)
|
|
|
|
DEFINE_GETTER(IndexReplicas, int, own_indices)
|
|
DEFINE_SETTER(IndexReplicas, int, own_indices)
|
|
|
|
int faiss_IndexReplicas_new(FaissIndexReplicas** p_index, idx_t d) {
|
|
try {
|
|
auto out = new IndexReplicas(d);
|
|
*p_index = reinterpret_cast<FaissIndexReplicas*>(out);
|
|
}
|
|
CATCH_AND_HANDLE
|
|
}
|
|
|
|
int faiss_IndexReplicas_new_with_options(
|
|
FaissIndexReplicas** p_index,
|
|
idx_t d,
|
|
int threaded) {
|
|
try {
|
|
auto out = new IndexReplicas(d, static_cast<bool>(threaded));
|
|
*p_index = reinterpret_cast<FaissIndexReplicas*>(out);
|
|
}
|
|
CATCH_AND_HANDLE
|
|
}
|
|
|
|
int faiss_IndexReplicas_add_replica(
|
|
FaissIndexReplicas* index,
|
|
FaissIndex* replica) {
|
|
try {
|
|
reinterpret_cast<IndexReplicas*>(index)->add_replica(
|
|
reinterpret_cast<Index*>(replica));
|
|
}
|
|
CATCH_AND_HANDLE
|
|
}
|
|
|
|
int faiss_IndexReplicas_remove_replica(
|
|
FaissIndexReplicas* index,
|
|
FaissIndex* replica) {
|
|
try {
|
|
reinterpret_cast<IndexReplicas*>(index)->remove_replica(
|
|
reinterpret_cast<Index*>(replica));
|
|
}
|
|
CATCH_AND_HANDLE
|
|
}
|
|
|
|
FaissIndex* faiss_IndexReplicas_at(FaissIndexReplicas* index, int i) {
|
|
auto replica = reinterpret_cast<IndexReplicas*>(index)->at(i);
|
|
return reinterpret_cast<FaissIndex*>(replica);
|
|
}
|