use web-time in case of wasm (#1060)

burn-common/Cargo.toml

@@ -20,6 +20,7 @@ wasm-sync = []
 [target.'cfg(target_family = "wasm")'.dependencies]
 async-trait = { workspace = true }
 getrandom = { workspace = true, features = ["js"] }
+web-time = { version = "0.2.3" }
 
 [dependencies]
 # ** Please make sure all dependencies support no_std when std is disabled **

burn-common/src/benchmark.rs

@@ -4,8 +4,10 @@ use alloc::vec::Vec;
 use core::fmt::Display;
 use core::time::Duration;
 
-#[cfg(feature = "std")]
+#[cfg(all(not(target_family = "wasm"), feature = "std"))]
 use std::time::Instant;
+#[cfg(all(target_family = "wasm", feature = "std"))]
+use web_time::Instant;
 
 /// Results of a benchmark run.
 #[derive(new, Debug)]

burn-compute/Cargo.toml

@@ -31,5 +31,8 @@ spin = { workspace = true }
 log = { workspace = true }
 hashbrown = { workspace = true }
 
+[target.'cfg(target_family = "wasm")'.dependencies]
+web-time = { version = "0.2.3" }
+
 [dev-dependencies]
 serial_test = "2.0.0"

burn-compute/src/memory_management/simple.rs

@@ -6,6 +6,11 @@ use crate::{
 use alloc::{sync::Arc, vec::Vec};
 use hashbrown::HashMap;
 
+#[cfg(all(not(target_family = "wasm"), feature = "std"))]
+use std::time;
+#[cfg(all(target_family = "wasm", feature = "std"))]
+use web_time as time;
+
 // The ChunkId allows to keep track of how many references there are to a specific chunk.
 memory_id_type!(ChunkId);
 // The SliceId allows to keep track of how many references there are to a specific slice.
@@ -48,9 +53,9 @@ pub enum DeallocStrategy {
     /// Once every period of time
     PeriodTime {
         /// Number of time before triggering the deallocation.
-        period: std::time::Duration,
+        period: time::Duration,
         /// Current state. Should start at now.
-        state: std::time::Instant,
+        state: time::Instant,
     },
     /// Never deallocate.
     Never,
@@ -100,7 +105,7 @@ impl DeallocStrategy {
             #[cfg(feature = "std")]
             DeallocStrategy::PeriodTime { period, state } => {
                 if &state.elapsed() > period {
-                    *state = std::time::Instant::now();
+                    *state = time::Instant::now();
                     true
                 } else {
                     false

burn-compute/src/tune/tuner.rs

@@ -1,4 +1,8 @@
 use core::marker::PhantomData;
+#[cfg(target_family = "wasm")]
+use web_time::Duration;
+
+#[cfg(not(target_family = "wasm"))]
 use core::time::Duration;
 
 use alloc::boxed::Box;