2022-09-05 02:22:56 +08:00
|
|
|
[workspace]
|
2023-11-19 23:35:03 +08:00
|
|
|
# Try
|
2023-02-21 21:35:24 +08:00
|
|
|
# require version 2 to avoid "feature" additiveness for dev-dependencies
|
|
|
|
|
# https://doc.rust-lang.org/cargo/reference/resolver.html#feature-resolver-version-2
|
|
|
|
|
resolver = "2"
|
|
|
|
|
|
2022-09-05 02:22:56 +08:00
|
|
|
members = [
|
2024-02-21 02:57:55 +08:00
|
|
|
"backend-comparison",
|
|
|
|
|
"crates/*",
|
2024-02-23 05:39:20 +08:00
|
|
|
"crates/burn-import/pytorch-tests",
|
|
|
|
|
"crates/burn-import/onnx-tests",
|
2023-10-05 22:29:13 +08:00
|
|
|
"examples/*",
|
2024-01-25 23:20:09 +08:00
|
|
|
"examples/pytorch-import/model",
|
2024-02-21 02:57:55 +08:00
|
|
|
"xtask",
|
2022-09-05 02:22:56 +08:00
|
|
|
]
|
2023-02-25 22:38:01 +08:00
|
|
|
|
2024-05-01 19:33:02 +08:00
|
|
|
exclude = [
|
|
|
|
|
"examples/notebook",
|
2024-05-24 23:31:47 +08:00
|
|
|
# "crates/burn-cuda" # comment this line to work on burn-cuda
|
2024-05-01 19:33:02 +08:00
|
|
|
]
|
2023-08-17 06:17:12 +08:00
|
|
|
|
2023-12-12 22:39:07 +08:00
|
|
|
[workspace.package]
|
|
|
|
|
edition = "2021"
|
2024-05-29 06:12:44 +08:00
|
|
|
version = "0.14.0"
|
2023-12-12 22:39:07 +08:00
|
|
|
readme = "README.md"
|
|
|
|
|
license = "MIT OR Apache-2.0"
|
|
|
|
|
|
2023-02-25 22:38:01 +08:00
|
|
|
[workspace.dependencies]
|
2024-06-25 21:23:36 +08:00
|
|
|
bytemuck = "1.16.1"
|
2024-06-21 20:55:27 +08:00
|
|
|
candle-core = { version = "0.5.1" }
|
2024-07-01 22:02:29 +08:00
|
|
|
clap = { version = "4.5.8", features = ["derive"] }
|
2024-04-04 03:41:35 +08:00
|
|
|
colored = "2.1.0"
|
2024-01-25 23:20:09 +08:00
|
|
|
console_error_panic_hook = "0.1.7"
|
2023-10-10 02:29:44 +08:00
|
|
|
csv = "1.3.0"
|
|
|
|
|
dashmap = "5.5.3"
|
2024-06-25 21:23:10 +08:00
|
|
|
data-encoding = { version = "2.6.0", default-features = false, features = [
|
|
|
|
|
"alloc",
|
|
|
|
|
] }
|
2023-05-21 21:07:39 +08:00
|
|
|
dirs = "5.0.1"
|
2024-05-11 04:25:19 +08:00
|
|
|
fake = "2.9.2"
|
|
|
|
|
flate2 = "1.0.30"
|
2023-08-05 04:51:51 +08:00
|
|
|
float-cmp = "0.9.0"
|
2024-05-11 04:25:19 +08:00
|
|
|
getrandom = { version = "0.2.15", default-features = false }
|
|
|
|
|
gix-tempfile = { version = "13.1.1", features = ["signals"] }
|
2024-02-03 05:32:38 +08:00
|
|
|
globwalk = "0.9.1"
|
2024-04-29 19:52:56 +08:00
|
|
|
hashbrown = "0.14.5"
|
2024-06-25 21:23:10 +08:00
|
|
|
hound = "3.5.1"
|
|
|
|
|
image = "0.25.1"
|
2024-02-12 23:42:15 +08:00
|
|
|
indicatif = "0.17.8"
|
2024-03-12 13:26:32 +08:00
|
|
|
js-sys = "0.3.69"
|
2023-10-10 02:29:44 +08:00
|
|
|
libm = "0.2.8"
|
2024-07-01 22:02:29 +08:00
|
|
|
log = { default-features = false, version = "0.4.22" }
|
2024-06-25 21:23:10 +08:00
|
|
|
md5 = "0.7.0"
|
|
|
|
|
percent-encoding = "2.3.1"
|
2024-05-11 04:25:19 +08:00
|
|
|
pretty_assertions = "1.4.0"
|
2024-06-25 21:23:36 +08:00
|
|
|
proc-macro2 = "1.0.86"
|
2024-05-11 04:25:19 +08:00
|
|
|
protobuf = "3.4.0"
|
|
|
|
|
protobuf-codegen = "3.4.0"
|
|
|
|
|
quote = "1.0.36"
|
2023-05-21 02:24:55 +08:00
|
|
|
r2d2 = "0.8.10"
|
2024-05-11 04:25:19 +08:00
|
|
|
r2d2_sqlite = { version = "0.24.0" }
|
2024-03-25 22:14:22 +08:00
|
|
|
rayon = "1.10.0"
|
2024-06-17 21:06:01 +08:00
|
|
|
regex = "1.10.5"
|
2024-07-01 22:02:29 +08:00
|
|
|
reqwest = "0.12.5"
|
2024-05-06 22:20:31 +08:00
|
|
|
rmp-serde = "1.3.0"
|
2024-05-11 04:25:19 +08:00
|
|
|
rstest = "0.19.0"
|
|
|
|
|
rusqlite = { version = "0.31.0" }
|
2024-01-25 23:20:09 +08:00
|
|
|
rust-format = { version = "0.3.4" }
|
2023-08-01 20:30:49 +08:00
|
|
|
sanitize-filename = "0.5.0"
|
2024-07-11 19:48:43 +08:00
|
|
|
serde_bytes = { version = "0.11.15", default-features = false, features = ["alloc"] } # alloc for no_std
|
2024-05-11 04:25:19 +08:00
|
|
|
serde_rusqlite = "0.35.0"
|
2024-06-25 21:23:10 +08:00
|
|
|
serial_test = "3.1.1"
|
2023-09-19 07:56:53 +08:00
|
|
|
spin = { version = "0.9.8", features = ["mutex", "spin_mutex"] }
|
2024-06-25 21:23:36 +08:00
|
|
|
strum = "0.26.3"
|
2024-06-11 04:22:08 +08:00
|
|
|
strum_macros = "0.26.4"
|
2024-07-09 03:54:54 +08:00
|
|
|
syn = { version = "2.0.69", features = ["full", "extra-traits"] }
|
2024-03-09 08:27:48 +08:00
|
|
|
tempfile = "3.10.1"
|
2024-05-22 21:08:21 +08:00
|
|
|
thiserror = "1.0.61"
|
2024-06-17 21:06:01 +08:00
|
|
|
tokio = { version = "1.38.0", features = ["rt", "macros"] }
|
2023-11-21 22:28:56 +08:00
|
|
|
tracing-appender = "0.2.3"
|
|
|
|
|
tracing-core = "0.1.32"
|
|
|
|
|
tracing-subscriber = "0.3.18"
|
2024-03-09 08:27:48 +08:00
|
|
|
web-time = "1.1.0"
|
2024-06-11 04:22:08 +08:00
|
|
|
zip = "2.1.3"
|
2023-11-21 22:28:56 +08:00
|
|
|
|
2024-07-02 20:25:28 +08:00
|
|
|
# Async handling
|
|
|
|
|
pollster = "0.3"
|
|
|
|
|
async-channel = "2.3"
|
|
|
|
|
|
2024-02-01 00:26:09 +08:00
|
|
|
# Terminal UI
|
2024-05-27 22:04:27 +08:00
|
|
|
ratatui = "0.26.3"
|
2024-05-11 04:25:19 +08:00
|
|
|
crossterm = "0.27.0"
|
2023-06-02 23:52:47 +08:00
|
|
|
|
|
|
|
|
# WGPU stuff
|
2024-07-09 03:54:54 +08:00
|
|
|
text_placeholder = "0.5.1"
|
2024-06-25 21:23:36 +08:00
|
|
|
wgpu = "0.20.1"
|
2023-06-15 23:59:37 +08:00
|
|
|
|
2024-03-27 03:58:38 +08:00
|
|
|
# Benchmarks and Burnbench
|
2024-05-11 04:25:19 +08:00
|
|
|
arboard = "3.4.0"
|
2024-02-14 00:16:53 +08:00
|
|
|
github-device-flow = "0.2.0"
|
2024-03-27 03:58:38 +08:00
|
|
|
os_info = "3.8.2"
|
|
|
|
|
wsl = "0.1.0"
|
2024-02-14 00:16:53 +08:00
|
|
|
|
2023-09-19 07:56:53 +08:00
|
|
|
bincode = { version = "2.0.0-rc.3", features = [
|
2023-10-05 22:29:13 +08:00
|
|
|
"alloc",
|
|
|
|
|
"serde",
|
2023-09-19 07:56:53 +08:00
|
|
|
], default-features = false }
|
2024-01-25 23:20:09 +08:00
|
|
|
|
|
|
|
|
#
|
|
|
|
|
# The following packages disable the "std" feature for no_std compatibility
|
|
|
|
|
#
|
|
|
|
|
derive-new = { version = "0.6.0", default-features = false }
|
2023-10-05 22:29:13 +08:00
|
|
|
|
2024-04-10 03:01:58 +08:00
|
|
|
half = { version = "2.4.1", features = [
|
2023-10-05 22:29:13 +08:00
|
|
|
"alloc",
|
|
|
|
|
"num-traits",
|
|
|
|
|
"serde",
|
2023-09-19 07:56:53 +08:00
|
|
|
], default-features = false }
|
|
|
|
|
ndarray = { version = "0.15.6", default-features = false }
|
2024-02-01 03:47:02 +08:00
|
|
|
matrixmultiply = { version = "0.3.8", default-features = false }
|
2024-02-06 04:48:34 +08:00
|
|
|
openblas-src = "0.10.9"
|
2024-05-11 04:25:19 +08:00
|
|
|
blas-src = { version = "0.10.0", default-features = false }
|
2024-05-06 22:20:31 +08:00
|
|
|
num-traits = { version = "0.2.19", default-features = false, features = [
|
2023-10-05 22:29:13 +08:00
|
|
|
"libm",
|
2023-09-19 07:56:53 +08:00
|
|
|
] } # libm is for no_std
|
|
|
|
|
rand = { version = "0.8.5", default-features = false, features = [
|
2023-10-05 22:29:13 +08:00
|
|
|
"std_rng",
|
2024-02-20 23:52:36 +08:00
|
|
|
] } # std_rng is for no_std
|
2023-09-19 07:56:53 +08:00
|
|
|
rand_distr = { version = "0.4.3", default-features = false }
|
2024-07-09 03:54:54 +08:00
|
|
|
serde = { version = "1.0.204", default-features = false, features = [
|
2023-10-05 22:29:13 +08:00
|
|
|
"derive",
|
|
|
|
|
"alloc",
|
2023-09-19 07:56:53 +08:00
|
|
|
] } # alloc is for no_std, derive is needed
|
2024-07-01 22:02:29 +08:00
|
|
|
serde_json = { version = "1.0.119", default-features = false }
|
2024-07-01 22:43:37 +08:00
|
|
|
uuid = { version = "1.9.1", default-features = false }
|
2023-09-30 03:13:16 +08:00
|
|
|
|
2024-05-27 22:04:27 +08:00
|
|
|
libc = "0.2.155"
|
2024-02-01 03:47:02 +08:00
|
|
|
tch = "0.15.0"
|
2024-05-11 04:25:19 +08:00
|
|
|
nvml-wrapper = "0.10.0"
|
|
|
|
|
sysinfo = "0.30.12"
|
2024-02-01 03:47:02 +08:00
|
|
|
systemstat = "0.2.3"
|
|
|
|
|
|
|
|
|
|
|
2023-09-30 03:13:16 +08:00
|
|
|
[profile.dev]
|
2024-01-23 22:50:51 +08:00
|
|
|
debug = 0 # Speed up compilation time and not necessary.
|
