mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
9c4d6262b3
Summary: I want to invoke norm computations by using CGO, but I find some functions which have been implemented in cpp are not exported in c api, so I commit the PR to solve the problem. Pull Request resolved: https://github.com/facebookresearch/faiss/pull/2036 Reviewed By: beauby Differential Revision: D30762172 Pulled By: mdouze fbshipit-source-id: 097b32f29658c1864bd794734daaef0dd75d17ef
97 lines
2.8 KiB
C
97 lines
2.8 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.
|
|
*/
|
|
|
|
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
// -*- c -*-
|
|
|
|
#ifndef FAISS_DISTANCES_C_H
|
|
#define FAISS_DISTANCES_C_H
|
|
|
|
#include <stdlib.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/*********************************************************
|
|
* Optimized distance/norm/inner prod computations
|
|
*********************************************************/
|
|
|
|
/// Compute pairwise distances between sets of vectors
|
|
void faiss_pairwise_L2sqr(
|
|
int64_t d,
|
|
int64_t nq,
|
|
const float* xq,
|
|
int64_t nb,
|
|
const float* xb,
|
|
float* dis,
|
|
int64_t ldq = -1,
|
|
int64_t ldb = -1,
|
|
int64_t ldd = -1);
|
|
|
|
/// compute the inner product between nx vectors x and one y
|
|
void faiss_fvec_inner_products_ny(
|
|
float* ip, /* output inner product */
|
|
const float* x,
|
|
const float* y,
|
|
size_t d,
|
|
size_t ny);
|
|
|
|
/// compute ny square L2 distance between x and a set of contiguous y vectors
|
|
void faiss_fvec_L2sqr_ny(
|
|
float* dis,
|
|
const float* x,
|
|
const float* y,
|
|
size_t d,
|
|
size_t ny);
|
|
|
|
/// squared norm of a vector
|
|
float faiss_fvec_norm_L2sqr(const float* x, size_t d);
|
|
|
|
/// compute the L2 norms for a set of vectors
|
|
void faiss_fvec_norms_L2(float* norms, const float* x, size_t d, size_t nx);
|
|
|
|
/// same as fvec_norms_L2, but computes squared norms
|
|
void faiss_fvec_norms_L2sqr(float* norms, const float* x, size_t d, size_t nx);
|
|
|
|
/// L2-renormalize a set of vector. Nothing done if the vector is 0-normed
|
|
void faiss_fvec_renorm_L2(size_t d, size_t nx, float* x);
|
|
|
|
/// Setter of threshold value on nx above which we switch to BLAS to compute
|
|
/// distances
|
|
void faiss_set_distance_compute_blas_threshold(int value);
|
|
|
|
/// Getter of threshold value on nx above which we switch to BLAS to compute
|
|
/// distances
|
|
int faiss_get_distance_compute_blas_threshold();
|
|
|
|
/// Setter of block sizes value for BLAS distance computations
|
|
void faiss_set_distance_compute_blas_query_bs(int value);
|
|
|
|
/// Getter of block sizes value for BLAS distance computations
|
|
int faiss_get_distance_compute_blas_query_bs();
|
|
|
|
/// Setter of block sizes value for BLAS distance computations
|
|
void faiss_set_distance_compute_blas_database_bs(int value);
|
|
|
|
/// Getter of block sizes value for BLAS distance computations
|
|
int faiss_get_distance_compute_blas_database_bs();
|
|
|
|
/// Setter of number of results we switch to a reservoir to collect results
|
|
/// rather than a heap
|
|
void faiss_set_distance_compute_min_k_reservoir(int value);
|
|
|
|
/// Getter of number of results we switch to a reservoir to collect results
|
|
/// rather than a heap
|
|
int faiss_get_distance_compute_min_k_reservoir();
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|