-
Notifications
You must be signed in to change notification settings - Fork 2
/
Cargo.toml
47 lines (38 loc) · 1.41 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
[package]
name = "burn-benches"
version = "0.1.0"
authors = ["nathanielsimard <[email protected]>"]
description = "Benchmarks for burn"
edition = "2021"
[features]
default = ["ndarray", "burn/std", "burn/autodiff", "burn/dataset"]
__benches = ["burn/std", "burn/autodiff", "burn/dataset"]
tch-cpu = ["burn-tch", "__benches"]
tch-gpu = ["burn-tch", "__benches"]
wgpu = ["burn-wgpu", "__benches"]
ndarray = ["burn-ndarray/std", "burn/autodiff", "burn/std"]
ndarray-blas-netlib = ["burn-ndarray/std", "burn-ndarray/blas-netlib"]
ndarray-blas-openblas = ["burn-ndarray/std", "burn-ndarray/blas-openblas"]
ndarray-no-std = ["burn-ndarray"]
[dependencies]
criterion = { version = "0.5", html_reports = true }
burn = { git = "https://github.com/burn-rs/burn/", branch = "main", default-features = false }
# Backends
burn-tch = { git = "https://github.com/burn-rs/burn/", branch = "main", optional = true }
burn-wgpu = { git = "https://github.com/burn-rs/burn/", branch = "main", optional = true }
burn-ndarray = { git = "https://github.com/burn-rs/burn/", branch = "main", optional = true, default-features = false }
derive-new = "0.5.9"
serde = { version = "1.0.151", features = ["derive"] }
sysinfo = "0.28.0"
nvml-wrapper = "0.8.0"
clap = { version = "4.1.6", features = ["derive"] }
ctrlc = "3.2.5"
[[bench]]
name = "mlp"
harness = false
[[bench]]
name = "transformer"
harness = false
[[bench]]
name = "conv2d"
harness = false