forked from jean-pierreBoth/annembed
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Cargo.toml
126 lines (91 loc) · 3.05 KB
/
Cargo.toml
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
[package]
name = "annembed"
version = "0.1.6"
authors = ["[email protected]"]
repository = "https://github.com/jean-pierreBoth/annembed"
description = "a Rust implementation of a dimension reduction à la Umap "
license = "MIT/Apache-2.0"
readme = "README.md"
keywords = ["algorithms", "ann", "embedding", "dimension-reduction", "umap"]
documentation = "https://docs.rs/annembed"
edition = "2021"
[profile.release]
lto = true
opt-level = 3
[lib]
# cargo rustc --lib -- --crate-type dylib [or staticlib] or rlib (default)
# if we want to avoid specifying in advance crate-type
path = "src/lib.rs"
#crate-type = ["cdylib"]
[[examples]]
name = "mnist_digits"
path = "examples/mnist_digits.rs"
[[examples]]
name = "mnist_fashion"
path = "examples/mnist_fashion.rs"
[[examples]]
name = "toripser"
path = "examples/toripser.rs"
[[examples]]
name = "higgs"
path = "examples/higgs.rs"
[[bin]]
name = "annembed"
path = "src/bin/annembed.rs"
[dependencies]
# default is version spec is ^ meaning can update up to max non null version number
# cargo doc --no-deps avoid dependencies doc generation
#
clap = { version = "4.4", features = ["derive"] }
# for //
parking_lot = "0.12"
rayon = { version = "1.10" }
num_cpus = { version = "1.16" }
cpu-time = { version = "1.0" }
ndarray = { version = ">=0.15, <=0.16", features = ["rayon", "serde"] }
# ndarray-linalg = { version = "0.16", default-features = false }
lax = { version = "0.16" }
sprs = { version = "0.11" }
# for macos
blas-src = { version = "0.8", optional = true, features = ["accelerate"] }
indexmap = { version = "2.2" }
memory-stats = { version = "1.1", features = ["always_use_statm"] }
# hnsw_rs = { git = "https://gitlab.com/jpboth/hnswlib-rs.git" }
hnsw_rs = { git = "https://github.com/jean-pierreBoth/hnswlib-rs" }
# hnsw_rs = { path = "../hnswlib-rs" }
# hnsw_rs = { version = "0.3" }
# rand utilis
rand = { version = "0.8" }
rand_distr = { version = "0.4" }
rand_xoshiro = { version = "0.6" }
quantiles = { version = "0.7" }
num-traits = { version = "0.2" }
lazy_static = { version = "1.4" }
# for hubness stats
hdrhistogram = { version = "7.5" }
indxvec = { version = "1.9" }
# for io
csv = { version = "1.3" }
serde = { version = "1.0", features = ["derive"] }
bincode = { version = "1.3" }
byteorder = { version = "1.4" }
bson = { version = "2.10" }
# decreasing order of log for debug build : (max_level_)trace debug info warn error off
# decreasing order of log for release build (release_max_level_) .. idem
#log = { version = "0.4", features = ["max_level_debug", "release_max_level_info"] }
log = { version = "0.4" }
env_logger = { version = "0.11" }
# no more interaction bug with intel-mkl
anyhow = { version = "1.0.58" }
katexit = { version = "0.1" }
[features]
default = []
# simd choice
stdsimd = ["hnsw_rs/stdsimd"]
simdeez_f = ["hnsw_rs/simdeez_f"]
# blas choice
intel-mkl-static = ["ndarray/blas", "lax/intel-mkl-static"]
openblas-static = ["ndarray/blas", "lax/openblas-static"]
openblas-system = ["ndarray/blas", "lax/openblas-system"]
# for macos
macos-accelerate = ["blas-src", "ndarray/blas"]