Skip to content

Commit 65aa8b5

Browse files
authored
Fix web examples with wgpu (#2433)
1 parent b422a65 commit 65aa8b5

File tree

4 files changed

+7
-10
lines changed

4 files changed

+7
-10
lines changed

Cargo.lock

+2-2
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

examples/image-classification-web/Cargo.toml

+2-5
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,9 @@ half_precision = []
1515

1616
[dependencies]
1717
burn = { path = "../../crates/burn", version = "0.15.0", default-features = false, features = [
18-
"ndarray",
19-
] }
20-
cubecl = { workspace = true, features = ["wgpu"] }
21-
burn-wgpu = { path = "../../crates/burn-wgpu", version = "0.15.0", default-features = false, features = [
22-
"autotune",
18+
"ndarray", "wgpu",
2319
] }
20+
cubecl-runtime = { version = "0.3.0", features = ["channel-mpsc"] } # missing feature flags
2421
burn-candle = { path = "../../crates/burn-candle", version = "0.15.0", default-features = false }
2522

2623
log = { workspace = true }

examples/image-classification-web/src/web.rs

+1-2
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,8 @@ use crate::model::{label::LABELS, normalizer::Normalizer, squeezenet::Model as S
1010

1111
use burn::{backend::NdArray, prelude::*, tensor::activation::softmax};
1212

13+
use burn::backend::wgpu::{init_async, AutoGraphicsApi, Wgpu, WgpuDevice};
1314
use burn_candle::Candle;
14-
use burn_wgpu::{Wgpu, WgpuDevice};
15-
use cubecl::wgpu::{init_async, AutoGraphicsApi};
1615

1716
use serde::Serialize;
1817
use wasm_bindgen::prelude::*;

examples/mnist-inference-web/Cargo.toml

+2-1
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,11 @@ crate-type = ["cdylib"]
1313
default = ["ndarray"]
1414

1515
ndarray = ["burn/ndarray"]
16-
wgpu = ["burn/wgpu"]
16+
wgpu = ["burn/wgpu", "cubecl-runtime"]
1717

1818
[dependencies]
1919
burn = { path = "../../crates/burn", default-features = false }
20+
cubecl-runtime = { version = "0.3.0", optional = true, features = ["channel-mpsc"] } # missing feature flag
2021
serde = { workspace = true }
2122
console_error_panic_hook = { workspace = true }
2223

0 commit comments

Comments
 (0)