DiskANN/diskann-benchmark/Cargo.toml at 57aa5c0d9de2adaccdaaf8a507bf4546bacd10e6 · microsoft/DiskANN · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
[package]
name = "diskann-benchmark"
version.workspace = true
description.workspace = true
authors.workspace = true
repository.workspace = true
license.workspace = true
edition.workspace = true

[dependencies]
anyhow.workspace = true
bf-tree.workspace = true
bit-set.workspace = true
bytemuck.workspace = true
roaring.workspace = true
clap = { workspace = true, features = ["derive"] }
diskann-providers = { workspace = true }
diskann.workspace = true
diskann-utils.workspace = true
half = { workspace = true, features = ["rand_distr", "num-traits"] }
indicatif = "0.18.3"
diskann-quantization = { workspace = true, features = ["linalg"] }
rand.workspace = true
rayon.workspace = true
serde = { workspace = true, features = ["derive"] }
serde_json.workspace = true
thiserror.workspace = true
tokio = { workspace = true, features = ["rt-multi-thread"] }
diskann-vector.workspace = true
diskann-wide.workspace = true
diskann-label-filter.workspace = true
diskann-tools = { workspace = true }
diskann-disk = { workspace = true, optional = true }
diskann-bftree = { workspace = true, optional = true }
cfg-if.workspace = true
diskann-benchmark-runner = { workspace = true }
opentelemetry = { workspace = true, optional = true }
opentelemetry_sdk = { workspace = true, optional = true }
scopeguard = { version = "1.2", optional = true }
diskann-benchmark-core = { workspace = true, features = ["bigann"] }
itertools.workspace = true
diskann-inmem = { workspace = true, optional = true }

[lints]
clippy.undocumented_unsafe_blocks = "warn"
clippy.uninlined_format_args = "allow"

[dev-dependencies]
rand.workspace = true
tempfile.workspace = true
half = { workspace = true, features = ["rand_distr", "num-traits"] }

[features]
default = []

# Enable spherical-quantization based algorithms
spherical-quantization = []

# Enable product-quantization based algorithms
product-quantization = []

# Enable scalar-quantization based algorithms
scalar-quantization = []

# Enable minmax-quantization based algorithms
minmax-quantization = []

# Enable multi-vector MaxSim distance benchmarks
multi-vector = []

# Enable inmem 2.0
inmem2 = ["dep:diskann-inmem"]

# Enable bftree backend
bftree = ["dep:diskann-bftree"]

# Enable Disk Index benchmarks
disk-index = [
    "diskann-disk/perf_test",
    "dep:diskann-disk",
    "dep:opentelemetry",
    "dep:opentelemetry_sdk",
    "dep:scopeguard",
]