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

Bump cube version (#2447)

This commit is contained in:
Arthur Brussee 2024-10-31 22:15:53 +00:00 committed by GitHub
parent 8e466d7ce1
commit 9eb25780ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
Cargo.lock generated
View File

@ -1460,7 +1460,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl" name = "cubecl"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"cubecl-core", "cubecl-core",
"cubecl-cuda", "cubecl-cuda",
@ -1491,7 +1491,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-common" name = "cubecl-common"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"derive-new", "derive-new",
"embassy-futures", "embassy-futures",
@ -1508,7 +1508,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-core" name = "cubecl-core"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
@ -1525,7 +1525,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-cpp" name = "cubecl-cpp"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
@ -1539,7 +1539,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-cuda" name = "cubecl-cuda"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
@ -1555,7 +1555,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-hip" name = "cubecl-hip"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
@ -1580,7 +1580,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-linalg" name = "cubecl-linalg"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"bytemuck", "bytemuck",
"cubecl-core", "cubecl-core",
@ -1591,7 +1591,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-macros" name = "cubecl-macros"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
"darling", "darling",
@ -1606,7 +1606,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-opt" name = "cubecl-opt"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
"cubecl-core", "cubecl-core",
@ -1643,7 +1643,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-runtime" name = "cubecl-runtime"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"async-channel", "async-channel",
"async-lock", "async-lock",
@ -1664,7 +1664,7 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-spirv" name = "cubecl-spirv"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
"cubecl-core", "cubecl-core",
@ -1677,11 +1677,12 @@ dependencies = [
[[package]] [[package]]
name = "cubecl-wgpu" name = "cubecl-wgpu"
version = "0.4.0" version = "0.4.0"
source = "git+https://github.com/tracel-ai/cubecl?rev=99404b1e29946832a42b72a5c26d4cf42c67692e#99404b1e29946832a42b72a5c26d4cf42c67692e" source = "git+https://github.com/tracel-ai/cubecl?rev=0dff475fec254e884f6b82e305e7a52adebf1dd7#0dff475fec254e884f6b82e305e7a52adebf1dd7"
dependencies = [ dependencies = [
"ash", "ash",
"async-channel", "async-channel",
"bytemuck", "bytemuck",
"cfg-if",
"cfg_aliases 0.2.1", "cfg_aliases 0.2.1",
"cubecl-common 0.4.0", "cubecl-common 0.4.0",
"cubecl-core", "cubecl-core",

4
Cargo.toml
View File

@ -152,8 +152,8 @@ ahash = { version = "0.8.11", default-features = false }
portable-atomic-util = { version = "0.2.2", features = ["alloc"] } portable-atomic-util = { version = "0.2.2", features = ["alloc"] }
### For the main burn branch. ### ### For the main burn branch. ###
cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "99404b1e29946832a42b72a5c26d4cf42c67692e" } cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "0dff475fec254e884f6b82e305e7a52adebf1dd7" }
cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "99404b1e29946832a42b72a5c26d4cf42c67692e" } cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "0dff475fec254e884f6b82e305e7a52adebf1dd7" }
### For local development. ### ### For local development. ###
# cubecl = { path = "../cubecl/crates/cubecl", default-features = false } # cubecl = { path = "../cubecl/crates/cubecl", default-features = false }
# cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false } # cubecl-common = { path = "../cubecl/crates/cubecl-common", default-features = false }

4
examples/image-classification-web/src/web.rs
View File

@ -9,7 +9,7 @@ use core::convert::Into;
use crate::model::{label::LABELS, normalizer::Normalizer, squeezenet::Model as SqueezenetModel}; use crate::model::{label::LABELS, normalizer::Normalizer, squeezenet::Model as SqueezenetModel};
use burn::{ use burn::{
backend::{wgpu::init_device, NdArray}, backend::{wgpu::init_setup_async, NdArray},
prelude::*, prelude::*,
tensor::activation::softmax, tensor::activation::softmax,
}; };
@ -110,7 +110,7 @@ impl ImageClassifier {
log::info!("Loading the model to the Wgpu backend"); log::info!("Loading the model to the Wgpu backend");
let start = Instant::now(); let start = Instant::now();
let device = WgpuDevice::default(); let device = WgpuDevice::default();
init_device::<AutoGraphicsApi>(&device, Default::default()).await; init_setup_async::<AutoGraphicsApi>(&device, Default::default()).await;
self.model = ModelType::WithWgpuBackend(Model::new(&device)); self.model = ModelType::WithWgpuBackend(Model::new(&device));
let duration = start.elapsed(); let duration = start.elapsed();
log::debug!("Model is loaded to the Wgpu backend in {:?}", duration); log::debug!("Model is loaded to the Wgpu backend in {:?}", duration);