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

Mnist inference web: Remove wee_alloc allocator and use rust's alloc for WASM32 (#231)

This commit is contained in:
Dilshod Tadjibaev 2023-03-14 07:57:10 -05:00 committed by GitHub
parent 7f3f8d2b22
commit e991ba38df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/mnist-inference-web/Cargo.toml
View File

@ -21,7 +21,6 @@ bincode = {version = "2.0.0-rc.2", git = "https://github.com/bincode-org/bincode
serde = {workspace = true}
wasm-bindgen = "0.2.84"
wee_alloc = "0.4.5"
[dev-dependencies]
burn-dataset = {path = "../../burn-dataset"}

6
examples/mnist-inference-web/src/web.rs
View File

@ -9,12 +9,6 @@ use burn::tensor::Tensor;
use wasm_bindgen::prelude::*;
#[allow(unused_imports)]
use wee_alloc::WeeAlloc;
#[global_allocator]
static ALLOC: wee_alloc::WeeAlloc = wee_alloc::WeeAlloc::INIT;
/// Mnist structure that corresponds to JavaScript class.
/// See: https://rustwasm.github.io/wasm-bindgen/contributing/design/exporting-rust-struct.html
#[wasm_bindgen]