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