102 lines
2.5 KiB
C++
102 lines
2.5 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.
|
|
*/
|
|
|
|
// -*- c++ -*-
|
|
|
|
#include "distances_c.h"
|
|
#include <faiss/utils/distances.h>
|
|
#include <cstdio>
|
|
|
|
void faiss_pairwise_L2sqr(
|
|
int64_t d,
|
|
int64_t nq,
|
|
const float* xq,
|
|
int64_t nb,
|
|
const float* xb,
|
|
float* dis,
|
|
int64_t ldq,
|
|
int64_t ldb,
|
|
int64_t ldd) {
|
|
faiss::pairwise_L2sqr(d, nq, xq, nb, xb, dis, ldq, ldb, ldd);
|
|
}
|
|
|
|
void faiss_pairwise_L2sqr_with_defaults(
|
|
int64_t d,
|
|
int64_t nq,
|
|
const float* xq,
|
|
int64_t nb,
|
|
const float* xb,
|
|
float* dis) {
|
|
faiss::pairwise_L2sqr(d, nq, xq, nb, xb, dis);
|
|
}
|
|
|
|
void faiss_fvec_inner_products_ny(
|
|
float* ip,
|
|
const float* x,
|
|
const float* y,
|
|
size_t d,
|
|
size_t ny) {
|
|
faiss::fvec_inner_products_ny(ip, x, y, d, ny);
|
|
}
|
|
|
|
void faiss_fvec_L2sqr_ny(
|
|
float* dis,
|
|
const float* x,
|
|
const float* y,
|
|
size_t d,
|
|
size_t ny) {
|
|
faiss::fvec_L2sqr_ny(dis, x, y, d, ny);
|
|
}
|
|
|
|
float faiss_fvec_norm_L2sqr(const float* x, size_t d) {
|
|
return faiss::fvec_norm_L2sqr(x, d);
|
|
}
|
|
|
|
void faiss_fvec_norms_L2(float* norms, const float* x, size_t d, size_t nx) {
|
|
faiss::fvec_norms_L2(norms, x, d, nx);
|
|
}
|
|
|
|
void faiss_fvec_norms_L2sqr(float* norms, const float* x, size_t d, size_t nx) {
|
|
faiss::fvec_norms_L2sqr(norms, x, d, nx);
|
|
}
|
|
|
|
void faiss_fvec_renorm_L2(size_t d, size_t nx, float* x) {
|
|
faiss::fvec_renorm_L2(d, nx, x);
|
|
}
|
|
|
|
void faiss_set_distance_compute_blas_threshold(int value) {
|
|
faiss::distance_compute_blas_threshold = value;
|
|
}
|
|
|
|
int faiss_get_distance_compute_blas_threshold() {
|
|
return faiss::distance_compute_blas_threshold;
|
|
}
|
|
|
|
void faiss_set_distance_compute_blas_query_bs(int value) {
|
|
faiss::distance_compute_blas_query_bs = value;
|
|
}
|
|
|
|
int faiss_get_distance_compute_blas_query_bs() {
|
|
return faiss::distance_compute_blas_query_bs;
|
|
}
|
|
|
|
void faiss_set_distance_compute_blas_database_bs(int value) {
|
|
faiss::distance_compute_blas_database_bs = value;
|
|
}
|
|
|
|
int faiss_get_distance_compute_blas_database_bs() {
|
|
return faiss::distance_compute_blas_database_bs;
|
|
}
|
|
|
|
void faiss_set_distance_compute_min_k_reservoir(int value) {
|
|
faiss::distance_compute_min_k_reservoir = value;
|
|
}
|
|
|
|
int faiss_get_distance_compute_min_k_reservoir() {
|
|
return faiss::distance_compute_min_k_reservoir;
|
|
}
|