1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
/*
* Copyright (c) Meta Platforms, Inc. and 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;
}
|
