Bump next version of Burn to 0.16.0 (#2434)

This commit is contained in:
Guillaume Lagrange 2024-10-28 16:41:58 -04:00 committed by GitHub
parent 65aa8b5eaa
commit b8e47dabe3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
29 changed files with 137 additions and 137 deletions

84
Cargo.lock generated
View File

@ -304,7 +304,7 @@ dependencies = [
[[package]] [[package]]
name = "backend-comparison" name = "backend-comparison"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"arboard", "arboard",
"burn", "burn",
@ -490,7 +490,7 @@ checksum = "79296716171880943b8470b5f8d03aa55eb2e645a4874bdbb28adb49162e012c"
[[package]] [[package]]
name = "burn" name = "burn"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-core", "burn-core",
"burn-train", "burn-train",
@ -498,7 +498,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-autodiff" name = "burn-autodiff"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"burn-tensor", "burn-tensor",
@ -510,7 +510,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-candle" name = "burn-candle"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-autodiff", "burn-autodiff",
"burn-tch", "burn-tch",
@ -522,7 +522,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-common" name = "burn-common"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"cubecl-common", "cubecl-common",
"dashmap", "dashmap",
@ -536,7 +536,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-core" name = "burn-core"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"ahash", "ahash",
"bincode", "bincode",
@ -572,7 +572,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-cuda" name = "burn-cuda"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-fusion", "burn-fusion",
"burn-jit", "burn-jit",
@ -586,7 +586,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-dataset" name = "burn-dataset"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"csv", "csv",
@ -618,7 +618,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-derive" name = "burn-derive"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"derive-new", "derive-new",
"proc-macro2", "proc-macro2",
@ -628,7 +628,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-fusion" name = "burn-fusion"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"burn-tensor", "burn-tensor",
@ -642,7 +642,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-hip" name = "burn-hip"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-fusion", "burn-fusion",
"burn-jit", "burn-jit",
@ -656,7 +656,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-import" name = "burn-import"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"candle-core", "candle-core",
@ -681,7 +681,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-jit" name = "burn-jit"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-autodiff", "burn-autodiff",
"burn-common", "burn-common",
@ -706,7 +706,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-ndarray" name = "burn-ndarray"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"atomic_float", "atomic_float",
"blas-src", "blas-src",
@ -726,7 +726,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-no-std-tests" name = "burn-no-std-tests"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-ndarray", "burn-ndarray",
@ -735,7 +735,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-router" name = "burn-router"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-autodiff", "burn-autodiff",
"burn-ndarray", "burn-ndarray",
@ -747,7 +747,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-tch" name = "burn-tch"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-autodiff", "burn-autodiff",
"burn-tensor", "burn-tensor",
@ -760,7 +760,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-tensor" name = "burn-tensor"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"burn-tensor-testgen", "burn-tensor-testgen",
@ -780,7 +780,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-tensor-testgen" name = "burn-tensor-testgen"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"proc-macro2", "proc-macro2",
"quote", "quote",
@ -788,7 +788,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-train" name = "burn-train"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-core", "burn-core",
"burn-ndarray", "burn-ndarray",
@ -806,7 +806,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-wgpu" name = "burn-wgpu"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-fusion", "burn-fusion",
"burn-jit", "burn-jit",
@ -1687,7 +1687,7 @@ dependencies = [
[[package]] [[package]]
name = "custom-csv-dataset" name = "custom-csv-dataset"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"csv", "csv",
@ -1697,7 +1697,7 @@ dependencies = [
[[package]] [[package]]
name = "custom-cubecl-kernel" name = "custom-cubecl-kernel"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-jit", "burn-jit",
@ -1710,7 +1710,7 @@ dependencies = [
[[package]] [[package]]
name = "custom-image-dataset" name = "custom-image-dataset"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"flate2", "flate2",
@ -1719,7 +1719,7 @@ dependencies = [
[[package]] [[package]]
name = "custom-renderer" name = "custom-renderer"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"bytemuck", "bytemuck",
@ -1731,7 +1731,7 @@ dependencies = [
[[package]] [[package]]
name = "custom-training-loop" name = "custom-training-loop"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"bytemuck", "bytemuck",
@ -1743,7 +1743,7 @@ dependencies = [
[[package]] [[package]]
name = "custom-wgpu-kernel" name = "custom-wgpu-kernel"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"bytemuck", "bytemuck",
@ -2793,7 +2793,7 @@ dependencies = [
[[package]] [[package]]
name = "guide" name = "guide"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"log", "log",
@ -3259,7 +3259,7 @@ dependencies = [
[[package]] [[package]]
name = "image-classification-web" name = "image-classification-web"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-candle", "burn-candle",
@ -3780,7 +3780,7 @@ dependencies = [
[[package]] [[package]]
name = "mnist" name = "mnist"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"log", "log",
@ -3789,7 +3789,7 @@ dependencies = [
[[package]] [[package]]
name = "mnist-inference-web" name = "mnist-inference-web"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"console_error_panic_hook", "console_error_panic_hook",
@ -3802,7 +3802,7 @@ dependencies = [
[[package]] [[package]]
name = "model" name = "model"
version = "0.4.0" version = "0.5.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-import", "burn-import",
@ -3874,7 +3874,7 @@ dependencies = [
[[package]] [[package]]
name = "named-tensor" name = "named-tensor"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"serde", "serde",
@ -4320,7 +4320,7 @@ dependencies = [
[[package]] [[package]]
name = "onnx-inference" name = "onnx-inference"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-import", "burn-import",
@ -4329,7 +4329,7 @@ dependencies = [
[[package]] [[package]]
name = "onnx-ir" name = "onnx-ir"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"half", "half",
@ -4346,7 +4346,7 @@ dependencies = [
[[package]] [[package]]
name = "onnx-tests" name = "onnx-tests"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-import", "burn-import",
@ -5244,7 +5244,7 @@ dependencies = [
[[package]] [[package]]
name = "pytorch-import" name = "pytorch-import"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-import", "burn-import",
@ -5253,7 +5253,7 @@ dependencies = [
[[package]] [[package]]
name = "pytorch-tests" name = "pytorch-tests"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"burn-autodiff", "burn-autodiff",
@ -6211,7 +6211,7 @@ checksum = "e3a9fe34e3e7a50316060351f37187a3f546bce95496156754b601a5fa71b76e"
[[package]] [[package]]
name = "simple-regression" name = "simple-regression"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"log", "log",
@ -6617,7 +6617,7 @@ dependencies = [
[[package]] [[package]]
name = "text-classification" name = "text-classification"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"derive-new", "derive-new",
@ -6627,7 +6627,7 @@ dependencies = [
[[package]] [[package]]
name = "text-generation" name = "text-generation"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"derive-new", "derive-new",
@ -7770,7 +7770,7 @@ checksum = "af4e2e2f7cba5a093896c1e150fbfe177d1883e7448200efb81d40b9d339ef26"
[[package]] [[package]]
name = "xtask" name = "xtask"
version = "1.0.0" version = "1.1.0"
dependencies = [ dependencies = [
"log", "log",
"rstest", "rstest",

View File

@ -24,7 +24,7 @@ exclude = [
edition = "2021" edition = "2021"
license = "MIT OR Apache-2.0" license = "MIT OR Apache-2.0"
readme = "README.md" readme = "README.md"
version = "0.15.0" version = "0.16.0"
[workspace.dependencies] [workspace.dependencies]
atomic_float = "1" atomic_float = "1"

View File

@ -33,8 +33,8 @@ wgpu-spirv-fusion = ["wgpu-spirv", "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.15.0" } burn-common = { path = "../crates/burn-common", version = "0.16.0" }
burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.15.0", optional = true } burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.16.0", optional = true }
clap = { workspace = true } clap = { workspace = true }
colored = { workspace = true } colored = { workspace = true }
cubecl = { workspace = true, features = ["wgpu"], default-features = true } cubecl = { workspace = true, features = ["wgpu"], default-features = true }

View File

@ -23,7 +23,7 @@ Some other dependencies have to be added
```toml ```toml
[dependencies] [dependencies]
embedded-alloc = "0.5.1" # Only if there is no default allocator for your chip embedded-alloc = "0.5.1" # Only if there is no default allocator for your chip
burn = { version = "0.15", default-features = false, features = ["ndarray"] } # Backend must be ndarray burn = { version = "0.16", default-features = false, features = ["ndarray"] } # Backend must be ndarray
[build-dependencies] [build-dependencies]
burn-import = { version = "0.14" } # Used to auto generate the rust code to import the model burn-import = { version = "0.14" } # Used to auto generate the rust code to import the model

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.15` and the corresponding branch to checkout is `main`. The current version of Burn is `0.16` 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:

View File

@ -20,7 +20,7 @@ version = "0.1.0"
edition = "2021" edition = "2021"
[dependencies] [dependencies]
burn = { version = "~0.15", features = ["train", "wgpu", "vision"] } burn = { version = "~0.16", 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

View File

@ -74,7 +74,7 @@ First, add the `burn-import` crate to your `Cargo.toml`:
```toml ```toml
[build-dependencies] [build-dependencies]
burn-import = "~0.15" burn-import = "~0.16"
``` ```
Then, in your `build.rs` file: Then, in your `build.rs` file:

View File

@ -18,16 +18,16 @@ std = []
async = [] # Require std async = [] # Require std
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.15.0" } burn-common = { path = "../burn-common", version = "0.16.0" }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false } burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false }
burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true } burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.16.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.15.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

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.15.0", default-features = false } burn-tensor = { path = "../burn-tensor", version = "0.16.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.15.0", default-features = false, features = [ burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }
burn-tch = { path = "../burn-tch", version = "0.15.0", default-features = false, features = [ burn-tch = { path = "../burn-tch", version = "0.16.0", default-features = false, features = [
] } ] }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

View File

@ -118,19 +118,19 @@ test-wgpu-spirv = [
# ** 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.15.0", default-features = false } burn-common = { path = "../burn-common", version = "0.16.0", default-features = false }
burn-dataset = { path = "../burn-dataset", version = "0.15.0", optional = true, default-features = false } burn-dataset = { path = "../burn-dataset", version = "0.16.0", optional = true, default-features = false }
burn-derive = { path = "../burn-derive", version = "0.15.0" } burn-derive = { path = "../burn-derive", version = "0.16.0" }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false } burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false }
# Backends # Backends
burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", optional = true } burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", optional = true }
burn-candle = { path = "../burn-candle", version = "0.15.0", optional = true } burn-candle = { path = "../burn-candle", version = "0.16.0", optional = true }
burn-cuda = { path = "../burn-cuda", version = "0.15.0", optional = true, default-features = false } burn-cuda = { path = "../burn-cuda", version = "0.16.0", optional = true, default-features = false }
burn-hip = { path = "../burn-hip", version = "0.15.0", optional = true, default-features = false } burn-hip = { path = "../burn-hip", version = "0.16.0", optional = true, default-features = false }
burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", optional = true, default-features = false } burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", optional = true, default-features = false }
burn-tch = { path = "../burn-tch", version = "0.15.0", optional = true } burn-tch = { path = "../burn-tch", version = "0.16.0", optional = true }
burn-wgpu = { path = "../burn-wgpu", version = "0.15.0", optional = true, default-features = false } burn-wgpu = { path = "../burn-wgpu", version = "0.16.0", optional = true, default-features = false }
data-encoding = { workspace = true } data-encoding = { workspace = true }
uuid = { workspace = true } uuid = { workspace = true }
@ -160,13 +160,13 @@ thiserror = { workspace = true, optional = true }
portable-atomic-util = { workspace = true } portable-atomic-util = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-dataset = { path = "../burn-dataset", version = "0.15.0", features = [ burn-dataset = { path = "../burn-dataset", version = "0.16.0", features = [
"fake", "fake",
] } ] }
tempfile = { workspace = true } tempfile = { workspace = true }
burn-autodiff = { path = "../burn-autodiff", version = "0.15.0" } burn-autodiff = { path = "../burn-autodiff", version = "0.16.0" }
burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", default-features = false } burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", default-features = false }
[package.metadata.docs.rs] [package.metadata.docs.rs]
features = ["doc"] features = ["doc"]

View File

@ -20,9 +20,9 @@ std = ["burn-jit/std", "cubecl/std"]
[dependencies] [dependencies]
cubecl = { workspace = true, features = ["cuda"] } cubecl = { workspace = true, features = ["cuda"] }
burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false } burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = ["cubecl-cuda"] } burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = ["cubecl-cuda"] }
burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true } burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
half = { workspace = true } half = { workspace = true }
bytemuck = { workspace = true } bytemuck = { workspace = true }
@ -31,7 +31,7 @@ log = { workspace = true }
derive-new = { workspace = true } derive-new = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [ burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

View File

@ -30,7 +30,7 @@ __sqlite-shared = [
dataframe = ["dep:polars"] dataframe = ["dep:polars"]
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.15.0", optional = true, features = [ burn-common = { path = "../burn-common", version = "0.16.0", optional = true, features = [
"network", "network",
] } ] }
csv = { workspace = true } csv = { workspace = true }

View File

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

View File

@ -20,9 +20,9 @@ std = ["burn-jit/std", "cubecl/std"]
[dependencies] [dependencies]
cubecl = { workspace = true, features = ["hip"] } cubecl = { workspace = true, features = ["hip"] }
burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false } burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = ["cubecl-hip"] } burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = ["cubecl-hip"] }
burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true } burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
half = { workspace = true } half = { workspace = true }
bytemuck = { workspace = true } bytemuck = { workspace = true }
@ -31,7 +31,7 @@ log = { workspace = true }
derive-new = { workspace = true } derive-new = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [ burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

View File

@ -20,8 +20,8 @@ onnx = []
pytorch = ["burn/record-item-custom-serde", "thiserror", "zip"] pytorch = ["burn/record-item-custom-serde", "thiserror", "zip"]
[dependencies] [dependencies]
burn = { path = "../burn", version = "0.15.0", features = ["ndarray"] } burn = { path = "../burn", version = "0.16.0", features = ["ndarray"] }
onnx-ir = { path = "../onnx-ir", version = "0.15.0" } onnx-ir = { path = "../onnx-ir", version = "0.16.0" }
candle-core = { workspace = true } candle-core = { workspace = true }
derive-new = { workspace = true } derive-new = { workspace = true }
half = { workspace = true } half = { workspace = true }

View File

@ -28,9 +28,9 @@ std = ["cubecl/std"]
template = [] template = []
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.15.0" } burn-common = { path = "../burn-common", version = "0.16.0" }
burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true } burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = [ burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = [
"cubecl", "repr", "cubecl", "repr",
] } ] }
cubecl = { workspace = true, features = ["linalg"] } cubecl = { workspace = true, features = ["linalg"] }
@ -50,12 +50,12 @@ futures-lite = { workspace = true, features = ["std"] }
serde = { workspace = true } serde = { workspace = true }
text_placeholder = { workspace = true, features = ["struct_context"] } text_placeholder = { workspace = true, features = ["struct_context"] }
burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true } burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.16.0", optional = true }
hashbrown = { workspace = true } hashbrown = { workspace = true }
# When exporting tests # When exporting tests
burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, optional = true } burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, optional = true }
burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", optional = true } burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", optional = true }
serial_test = { workspace = true, optional = true } serial_test = { workspace = true, optional = true }
[package.metadata.docs.rs] [package.metadata.docs.rs]

View File

@ -43,9 +43,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.15.0", optional = true } burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", optional = true }
burn-common = { path = "../burn-common", version = "0.15.0", default-features = false } burn-common = { path = "../burn-common", version = "0.16.0", default-features = false }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = ["repr"] } burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = ["repr"] }
atomic_float = { workspace = true } atomic_float = { workspace = true }
blas-src = { workspace = true, default-features = false, optional = true } # no-std compatible blas-src = { workspace = true, default-features = false, optional = true } # no-std compatible
@ -62,10 +62,10 @@ spin = { workspace = true } # usi
portable-atomic-util = { workspace = true } portable-atomic-util = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [ burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

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.15.0", default-features = false } burn = { path = "../burn", version = "0.16.0", default-features = false }
burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", default-features = false } burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", default-features = false }
serde = { workspace = true } serde = { workspace = true }

View File

@ -17,21 +17,21 @@ std = []
doc = ["default"] doc = ["default"]
[dependencies] [dependencies]
burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = ["repr"]} burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = ["repr"]}
hashbrown = { workspace = true } hashbrown = { workspace = true }
spin = { workspace = true } spin = { workspace = true }
[dev-dependencies] [dev-dependencies]
burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [ burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [ burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }
burn-ndarray = { path = "../burn-ndarray", version = "0.15.0" } burn-ndarray = { path = "../burn-ndarray", version = "0.16.0" }
burn-wgpu = { path = "../burn-wgpu", version = "0.15.0" } burn-wgpu = { path = "../burn-wgpu", version = "0.16.0" }
[package.metadata.docs.rs] [package.metadata.docs.rs]

View File

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

View File

@ -31,8 +31,8 @@ cubecl-cuda = ["cubecl", "cubecl/cuda"]
cubecl-hip = ["cubecl", "cubecl/hip"] cubecl-hip = ["cubecl", "cubecl/hip"]
[dependencies] [dependencies]
burn-common = { path = "../burn-common", version = "0.15.0", default-features = false } burn-common = { path = "../burn-common", version = "0.16.0", default-features = false }
burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true } burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.16.0", optional = true }
cubecl = { workspace = true, optional = true } cubecl = { workspace = true, optional = true }
derive-new = { workspace = true } derive-new = { workspace = true }

View File

@ -18,7 +18,7 @@ metrics = ["nvml-wrapper", "sysinfo", "systemstat"]
tui = ["ratatui"] tui = ["ratatui"]
[dependencies] [dependencies]
burn-core = { path = "../burn-core", version = "0.15.0", features = [ burn-core = { path = "../burn-core", version = "0.16.0", features = [
"dataset", "dataset",
"std", "std",
], default-features = false } ], default-features = false }
@ -41,7 +41,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.15.0" } burn-ndarray = { path = "../burn-ndarray", version = "0.16.0" }
[package.metadata.docs.rs] [package.metadata.docs.rs]
features = ["doc"] features = ["doc"]

View File

@ -24,15 +24,15 @@ template = ["burn-jit/template", "cubecl/template"]
[dependencies] [dependencies]
cubecl = { workspace = true, features = ["wgpu"] } cubecl = { workspace = true, features = ["wgpu"] }
burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true } burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false } burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false }
burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = [ burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = [
"cubecl-wgpu", "cubecl-wgpu",
] } ] }
[dev-dependencies] [dev-dependencies]
burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [ burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false, features = [
"export_tests", "export_tests",
] } ] }

View File

@ -70,5 +70,5 @@ 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.15.0", default-features = false } burn-core = { path = "../burn-core", version = "0.16.0", default-features = false }
burn-train = { path = "../burn-train", version = "0.15.0", optional = true, default-features = false } burn-train = { path = "../burn-train", version = "0.16.0", optional = true, default-features = false }

View File

@ -14,11 +14,11 @@ default = []
half_precision = [] half_precision = []
[dependencies] [dependencies]
burn = { path = "../../crates/burn", version = "0.15.0", default-features = false, features = [ burn = { path = "../../crates/burn", version = "0.16.0", default-features = false, features = [
"ndarray", "wgpu", "ndarray", "wgpu",
] } ] }
cubecl-runtime = { version = "0.3.0", features = ["channel-mpsc"] } # missing feature flags cubecl-runtime = { version = "0.3.0", features = ["channel-mpsc"] } # missing feature flags
burn-candle = { path = "../../crates/burn-candle", version = "0.15.0", default-features = false } burn-candle = { path = "../../crates/burn-candle", version = "0.16.0", default-features = false }
log = { workspace = true } log = { workspace = true }
serde = { workspace = true } serde = { workspace = true }

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.15.0" version = "0.16.0"
[dependencies] [dependencies]
burn = { path = "../../crates/burn", features = [ burn = { path = "../../crates/burn", features = [

View File

@ -1,6 +1,6 @@
[package] [package]
name = "model" name = "model"
version = "0.4.0" version = "0.5.0"
edition = "2021" edition = "2021"
[dependencies] [dependencies]

View File

@ -286,7 +286,7 @@ checksum = "79296716171880943b8470b5f8d03aa55eb2e645a4874bdbb28adb49162e012c"
[[package]] [[package]]
name = "burn" name = "burn"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-core", "burn-core",
"burn-train", "burn-train",
@ -294,7 +294,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-autodiff" name = "burn-autodiff"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"burn-tensor", "burn-tensor",
@ -305,7 +305,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-candle" name = "burn-candle"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-tensor", "burn-tensor",
"candle-core", "candle-core",
@ -315,7 +315,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-common" name = "burn-common"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"cubecl-common", "cubecl-common",
"data-encoding", "data-encoding",
@ -326,7 +326,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-core" name = "burn-core"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"bincode", "bincode",
"burn-autodiff", "burn-autodiff",
@ -357,7 +357,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-cuda" name = "burn-cuda"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-fusion", "burn-fusion",
"burn-jit", "burn-jit",
@ -371,7 +371,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-dataset" name = "burn-dataset"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"csv", "csv",
"derive-new", "derive-new",
@ -395,7 +395,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-derive" name = "burn-derive"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"derive-new", "derive-new",
"proc-macro2", "proc-macro2",
@ -405,7 +405,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-fusion" name = "burn-fusion"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"burn-tensor", "burn-tensor",
@ -418,7 +418,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-import" name = "burn-import"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn", "burn",
"candle-core", "candle-core",
@ -441,7 +441,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-jit" name = "burn-jit"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"burn-fusion", "burn-fusion",
@ -461,7 +461,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-ndarray" name = "burn-ndarray"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-autodiff", "burn-autodiff",
"burn-common", "burn-common",
@ -478,7 +478,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-tch" name = "burn-tch"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-tensor", "burn-tensor",
"half", "half",
@ -489,7 +489,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-tensor" name = "burn-tensor"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-common", "burn-common",
"bytemuck", "bytemuck",
@ -507,7 +507,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-train" name = "burn-train"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-core", "burn-core",
"crossterm", "crossterm",
@ -525,7 +525,7 @@ dependencies = [
[[package]] [[package]]
name = "burn-wgpu" name = "burn-wgpu"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"burn-fusion", "burn-fusion",
"burn-jit", "burn-jit",
@ -2959,7 +2959,7 @@ checksum = "3fdb12b2476b595f9358c5161aa467c2438859caa136dec86c26fdd2efe17b92"
[[package]] [[package]]
name = "onnx-ir" name = "onnx-ir"
version = "0.15.0" version = "0.16.0"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"half", "half",

View File

@ -1,6 +1,6 @@
[package] [package]
name = "xtask" name = "xtask"
version = "1.0.0" version = "1.1.0"
edition = "2021" edition = "2021"
license = "MIT OR Apache-2.0" license = "MIT OR Apache-2.0"