Faiss

- Main Page
- Namespaces
- Classes
- Files

- File List

All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
L2Select.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 #pragma once
11
12 #include "../utils/Float16.cuh"
13 #include "../utils/Tensor.cuh"
14
15 namespace faiss { namespace gpu {
16
17 void runL2SelectMin(Tensor<float, 2, true>& productDistances,
18 Tensor<float, 1, true>& centroidDistances,
19 Tensor<float, 2, true>& outDistances,
20 Tensor<int, 2, true>& outIndices,
21 int k,
22 cudaStream_t stream);
23
24 #ifdef FAISS_USE_FLOAT16
25 void runL2SelectMin(Tensor<half, 2, true>& productDistances,
26 Tensor<half, 1, true>& centroidDistances,
27 Tensor<half, 2, true>& outDistances,
28 Tensor<int, 2, true>& outIndices,
29 int k,
30 cudaStream_t stream);
31 #endif
32
33 } } // namespace