Current state: ["Under DiscussionDeclined"]
ISSUE: #17754
Keywords: IDMAP, BinaryIDMAP, brute force search, chunk
...
Add new interface only in Knowhere
Code Block | ||
---|---|---|
| ||
/**class knowhereBruteForce wrapper{ API topublic: call faiss brute force search for all metric typesstatic DatasetPtr * * @param metric_type * @param xb training vecors, size nb * dim * @param xq query vecors, size nq * dim * @param dim * @param nb rows of training vectors * @param nq rows of query vectors * @param topk * @param labels output, memory allocated and freed by caller * @param distances output, memory allocated and freed by coller * @param bitset */ void BruteForceSearch( const knowhere::MetricType& metric_type Search(const DatasetPtr base_dataset, const void* xb, const void* xq, const int64_t dimDatasetPtr query_dataset, const int64_t nb, const int64_t nqConfig& config, const int64_t topk, int64_t* labels, float* distances, const faiss::BitsetView bitset); /** knowhere wrapper API to call faiss brute force range search for all metric types * * @param metric_type * @param xb training vecors, size nb * dim * @param xq query vecors, size nq * dim * @param dim * @param nb rows of training vectors * @param nq rows of query vectors * @param radius range search radius * @param labels output, memory allocated inside and freed by caller * @param distances output, memory allocated inside and freed by coller * @param lims output, memory allocated inside and freed by coller * @param bitset */ void BruteForceRangeSearch( const knowhere::MetricType& metric_typestatic DatasetPtr RangeSearch(const DatasetPtr base_dataset, const void* xb, const void* xq, const DatasetPtr const int64_t dim, query_dataset, const int64_t nb, const int64_t nq, const float radius, int64_t*Config& labelsconfig, float*& distances, size_t*& lims, const faiss::BitsetView bitset); }; |
BruteForceSearchBruteForce::Search() is wrapper API to call Faiss brute force search for all metric types. The output parameters "labels" and "distances" are pointers to the memory allocated by caller. Caller will also free these memory after use.BruteForceRangeSearch.
BruteForce::RangeSearch() is wrapper API to call Faiss brute force range search for all metric types. The output parameters "labels", "distances" and "lims" are reference of pointers to the memory allocated in Faiss. This is because the length of range search result is unknown, caller cannot allocate memory in advance. Caller need free these memory after use.
Design Details(required)
For growing segment in segcore, all brute force search related code can be removed, call knowhere's BruteForceSearch() and BruteForceRangeSearch() instead.
...