Faiss

- Main Page
- Namespaces
- Classes
- Files

- File List

All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
L2Norm.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 runL2Norm(Tensor<float, 2, true>& input,
20 Tensor<float, 1, true>& output,
21 bool normSquared,
22 cudaStream_t stream);
23
24 #ifdef FAISS_USE_FLOAT16
25 void runL2Norm(Tensor<half, 2, true>& input,
26 Tensor<half, 1, true>& output,
27 bool normSquared,
28 cudaStream_t stream);
29 #endif
30
31 } } // namespace