Fixed ndarray deprecated warnings

Author: BjornTheProgrammer

Cargo.lock

@@ -151,8 +151,8 @@ 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 = "30d090c8b725e83a829a6459d9c09578d9a4ba68" }
-cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "30d090c8b725e83a829a6459d9c09578d9a4ba68" }
+cubecl = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "034f667da6e92a81b7da9f303e8507db944cc2a4" }
+cubecl-common = { git = "https://github.com/tracel-ai/cubecl", default-features = false, rev = "034f667da6e92a81b7da9f303e8507db944cc2a4" }
 ### For local development. ###
 # cubecl = { path = "../cubecl/crates/cubecl" }
 # cubecl-common = { path = "../cubecl/crates/cubecl-common" }

Cargo.toml

@@ -612,7 +612,7 @@ fn arg<E: NdArrayElement, const D: usize>(
         idx as i64
     });
 
-    let output = output.into_shape(Dim(reshape.as_slice())).unwrap();
+    let output = output.into_shape_with_order(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([batch_size, out_channels, out_height, out_width])
+        .into_shape_with_order([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([batch_size, out_channels, out_depth, out_height, out_width])
+        .into_shape_with_order([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(dim)
+                .into_shape_with_order(dim)
                 .expect("Safe to change shape without relayout")
                 .into_shared(),
-            false => $array.reshape(dim),
+            false => $array.into_shape_with_order(dim).unwrap(),
         };
         let array = array.into_dyn();
 

crates/burn-ndarray/src/tensor.rs

@@ -12,13 +12,15 @@ Shell commands to build and test the package:
 ```sh
 
 # install the new targets if not installed previously
+rustup target add thumbv6m-none-eabi
 rustup target add thumbv7m-none-eabi
 rustup target add wasm32-unknown-unknown
 
 # build for various targets 
 cargo build # regular build
 cargo build --target thumbv7m-none-eabi
 cargo build --target wasm32-unknown-unknown
+RUSTFLAGS="--cfg portable_atomic_unsafe_assume_single_core" cargo build --target thumbv6m-none-eabi
 
 # test
 cargo test