Faiss

- Main Page
- Namespaces
- Classes
- Files

- File List

All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
BinaryDistance.cuh
1 /**
2 * Copyright (c) 2015-present, Facebook, Inc.
3 * All rights reserved.
4 *
5 * This source code is licensed under the BSD+Patents license found in the
6 * LICENSE file in the root directory of this source tree.
7 */
8
9
10 #include "../utils/DeviceTensor.cuh"
11
12 namespace faiss { namespace gpu {
13
14 // Performs brute-force k-NN comparison between `vecs` and `query`, where they
15 // are encoded as binary vectors
16 void runBinaryDistance(Tensor<unsigned char, 2, true>& vecs,
17 Tensor<unsigned char, 2, true>& query,
18 Tensor<int, 2, true>& outK,
19 Tensor<int, 2, true>& outV,
20 int k, cudaStream_t stream);
21
22 } } // namespace