dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#1747 Upgrade Rust dependencies (#1748) 

* #1747
Upgrade Rust dependencies

* Revert upgrade for tch

The update of tch on windows gives an error:

INTEL MKL ERROR: The specified module could not be found. mkl_vml_avx2.1.dll.
Intel MKL FATAL ERROR: cannot load mkl_vml_avx2.1.dll or mkl_vml_def.1.dll.

* Keep only .cargo/config.toml file which works with rust > 1.75

---------

Co-authored-by: Sylvain Benner <sylvain@benner.online>
This commit is contained in:
Ahmed Yarub Hani Al Nuaimi 2024-05-10 17:25:19 -03:00 committed by GitHub
parent b09d8431df
commit 10737527d8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
27 changed files with 779 additions and 292 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
.cargo/config → .cargo/config.toml
View File

1
.gitignore vendored
View File

@ -2,7 +2,6 @@ target
# These are backup files generated by rustfmt # These are backup files generated by rustfmt
**/*.rs.bk **/*.rs.bk
.DS_Store .DS_Store
.cargo/config.toml
.dir-locals.el .dir-locals.el
.idea .idea

858
Cargo.lock generated
View File

File diff suppressed because it is too large Load Diff

68
Cargo.toml
View File

@ -21,61 +21,61 @@ exclude = [
[workspace.package] [workspace.package]
edition = "2021" edition = "2021"
version = "0.14.0" version = "0.15.0"
readme = "README.md" readme = "README.md"
license = "MIT OR Apache-2.0" license = "MIT OR Apache-2.0"
[workspace.dependencies] [workspace.dependencies]
async-trait = "0.1.80" async-trait = "0.1.80"
bytemuck = "1.15" bytemuck = "1.15.0"
# candle-core = { version = "0.4.1" } # candle-core = { version = "0.4.1" }
candle-core = { git = "https://github.com/huggingface/candle.git", rev = "2817643" } candle-core = { git = "https://github.com/huggingface/candle.git", rev = "82b641f" }
clap = { version = "4.5.4", features = ["derive"] } clap = { version = "4.5.4", features = ["derive"] }
colored = "2.1.0" colored = "2.1.0"
console_error_panic_hook = "0.1.7" console_error_panic_hook = "0.1.7"
csv = "1.3.0" csv = "1.3.0"
dashmap = "5.5.3" dashmap = "5.5.3"
dirs = "5.0.1" dirs = "5.0.1"
fake = "2.9.1" fake = "2.9.2"
flate2 = "1.0.29" flate2 = "1.0.30"
float-cmp = "0.9.0" float-cmp = "0.9.0"
getrandom = { version = "0.2.14", default-features = false } getrandom = { version = "0.2.15", default-features = false }
gix-tempfile = { version = "11.0.0", features = ["signals"] } gix-tempfile = { version = "13.1.1", features = ["signals"] }
globwalk = "0.9.1" globwalk = "0.9.1"
hashbrown = "0.14.5" hashbrown = "0.14.5"
indicatif = "0.17.8" indicatif = "0.17.8"
js-sys = "0.3.69" js-sys = "0.3.69"
libm = "0.2.8" libm = "0.2.8"
log = { default-features = false, version = "0.4.21" } log = { default-features = false, version = "0.4.21" }
pretty_assertions = "1.4" pretty_assertions = "1.4.0"
proc-macro2 = "1.0.79" proc-macro2 = "1.0.82"
protobuf = "3.3" protobuf = "3.4.0"
protobuf-codegen = "3.3" protobuf-codegen = "3.4.0"
quote = "1.0.33" quote = "1.0.36"
percent-encoding = "2.3.1" percent-encoding = "2.3.1"
r2d2 = "0.8.10" r2d2 = "0.8.10"
r2d2_sqlite = { version = "0.23.0" } r2d2_sqlite = { version = "0.24.0" }
rayon = "1.10.0" rayon = "1.10.0"
regex = "1.10.4" regex = "1.10.4"
reqwest = "0.11.27" reqwest = "0.12.4"
rmp-serde = "1.3.0" rmp-serde = "1.3.0"
rstest = "0.18.2" rstest = "0.19.0"
rusqlite = { version = "0.30.0" } rusqlite = { version = "0.31.0" }
rust-format = { version = "0.3.4" } rust-format = { version = "0.3.4" }
sanitize-filename = "0.5.0" sanitize-filename = "0.5.0"
serde_rusqlite = "0.34.0" serde_rusqlite = "0.35.0"
serde-wasm-bindgen = "0.6.5" serde-wasm-bindgen = "0.6.5"
spin = { version = "0.9.8", features = ["mutex", "spin_mutex"] } spin = { version = "0.9.8", features = ["mutex", "spin_mutex"] }
strum = "0.25.0" strum = "0.26.2"
strum_macros = "0.25.3" strum_macros = "0.26.2"
syn = { version = "2.0", features = ["full", "extra-traits"] } syn = { version = "2.0.61", features = ["full", "extra-traits"] }
tempfile = "3.10.1" tempfile = "3.10.1"
thiserror = "1.0.58" thiserror = "1.0.60"
tokio = { version = "1.37.0", features = ["rt", "macros"] } tokio = { version = "1.37.0", features = ["rt", "macros"] }
tracing-appender = "0.2.3" tracing-appender = "0.2.3"
tracing-core = "0.1.32" tracing-core = "0.1.32"
tracing-subscriber = "0.3.18" tracing-subscriber = "0.3.18"
wasm-bindgen = "0.2.88" wasm-bindgen = "0.2.92"
wasm-bindgen-futures = "0.4.42" wasm-bindgen-futures = "0.4.42"
wasm-logger = "0.2.0" wasm-logger = "0.2.0"
wasm-timer = "0.2.5" wasm-timer = "0.2.5"
@ -84,20 +84,20 @@ serial_test = "3.1.1"
web-time = "1.1.0" web-time = "1.1.0"
hound = "3.5.1" hound = "3.5.1"
image = "0.25.1" image = "0.25.1"
zip = "0.6.6" zip = "1.2.1"
# Terminal UI # Terminal UI
ratatui = "0.25" ratatui = "0.26.2"
crossterm = "0.27" crossterm = "0.27.0"
# WGPU stuff # WGPU stuff
futures-intrusive = "0.5" futures-intrusive = "0.5.0"
text_placeholder = "0.5.0" text_placeholder = "0.5.0"
pollster = "0.3" pollster = "0.3.0"
wgpu = "0.19.4" wgpu = "0.20.0"
# Benchmarks and Burnbench # Benchmarks and Burnbench
arboard = "3.3.2" arboard = "3.4.0"
github-device-flow = "0.2.0" github-device-flow = "0.2.0"
os_info = "3.8.2" os_info = "3.8.2"
wsl = "0.1.0" wsl = "0.1.0"
@ -120,7 +120,7 @@ half = { version = "2.4.1", features = [
ndarray = { version = "0.15.6", default-features = false } ndarray = { version = "0.15.6", default-features = false }
matrixmultiply = { version = "0.3.8", default-features = false } matrixmultiply = { version = "0.3.8", default-features = false }
openblas-src = "0.10.9" openblas-src = "0.10.9"
blas-src = { version = "0.9.0", default-features = false } blas-src = { version = "0.10.0", default-features = false }
num-traits = { version = "0.2.19", default-features = false, features = [ num-traits = { version = "0.2.19", default-features = false, features = [
"libm", "libm",
] } # libm is for no_std ] } # libm is for no_std
@ -128,17 +128,17 @@ rand = { version = "0.8.5", default-features = false, features = [
"std_rng", "std_rng",
] } # std_rng is for no_std ] } # std_rng is for no_std
rand_distr = { version = "0.4.3", default-features = false } rand_distr = { version = "0.4.3", default-features = false }
serde = { version = "1.0.200", default-features = false, features = [ serde = { version = "1.0.201", default-features = false, features = [
"derive", "derive",
"alloc", "alloc",
] } # alloc is for no_std, derive is needed ] } # alloc is for no_std, derive is needed
serde_json = { version = "1.0.115", default-features = false } serde_json = { version = "1.0.117", default-features = false }
uuid = { version = "1.8.0", default-features = false } uuid = { version = "1.8.0", default-features = false }
libc = "0.2.154" libc = "0.2.154"
tch = "0.15.0" tch = "0.15.0"
nvml-wrapper = "0.9.0" nvml-wrapper = "0.10.0"
sysinfo = "0.30.10" sysinfo = "0.30.12"
systemstat = "0.2.3" systemstat = "0.2.3"

4
backend-comparison/Cargo.toml
View File

@ -28,8 +28,8 @@ wgpu-fusion = ["wgpu", "burn/fusion"]
[dependencies] [dependencies]
arboard = { workspace = true } arboard = { workspace = true }
burn = { path = "../crates/burn", default-features = false } burn = { path = "../crates/burn", default-features = false }
burn-common = { path = "../crates/burn-common", version = "0.14.0" } burn-common = { path = "../crates/burn-common", version = "0.15.0" }
burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.14.0" } burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.15.0" }
clap = { workspace = true } clap = { workspace = true }
colored = { workspace = true } colored = { workspace = true }
derive-new = { workspace = true } derive-new = { workspace = true }

2
burn-book/src/basic-workflow/README.md
View File

@ -14,7 +14,7 @@ automatically add the missing imports as you add the code snippets to your code.
Be sure to checkout the git branch corresponding to the version of Burn you are using to follow Be sure to checkout the git branch corresponding to the version of Burn you are using to follow
this guide. this guide.
The current version of Burn is `0.14` and the corresponding branch to checkout is `main`. The current version of Burn is `0.15` and the corresponding branch to checkout is `main`.
</div> </div>
The code for this demo can be executed from Burn's base directory using the command: The code for this demo can be executed from Burn's base directory using the command:

2
burn-book/src/basic-workflow/model.md
View File

@ -20,7 +20,7 @@ version = "0.1.0"
edition = "2021" edition = "2021"
[dependencies] [dependencies]
burn = { version = "0.14.0", features = ["train", "wgpu", "vision"] } burn = { version = "0.15.0", features = ["train", "wgpu", "vision"] }
``` ```
Our goal will be to create a basic convolutional neural network used for image classification. We Our goal will be to create a basic convolutional neural network used for image classification. We

8
crates/burn-autodiff/Cargo.toml
View File

@ -17,15 +17,15 @@ std = []
async = [] # Require std async = [] # Require std
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.14.0" } burn-common = { path = "../burn-common", version = "0.15.0" }
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false } burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false }
burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.14.0", optional = true } burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true }
derive-new = { workspace = true } derive-new = { workspace = true }
spin = { workspace = true } spin = { workspace = true }
log = { workspace = true } log = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

8
crates/burn-candle/Cargo.toml
View File

@ -21,17 +21,17 @@ accelerate = ["candle-core/accelerate"]
[dependencies] [dependencies]
derive-new = { workspace = true } derive-new = { workspace = true }
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false } burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false }
half = { workspace = true } half = { workspace = true }
candle-core = { workspace = true } candle-core = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-autodiff = { path = "../burn-autodiff", version = "0.14.0", default-features = false, features = [ burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }
burn-tch = { path = "../burn-tch", version = "0.14.0", default-features = false, features = [ burn-tch = { path = "../burn-tch", version = "0.15.0", default-features = false, features = [
] } ] }
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

2
crates/burn-compute/Cargo.toml
View File

@ -34,7 +34,7 @@ autotune-persistent-cache = [
] # Assume std ] # Assume std
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.14.0", default-features = false } burn-common = { path = "../burn-common", version = "0.15.0", default-features = false }
derive-new = { workspace = true } derive-new = { workspace = true }
spin = { workspace = true } spin = { workspace = true }
log = { workspace = true } log = { workspace = true }

24
crates/burn-core/Cargo.toml
View File

@ -99,17 +99,17 @@ test-wgpu = ["wgpu"] # To use wgpu during testing, default uses ndarray.
# ** Please make sure all dependencies support no_std when std is disabled ** # ** Please make sure all dependencies support no_std when std is disabled **
burn-common = { path = "../burn-common", version = "0.14.0", default-features = false } burn-common = { path = "../burn-common", version = "0.15.0", default-features = false }
burn-dataset = { path = "../burn-dataset", version = "0.14.0", optional = true, default-features = false } burn-dataset = { path = "../burn-dataset", version = "0.15.0", optional = true, default-features = false }
burn-derive = { path = "../burn-derive", version = "0.14.0" } burn-derive = { path = "../burn-derive", version = "0.15.0" }
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false } burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false }
# Backends # Backends
burn-ndarray = { path = "../burn-ndarray", version = "0.14.0", optional = true, default-features = false } burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", optional = true, default-features = false }
burn-wgpu = { path = "../burn-wgpu", version = "0.14.0", optional = true, default-features = false } burn-wgpu = { path = "../burn-wgpu", version = "0.15.0", optional = true, default-features = false }
burn-autodiff = { path = "../burn-autodiff", version = "0.14.0", optional = true } burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", optional = true }
burn-tch = { path = "../burn-tch", version = "0.14.0", optional = true } burn-tch = { path = "../burn-tch", version = "0.15.0", optional = true }
burn-candle = { path = "../burn-candle", version = "0.14.0", optional = true } burn-candle = { path = "../burn-candle", version = "0.15.0", optional = true }
derive-new = { workspace = true } derive-new = { workspace = true }
log = { workspace = true, optional = true } log = { workspace = true, optional = true }
@ -135,12 +135,12 @@ num-traits = { workspace = true }
[dev-dependencies] [dev-dependencies]
tempfile = { workspace = true } tempfile = { workspace = true }
burn-dataset = { path = "../burn-dataset", version = "0.14.0", features = [ burn-dataset = { path = "../burn-dataset", version = "0.15.0", features = [
"fake", "fake",
] } ] }
burn-ndarray = { path = "../burn-ndarray", version = "0.14.0", default-features = false } burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", default-features = false }
burn-autodiff = { path = "../burn-autodiff", version = "0.14.0" } burn-autodiff = { path = "../burn-autodiff", version = "0.15.0" }
[package.metadata.docs.rs] [package.metadata.docs.rs]
features = ["doc"] features = ["doc"]

12
crates/burn-cuda/Cargo.toml
View File

@ -18,11 +18,11 @@ doc = ["burn-jit/doc"]
std = ["burn-jit/std"] std = ["burn-jit/std"]
[dependencies] [dependencies]
burn-jit = { path = "../burn-jit", version = "0.14.0", default-features = false } burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false }
burn-compute = { path = "../burn-compute", version = "0.14.0" } burn-compute = { path = "../burn-compute", version = "0.15.0" }
burn-tensor = { path = "../burn-tensor", version = "0.14.0" } burn-tensor = { path = "../burn-tensor", version = "0.15.0" }
burn-common = { path = "../burn-common", version = "0.14.0" } burn-common = { path = "../burn-common", version = "0.15.0" }
burn-fusion = { path = "../burn-fusion", version = "0.14.0", optional = true } burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true }
half = { workspace = true } half = { workspace = true }
bytemuck = { workspace = true } bytemuck = { workspace = true }
@ -32,7 +32,7 @@ log = { workspace = true }
derive-new = { workspace = true } derive-new = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-jit = { path = "../burn-jit", version = "0.14.0", default-features = false, features = [ burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

2
crates/burn-dataset/Cargo.toml
View File

@ -33,7 +33,7 @@ __sqlite-shared = [
] ]
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.14.0", optional = true, features = [ burn-common = { path = "../burn-common", version = "0.15.0", optional = true, features = [
"network", "network",
] } ] }
csv = { workspace = true } csv = { workspace = true }

4
crates/burn-fusion/Cargo.toml
View File

@ -16,8 +16,8 @@ std = ["serde/std"]
doc = ["default"] doc = ["default"]
[dependencies] [dependencies]
burn-tensor = { path = "../burn-tensor", version = "0.14.0" } burn-tensor = { path = "../burn-tensor", version = "0.15.0" }
burn-common = { path = "../burn-common", version = "0.14.0" } burn-common = { path = "../burn-common", version = "0.15.0" }
hashbrown = { workspace = true } hashbrown = { workspace = true }
derive-new = {workspace = true } derive-new = {workspace = true }
spin = { workspace = true } spin = { workspace = true }

2
crates/burn-import/Cargo.toml
View File

@ -19,7 +19,7 @@ onnx = []
pytorch = ["burn/record-item-custom-serde", "thiserror", "zip"] pytorch = ["burn/record-item-custom-serde", "thiserror", "zip"]
[dependencies] [dependencies]
burn = { path = "../burn", version = "0.14.0", features = ["ndarray"] } burn = { path = "../burn", version = "0.15.0", features = ["ndarray"] }
bytemuck = { workspace = true } bytemuck = { workspace = true }
candle-core = { workspace = true } candle-core = { workspace = true }

14
crates/burn-jit/Cargo.toml
View File

@ -27,9 +27,9 @@ export_tests = [
] ]
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.14.0" } burn-common = { path = "../burn-common", version = "0.15.0" }
burn-tensor = { path = "../burn-tensor", version = "0.14.0" } burn-tensor = { path = "../burn-tensor", version = "0.15.0" }
burn-fusion = { path = "../burn-fusion", version = "0.14.0", optional = true } burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true }
bytemuck = { workspace = true } bytemuck = { workspace = true }
derive-new = { workspace = true } derive-new = { workspace = true }
@ -44,16 +44,16 @@ serde = { workspace = true }
text_placeholder = { workspace = true, features = ["struct_context"] } text_placeholder = { workspace = true, features = ["struct_context"] }
hashbrown = { workspace = true } hashbrown = { workspace = true }
burn-compute = { path = "../burn-compute", version = "0.14.0", default-features = false, features = [ burn-compute = { path = "../burn-compute", version = "0.15.0", default-features = false, features = [
"channel-mutex", "channel-mutex",
"std", "std",
] } ] }
burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.14.0", optional = true } burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true }
# When exporting tests # When exporting tests
serial_test = { workspace = true, optional = true } serial_test = { workspace = true, optional = true }
burn-autodiff = { path = "../burn-autodiff", version = "0.14.0", default-features = false, optional = true } burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, optional = true }
burn-ndarray = { path = "../burn-ndarray", version = "0.14.0", optional = true } burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", optional = true }
[package.metadata.docs.rs] [package.metadata.docs.rs]
features = ["doc"] features = ["doc"]

10
crates/burn-ndarray/Cargo.toml
View File

@ -42,9 +42,9 @@ blas-openblas-system = [
# ** Please make sure all dependencies support no_std when std is disabled ** # ** Please make sure all dependencies support no_std when std is disabled **
burn-autodiff = { path = "../burn-autodiff", version = "0.14.0", optional = true } burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", optional = true }
burn-common = { path = "../burn-common", version = "0.14.0", default-features = false } burn-common = { path = "../burn-common", version = "0.15.0", default-features = false }
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false } burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false }
matrixmultiply = { workspace = true, default-features = false } matrixmultiply = { workspace = true, default-features = false }
rayon = { workspace = true, optional = true } rayon = { workspace = true, optional = true }
@ -58,10 +58,10 @@ rand = { workspace = true }
spin = { workspace = true } # using in place of use std::sync::Mutex; spin = { workspace = true } # using in place of use std::sync::Mutex;
[dev-dependencies] [dev-dependencies]
burn-autodiff = { path = "../burn-autodiff", version = "0.14.0", default-features = false, features = [ burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

4
crates/burn-no-std-tests/Cargo.toml
View File

@ -14,7 +14,7 @@ version.workspace = true
# ** Please make sure all dependencies support no_std ** # ** Please make sure all dependencies support no_std **
burn = { path = "../burn", version = "0.14.0", default-features = false } burn = { path = "../burn", version = "0.15.0", default-features = false }
burn-ndarray = { path = "../burn-ndarray", version = "0.14.0", default-features = false } burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", default-features = false }
serde = { workspace = true } serde = { workspace = true }

6
crates/burn-tch/Cargo.toml
View File

@ -15,7 +15,7 @@ default = []
doc = ["tch/doc-only"] doc = ["tch/doc-only"]
[dependencies] [dependencies]
burn-tensor = { path = "../burn-tensor", version = "0.14.0" } burn-tensor = { path = "../burn-tensor", version = "0.15.0" }
half = { workspace = true, features = ["std"] } half = { workspace = true, features = ["std"] }
libc = { workspace = true } libc = { workspace = true }
@ -23,10 +23,10 @@ rand = { workspace = true, features = ["std"] }
tch = { workspace = true, features = ["download-libtorch"] } tch = { workspace = true, features = ["download-libtorch"] }
[dev-dependencies] [dev-dependencies]
burn-autodiff = { path = "../burn-autodiff", version = "0.14.0", default-features = false, features = [ burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }
burn-tensor = { path = "../burn-tensor", version = "0.14.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

4
crates/burn-tensor/Cargo.toml
View File

@ -20,8 +20,8 @@ repr = []
wasm-sync = [] wasm-sync = []
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.14.0", default-features = false } burn-common = { path = "../burn-common", version = "0.15.0", default-features = false }
burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.14.0", optional = true } burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true }
derive-new = { workspace = true } derive-new = { workspace = true }
half = { workspace = true } half = { workspace = true }

4
crates/burn-train/Cargo.toml
View File

@ -17,7 +17,7 @@ metrics = ["nvml-wrapper", "sysinfo", "systemstat"]
tui = ["ratatui", "crossterm"] tui = ["ratatui", "crossterm"]
[dependencies] [dependencies]
burn-core = { path = "../burn-core", version = "0.14.0", features = ["dataset"] } burn-core = { path = "../burn-core", version = "0.15.0", features = ["dataset"] }
log = { workspace = true } log = { workspace = true }
tracing-subscriber = { workspace = true } tracing-subscriber = { workspace = true }
@ -38,7 +38,7 @@ derive-new = { workspace = true }
serde = { workspace = true, features = ["std", "derive"] } serde = { workspace = true, features = ["std", "derive"] }
[dev-dependencies] [dev-dependencies]
burn-ndarray = { path = "../burn-ndarray", version = "0.14.0" } burn-ndarray = { path = "../burn-ndarray", version = "0.15.0" }
[package.metadata.docs.rs] [package.metadata.docs.rs]
features = ["doc"] features = ["doc"]

7
crates/burn-train/src/renderer/tui/metric_numeric.rs
View File

@ -219,12 +219,7 @@ impl<'a> NumericMetricView<'a> {
) )
.split(size); .split(size);
let titles = titles let tabs = Tabs::new(titles.iter().map(|i| Line::from(vec![i.clone().yellow()])))
.iter()
.map(|i| Line::from(vec![i.clone().yellow()]))
.collect();
let tabs = Tabs::new(titles)
.select(selected) .select(selected)
.style(Style::default()) .style(Style::default())
.highlight_style( .highlight_style(

12
crates/burn-wgpu/Cargo.toml
View File

@ -19,11 +19,11 @@ doc = ["burn-jit/doc"]
std = ["burn-jit/std"] std = ["burn-jit/std"]
[dependencies] [dependencies]
burn-jit = { path = "../burn-jit", version = "0.14.0", default-features = false } burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false }
burn-compute = { path = "../burn-compute", version = "0.14.0" } burn-compute = { path = "../burn-compute", version = "0.15.0" }
burn-tensor = { path = "../burn-tensor", version = "0.14.0" } burn-tensor = { path = "../burn-tensor", version = "0.15.0" }
burn-common = { path = "../burn-common", version = "0.14.0" } burn-common = { path = "../burn-common", version = "0.15.0" }
burn-fusion = { path = "../burn-fusion", version = "0.14.0", optional = true } burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true }
bytemuck = { workspace = true } bytemuck = { workspace = true }
wgpu = { workspace = true, features = ["fragile-send-sync-non-atomic-wasm"] } wgpu = { workspace = true, features = ["fragile-send-sync-non-atomic-wasm"] }
@ -35,6 +35,6 @@ derive-new = { workspace = true }
hashbrown = { workspace = true } hashbrown = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-jit = { path = "../burn-jit", version = "0.14.0", default-features = false, features = [ burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

1
crates/burn-wgpu/src/compute/server.rs
View File

@ -109,6 +109,7 @@ where
layout: None, layout: None,
module: &module, module: &module,
entry_point: "main", entry_point: "main",
compilation_options: Default::default(),
}), }),
) )
} }

4
crates/burn/Cargo.toml
View File

@ -68,8 +68,8 @@ record-item-custom-serde = ["burn-core/record-item-custom-serde"]
# ** Please make sure all dependencies support no_std when std is disabled ** # ** Please make sure all dependencies support no_std when std is disabled **
burn-core = { path = "../burn-core", version = "0.14.0", default-features = false } burn-core = { path = "../burn-core", version = "0.15.0", default-features = false }
burn-train = { path = "../burn-train", version = "0.14.0", optional = true, default-features = false } burn-train = { path = "../burn-train", version = "0.15.0", optional = true, default-features = false }
[package.metadata.docs.rs] [package.metadata.docs.rs]
features = ["doc"] features = ["doc"]

6
examples/image-classification-web/Cargo.toml
View File

@ -14,11 +14,11 @@ default = []
half_precision = [] half_precision = []
[dependencies] [dependencies]
burn = { path = "../../crates/burn", version = "0.14.0", default-features = false, features = [ burn = { path = "../../crates/burn", version = "0.15.0", default-features = false, features = [
"ndarray", "ndarray",
] } ] }
burn-wgpu = { path = "../../crates/burn-wgpu", version = "0.14.0", default-features = false } burn-wgpu = { path = "../../crates/burn-wgpu", version = "0.15.0", default-features = false }
burn-candle = { path = "../../crates/burn-candle", version = "0.14.0", default-features = false } burn-candle = { path = "../../crates/burn-candle", version = "0.15.0", default-features = false }
js-sys = { workspace = true } js-sys = { workspace = true }
log = { workspace = true } log = { workspace = true }

2
examples/pytorch-import/Cargo.toml
View File

@ -4,7 +4,7 @@ edition = "2021"
license = "MIT OR Apache-2.0" license = "MIT OR Apache-2.0"
name = "pytorch-import" name = "pytorch-import"
publish = false publish = false
version = "0.14.0" version = "0.15.0"
[dependencies] [dependencies]
burn = { path = "../../crates/burn", features = [ burn = { path = "../../crates/burn", features = [