Faiss

- Main Page
- Namespaces
- Classes
- Files

- File List

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