Faiss

- Main Page
- Namespaces
- Classes
- Files

- File List

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