-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathCargo.toml
53 lines (43 loc) · 1018 Bytes
/
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
[package]
name = "plant"
version = "0.1.0"
authors = ["MashPlant <[email protected]>"]
edition = "2018"
[features]
gpu-runtime = ["plant-runtime/gpu-runtime"]
[dependencies]
plant-macros = { path = "macros" }
plant-runtime = { path = "runtime" }
tools = { git = "https://github.com/MashPlant/tools", features = ["debug-panic"] }
xgboost = { git = "https://github.com/MashPlant/rust-xgboost" }
isl = { path = "isl-rs" }
log = "0.4"
env_logger = "0.8"
tempfile = "3"
scoped_threadpool = "0.1"
byteorder = "1"
[dev-dependencies]
test-case = "1"
[[bin]]
name = "gemm_cpu"
path = "bench/gemm_cpu.rs"
[[bin]]
name = "conv_cpu"
path = "bench/conv_cpu.rs"
[[bin]]
name = "gemm_gpu"
path = "bench/gemm_gpu.rs"
required-features = ["gpu-runtime"]
[[bin]]
name = "conv_gpu"
path = "bench/conv_gpu.rs"
required-features = ["gpu-runtime"]
[[bin]]
path = "bench/resnet.rs"
name = "resnet"
[[bin]]
path = "bench/resnet_aarch64.rs"
name = "resnet_aarch64"
[[bin]]
path = "bench/tune_resnet_conv.rs"
name = "tune_resnet_conv"