Fix web examples with wgpu (#2433)

tracel-ai · Oct 28, 2024 · 65aa8b5 · 65aa8b5
1 parent b422a65
commit 65aa8b5
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 10 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/examples/image-classification-web/Cargo.toml b/examples/image-classification-web/Cargo.toml
@@ -15,12 +15,9 @@ half_precision = []
 
 [dependencies]
 burn = { path = "../../crates/burn", version = "0.15.0", default-features = false, features = [
-    "ndarray",
-] }
-cubecl = { workspace = true, features = ["wgpu"] }
-burn-wgpu = { path = "../../crates/burn-wgpu", version = "0.15.0", default-features = false, features = [
-    "autotune",
+    "ndarray", "wgpu",
 ] }
+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 }
 
 log = { workspace = true }

diff --git a/examples/image-classification-web/src/web.rs b/examples/image-classification-web/src/web.rs
@@ -10,9 +10,8 @@ use crate::model::{label::LABELS, normalizer::Normalizer, squeezenet::Model as S
 
 use burn::{backend::NdArray, prelude::*, tensor::activation::softmax};
 
+use burn::backend::wgpu::{init_async, AutoGraphicsApi, Wgpu, WgpuDevice};
 use burn_candle::Candle;
-use burn_wgpu::{Wgpu, WgpuDevice};
-use cubecl::wgpu::{init_async, AutoGraphicsApi};
 
 use serde::Serialize;
 use wasm_bindgen::prelude::*;

diff --git a/examples/mnist-inference-web/Cargo.toml b/examples/mnist-inference-web/Cargo.toml
@@ -13,10 +13,11 @@ crate-type = ["cdylib"]
 default = ["ndarray"]
 
 ndarray = ["burn/ndarray"]
-wgpu = ["burn/wgpu"]
+wgpu = ["burn/wgpu", "cubecl-runtime"]
 
 [dependencies]
 burn = { path = "../../crates/burn", default-features = false }
+cubecl-runtime = { version = "0.3.0", optional = true, features = ["channel-mpsc"] } # missing feature flag
 serde = { workspace = true }
 console_error_panic_hook = { workspace = true }