Faiss

- Main Page
- Namespaces
- Classes
- Files

- File List

All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
L2Select.cuh
1 /**
2 * Copyright (c) Facebook, Inc. and its affiliates.
3 *
4 * This source code is licensed under the MIT license found in the
5 * LICENSE file in the root directory of this source tree.
6 */
7
8
9 #pragma once
10
11 #include "../utils/Float16.cuh"
12 #include "../utils/Tensor.cuh"
13
14 namespace faiss { namespace gpu {
15
16 void runL2SelectMin(Tensor<float, 2, true>& productDistances,
17 Tensor<float, 1, true>& centroidDistances,
18 Tensor<float, 2, true>& outDistances,
19 Tensor<int, 2, true>& outIndices,
20 int k,
21 cudaStream_t stream);
22
23 #ifdef FAISS_USE_FLOAT16
24 void runL2SelectMin(Tensor<half, 2, true>& productDistances,
25 Tensor<half, 1, true>& centroidDistances,
26 Tensor<half, 2, true>& outDistances,
27 Tensor<int, 2, true>& outIndices,
28 int k,
29 cudaStream_t stream);
30 #endif
31
32 } } // namespace