Faiss

- Main Page
- Namespaces
- Classes
- Files

- File List

All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
GpuClonerOptions.h
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 #pragma once
9
10 #include "GpuIndicesOptions.h"
11
12 namespace faiss { namespace gpu {
13
14 /// set some options on how to copy to GPU
15 struct GpuClonerOptions {
16 GpuClonerOptions();
17
18 /// how should indices be stored on index types that support indices
19 /// (anything but GpuIndexFlat*)?
20 IndicesOptions indicesOptions;
21
22 /// is the coarse quantizer in float16?
23 bool useFloat16CoarseQuantizer;
24
25 /// for GpuIndexIVFFlat, is storage in float16?
26 /// for GpuIndexIVFPQ, are intermediate calculations in float16?
27 bool useFloat16;
28
29 /// use precomputed tables?
30 bool usePrecomputed;
31
32 /// reserve vectors in the invfiles?
33 long reserveVecs;
34
35 /// For GpuIndexFlat, store data in transposed layout?
36 bool storeTransposed;
37
38 /// Set verbose options on the index
39 bool verbose;
40 };
41
42 struct GpuMultipleClonerOptions : public GpuClonerOptions {
43 GpuMultipleClonerOptions ();
44
45 /// Whether to shard the index across GPUs, versus replication
46 /// across GPUs
47 bool shard;
48
49 /// IndexIVF::copy_subset_to subset type
50 int shard_type;
51 };
52
53 } } // namespace
faiss::gpu::GpuMultipleClonerOptions::shard
bool shard
Definition: GpuClonerOptions.h:47
faiss::gpu::GpuClonerOptions::storeTransposed
bool storeTransposed
For GpuIndexFlat, store data in transposed layout?
Definition: GpuClonerOptions.h:36
faiss::gpu::GpuClonerOptions
set some options on how to copy to GPU
Definition: GpuClonerOptions.h:15
faiss::gpu::GpuClonerOptions::usePrecomputed
bool usePrecomputed
use precomputed tables?
Definition: GpuClonerOptions.h:30
faiss::gpu::GpuMultipleClonerOptions
Definition: GpuClonerOptions.h:42
faiss::gpu::GpuClonerOptions::useFloat16
bool useFloat16
Definition: GpuClonerOptions.h:27
faiss::gpu::GpuClonerOptions::indicesOptions
IndicesOptions indicesOptions
Definition: GpuClonerOptions.h:20
faiss::gpu::GpuMultipleClonerOptions::shard_type
int shard_type
IndexIVF::copy_subset_to subset type.
Definition: GpuClonerOptions.h:50
faiss::gpu::GpuClonerOptions::useFloat16CoarseQuantizer
bool useFloat16CoarseQuantizer
is the coarse quantizer in float16?
Definition: GpuClonerOptions.h:23
faiss::gpu::GpuClonerOptions::verbose
bool verbose
Set verbose options on the index.
Definition: GpuClonerOptions.h:39
faiss::gpu::GpuClonerOptions::reserveVecs
long reserveVecs
reserve vectors in the invfiles?
Definition: GpuClonerOptions.h:33