-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Cargo.toml
40 lines (35 loc) · 1.14 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
[package]
name = "ha-ndarray"
version = "0.2.0"
edition = "2021"
authors = ["code@tinychain.net"]
description = "A hardware-accelerated n-dimensional array"
license = "Apache-2.0"
readme = "README.md"
repository = "https://github.com/haydnv/ha-ndarray.git"
keywords = ["ndarray", "tensor", "gpu", "ml", "parallel"]
categories = ["data-structures", "hardware-support", "mathematics"]
[features]
all = ["freqfs", "opencl", "stream"]
opencl = ["ocl"]
stream = ["async-trait", "destream", "futures"]
benchmark = ["ndarray", "ndarray-rand"] # run with `cargo run --bin benchmark --features benchmark`
[build-dependencies]
pkg-config = "0.3"
[dependencies]
async-trait = { version = "0.1", optional = true }
destream = { version = "0.7", optional = true }
futures = { version = "0.3", optional = true }
freqfs = { version = "0.9", optional = true }
get-size = "0.1"
num_cpus = "1.15"
ocl = { version = "0.19", optional = true }
rand = "0.8"
rayon = "1.8"
transpose = "0.2"
ndarray-rand = { version = "0.14.0", optional = true }
ndarray = { version = "0.15.6", optional = true }
[[bin]]
name = "benchmark"
path = "benchmark/benchmark.rs"
required-features = ["benchmark"]