Added tests

Author: BjornTheProgrammer

Cargo.lock

@@ -16,6 +16,8 @@ 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
 ]
 
 [workspace.package]
@@ -72,7 +74,11 @@ 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.74", features = ["full", "extra-traits"] }
@@ -118,7 +124,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 }
@@ -142,9 +148,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 = "bee7886b5c3016c425d244136f77442655097f3e" }
-cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "bee7886b5c3016c425d244136f77442655097f3e" }
+cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "b09821d1d5bd1ee0cb8d80b04916acb4a9096c29" }
+cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "b09821d1d5bd1ee0cb8d80b04916acb4a9096c29" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl" }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common" }

Cargo.toml

@@ -1,6 +1,6 @@
 # No Standard Library
 
-In this seciton, you will learn how to run an onnx inference model on an embedded system, with no standard library support on a Raspberry Pi Pico. This should be universally applicable to other platforms. All the code can be found under the
+In this section, you will learn how to run an onnx inference model on an embedded system, with no standard library support on a Raspberry Pi Pico. This should be universally applicable to other platforms. All the code can be found under the
 [examples directory](https://github.com/tracel-ai/burn/tree/main/examples/onnx-inference-rp2040).
 
 ## Step-by-Step Guide

burn-book/src/advanced/no-std.md

@@ -612,7 +612,7 @@ fn arg<E: NdArrayElement, const D: usize>(
         idx as i64
     });
 
-    let output = output.into_shape_with_order(Dim(reshape.as_slice())).unwrap();
+    let output = output.to_shape(Dim(reshape.as_slice())).unwrap();
 
     NdArrayTensor {
         array: output.into_shared(),

crates/burn-ndarray/src/ops/base.rs

@@ -209,7 +209,7 @@ pub(crate) fn conv2d<E: FloatNdArrayElement, Q: QuantElement>(
     });
 
     let output = output
-        .into_shape_with_order([batch_size, out_channels, out_height, out_width])
+        .to_shape([batch_size, out_channels, out_height, out_width])
         .unwrap()
         .into_dyn()
         .into_shared();
@@ -437,7 +437,7 @@ pub(crate) fn conv3d<E: FloatNdArrayElement, Q: QuantElement>(
     });
 
     let output = output
-        .into_shape_with_order([batch_size, out_channels, out_depth, out_height, out_width])
+        .to_shape([batch_size, out_channels, out_depth, out_height, out_width])
         .unwrap()
         .into_dyn()
         .into_shared();

crates/burn-ndarray/src/ops/conv.rs

@@ -70,10 +70,10 @@ macro_rules! reshape {
         let dim = $crate::to_typed_dims!($n, $shape.dims, justdim);
         let array: ndarray::ArcArray<$ty, Dim<[usize; $n]>> = match $array.is_standard_layout() {
             true => $array
-                .into_shape_with_order(dim)
+                .to_shape(dim)
                 .expect("Safe to change shape without relayout")
                 .into_shared(),
-            false => $array.into_shape_with_order(dim).unwrap(),
+            false => $array.to_shape(dim).unwrap().as_standard_layout().into_shared(),
         };
         let array = array.into_dyn();
 

crates/burn-ndarray/src/tensor.rs

@@ -39,7 +39,7 @@ async fn main(_spawner: Spawner) {
         // Run the model
         let output = run_model(&model, &device, input);
 
-        // Ouput the values
+        // Output the values
         match output.into_primitive().tensor().array.as_slice() {
             Some(slice) => info!("input: {} - output: {}", input, slice),
             None => defmt::panic!("Failed to get value")

examples/onnx-inference-rp2040/src/bin/main.rs

@@ -22,6 +22,7 @@ use crate::{endgroup, group};
 // Targets constants
 const WASM32_TARGET: &str = "wasm32-unknown-unknown";
 const ARM_TARGET: &str = "thumbv7m-none-eabi";
+const ARM_NO_ATOMIC_PTR_TARGET: &str = "thumbv6m-none-eabi";
 
 #[derive(clap::ValueEnum, Default, Copy, Clone, PartialEq, Eq)]
 pub(crate) enum CheckType {
@@ -81,12 +82,12 @@ impl CheckType {
 }
 
 /// Run cargo build command
-fn cargo_build(params: Params) {
+fn cargo_build(params: Params, envs: Option<HashMap<&str, String>>) {
     // Run cargo build
     run_cargo(
         "build",
         params + "--color=always",
-        HashMap::new(),
+        envs.unwrap_or_default(),
         "Failed to run cargo build",
     );
 }
@@ -155,7 +156,10 @@ fn build_and_test_no_std<const N: usize>(crate_name: &str, extra_args: [&str; N]
     group!("Checks: {} (no-std)", crate_name);
 
     // Run cargo build --no-default-features
-    cargo_build(Params::from(["-p", crate_name, "--no-default-features"]) + extra_args);
+    cargo_build(
+        Params::from(["-p", crate_name, "--no-default-features"]) + extra_args,
+        None,
+    );
 
     // Run cargo test --no-default-features
     cargo_test(Params::from(["-p", crate_name, "--no-default-features"]) + extra_args);
@@ -169,6 +173,7 @@ fn build_and_test_no_std<const N: usize>(crate_name: &str, extra_args: [&str; N]
             "--target",
             WASM32_TARGET,
         ]) + extra_args,
+        None,
     );
 
     // Run cargo build --no-default-features --target thumbv7m-none-eabi
@@ -180,6 +185,22 @@ fn build_and_test_no_std<const N: usize>(crate_name: &str, extra_args: [&str; N]
             "--target",
             ARM_TARGET,
         ]) + extra_args,
+        None,
+    );
+
+    // Run cargo build --no-default-features --target thumbv6m-none-eabi
+    cargo_build(
+        Params::from([
+            "-p",
+            crate_name,
+            "--no-default-features",
+            "--target",
+            ARM_NO_ATOMIC_PTR_TARGET,
+        ]) + extra_args,
+        Some(HashMap::from([(
+            "RUSTFLAGS",
+            "--cfg portable_atomic_unsafe_assume_single_core".to_string(),
+        )])),
     );
 
     endgroup!();
@@ -228,6 +249,9 @@ fn no_std_checks() {
     // Install ARM target
     rustup_add_target(ARM_TARGET);
 
+    // Install ARM no atomic ptr target
+    rustup_add_target(ARM_NO_ATOMIC_PTR_TARGET);
+
     // Run checks for the following crates
     build_and_test_no_std("burn", []);
     build_and_test_no_std("burn-core", []);
@@ -265,7 +289,7 @@ fn burn_dataset_features_std() {
     group!("Checks: burn-dataset (all-features)");
 
     // Run cargo build --all-features
-    cargo_build(["-p", "burn-dataset", "--all-features"].into());
+    cargo_build(["-p", "burn-dataset", "--all-features"].into(), None);
 
     // Run cargo test --all-features
     cargo_test(["-p", "burn-dataset", "--all-features"].into());
@@ -334,7 +358,7 @@ fn std_checks() {
         }
 
         group!("Checks: {}", member.name);
-        cargo_build(Params::from(["-p", &member.name]));
+        cargo_build(Params::from(["-p", &member.name]), None);
         cargo_test(Params::from(["-p", &member.name]));
         endgroup!();
     }
@@ -373,6 +397,7 @@ fn check_typos() {
 
     // Run typos command as child process
     let typos = Command::new("typos")
+        .args(["--exclude", "**/*.onnx"])
         .stdout(Stdio::inherit()) // Send stdout directly to terminal
         .stderr(Stdio::inherit()) // Send stderr directly to terminal
         .spawn()