Made compatible with thumbv6m-none-eabi

Author: BjornTheProgrammer

Cargo.lock

@@ -16,6 +16,7 @@ members = [
 
 exclude = [
     "examples/notebook",
+    "examples/onnx-inference-rp2040", # will cause dependency building issues otherwise
     # "crates/burn-cuda",     # comment this line to work on burn-cuda
 ]
 
@@ -73,7 +74,7 @@ serde_bytes = { version = "0.11.15", default-features = false, features = [
 ] } # alloc for no_std
 serde_rusqlite = "0.35.0"
 serial_test = "3.1.1"
-spin = { version = "0.9.8", features = ["mutex", "spin_mutex"] }
+spin = { version = "0.9.8", features = ["mutex", "spin_mutex", "portable-atomic"] }
 strum = "0.26.3"
 strum_macros = "0.26.4"
 syn = { version = "2.0.71", features = ["full", "extra-traits"] }
@@ -119,7 +120,7 @@ half = { version = "2.4.1", features = [
     "num-traits",
     "serde",
 ], default-features = false }
-ndarray = { version = "0.15.6", default-features = false }
+ndarray = { version = "0.16.0", default-features = false }
 matrixmultiply = { version = "0.3.9", default-features = false }
 openblas-src = "0.10.9"
 blas-src = { version = "0.10.0", default-features = false }
@@ -143,9 +144,11 @@ nvml-wrapper = "0.10.0"
 sysinfo = "0.30.13"
 systemstat = "0.2.3"
 
+portable-atomic-util = { version = "0.2.2", features = ["alloc"] }
+
 ### For the main burn branch. ###
-cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "dd274b6515f853ab1c82db410d6fc7697060f69e" }
-cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "dd274b6515f853ab1c82db410d6fc7697060f69e" }
+cubecl = { git = "https://github.com/BjornTheProgrammer/cubecl", default-features = false, rev = "d1d7dff8a6e1246b06e5acce077fb0cc2b8d5f11" }
+cubecl-common = { git = "https://github.com/BjornTheProgrammer/cubecl", default-features = false, rev = "d1d7dff8a6e1246b06e5acce077fb0cc2b8d5f11" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl" }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common" }

Cargo.toml

@@ -116,9 +116,6 @@ derive-new = { workspace = true }
 log = { workspace = true, optional = true }
 rand = { workspace = true, features = ["std_rng"] } # Default enables std
 
-# Using in place of use std::sync::Mutex when std is disabled
-spin = { workspace = true, features = ["mutex", "spin_mutex"] }
-
 # The same implementation of HashMap in std but with no_std support (only alloc crate is needed)
 hashbrown = { workspace = true, features = ["serde"] } # no_std compatible
 
@@ -133,6 +130,10 @@ serde_json = { workspace = true, features = ["alloc"] } #Default enables std
 thiserror = { workspace = true, optional = true }
 regex = { workspace = true, optional = true }
 num-traits = { workspace = true }
+spin = { workspace = true } # Using in place of use std::sync::Mutex when std is disabled
+
+[target.'cfg(not(target_has_atomic = "ptr"))'.dependencies]
+portable-atomic-util = { workspace = true }
 
 [dev-dependencies]
 tempfile = { workspace = true }

crates/burn-core/Cargo.toml

@@ -5,9 +5,14 @@ use crate::module::{
 };
 
 use alloc::string::ToString;
-use alloc::sync::Arc;
 use alloc::vec::Vec;
 
+#[cfg(target_has_atomic = "ptr")]
+use alloc::sync::Arc;
+
+#[cfg(not(target_has_atomic = "ptr"))]
+use portable_atomic_util::Arc;
+
 use burn_common::stub::Mutex;
 use burn_tensor::{
     backend::{AutodiffBackend, Backend},

crates/burn-core/src/module/param/running.rs

@@ -56,6 +56,9 @@ openblas-src = { workspace = true, optional = true }
 rand = { workspace = true }
 spin = { workspace = true }                                                # using in place of use std::sync::Mutex;
 
+[target.'cfg(not(target_has_atomic = "ptr"))'.dependencies]
+portable-atomic-util = { workspace = true }
+
 [dev-dependencies]
 burn-autodiff = { path = "../burn-autodiff", version = "0.14.0", default-features = false, features = [
     "export_tests",

crates/burn-ndarray/Cargo.toml

@@ -0,0 +1,10 @@
+[target.'cfg(all(target_arch = "arm", target_os = "none"))']
+rustflags = ["--cfg", "portable_atomic_critical_section"]
+runner = "probe-rs run --chip RP2040"
+# runner = "elf2uf2-rs -d -s"
+
+[build]
+target = "thumbv6m-none-eabi"        # Cortex-M0 and Cortex-M0+
+
+[env]
+DEFMT_LOG = "debug"