-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathCargo.toml
66 lines (53 loc) · 1.6 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
[package]
name = "rust-optimal-transport"
version = "0.2.0"
authors = ["kachark <[email protected]>"]
edition = "2018"
repository = "https://github.com/kachark/rust-optimal-transport"
exclude = [
"assets/*",
".github/*",
]
readme = "README.md"
license = "MIT"
description = "A library of optimal transport solvers for Rust"
keywords = ["optimal-transport", "ot", "machine-learning", "optimization", "ai"]
categories = ["algorithms", "mathematics", "science"]
[dependencies]
cxx = "1.0"
thiserror = "1.0"
anyhow = "1.0"
num-traits = "0.2"
ndarray = "0.15.4"
blas-src = { version = "0.8", default-features = false, optional = true }
openblas-src = { version = "0.10", optional = true }
ndarray-linalg = "0.14"
ndarray_einsum_beta = "0.7.0"
ndarray-stats = "0.5"
ndarray-rand = "0.14"
[build-dependencies]
cxx-build = "1.0"
[dev-dependencies]
pyo3 = { version = "0.20", features = ["auto-initialize"] }
numpy = "0.20"
rand = "0.8"
criterion = { version = "0.4", features = ["html_reports"] }
[features]
default = ["ndarray-linalg-openblas-system"]
blas = ["default", "openblas-system"]
# ndarray-linalg LAPACK FFI
ndarray-linalg-openblas-static = ["ndarray-linalg/openblas-static"]
ndarray-linalg-openblas-system = ["ndarray-linalg/openblas-system"]
# Compile openblas
openblas = ["ndarray/blas", "blas-src", "blas-src/openblas", "openblas-src"]
# Pre-compiled/System openblas
openblas-system = ["ndarray/blas", "blas-src", "blas-src/openblas", "openblas-src/cblas", "openblas-src/system"]
[[example]]
name = "emd_2D"
[[example]]
name = "sinkhorn_1D"
[[bench]]
name = "sinkhorn"
harness = false
[profile.bench]
debug = true