Fix CI (#2268)

.github/workflows/test.yml

@@ -31,7 +31,7 @@ env:
   # Note: It is not possible to define env vars in composite actions.
   # To work around this issue we use inputs and define all the env vars here.
 
-  RUST_PREVIOUS_VERSION: 1.79.0
+  RUST_PREVIOUS_VERSION: 1.80.0
 
   # Cargo
   CARGO_TERM_COLOR: "always"

crates/burn-jit/src/kernel/conv/conv2d.rs

@@ -41,11 +41,10 @@ fn conv2d_kernel<F: Float>(
 
     let in_channels = weight.shape(1);
 
-    let kernel_size_0 = kernel_size_0_unroll.unwrap_or_else(|| weight.shape(2));
+    let kernel_size_0 = weight.shape(2);
     let kernel_size_1 = kernel_size_1_unroll.unwrap_or_else(|| weight.shape(3));
     let unroll_1 = kernel_size_1_unroll.is_some();
 
-
     let b = ABSOLUTE_POS / output.stride(0) % output.shape(0);
     let oc = ABSOLUTE_POS / output.stride(1) % output.shape(1);
     let oh = ABSOLUTE_POS / output.stride(2) % output.shape(2);
@@ -130,7 +129,7 @@ pub(crate) fn conv2d<R: JitRuntime, E: FloatElement>(
     let kernel_1_unroll = if kernel_1 > 8 {
         None
     } else {
-        Some(kernel_1.into())
+        Some(kernel_1 as u32)
     };
 
     let out_0 = calculate_conv_output_size(
@@ -188,7 +187,7 @@ pub(crate) fn conv2d<R: JitRuntime, E: FloatElement>(
             ScalarArg::new(options.padding[1] as u32),
             ScalarArg::new(options.groups as u32),
         ),
-        Some(kernel_1 as u32),
+        kernel_1_unroll,
     );
 
     output

crates/burn/Cargo.toml

@@ -9,7 +9,7 @@ name = "burn"
 readme.workspace = true
 repository = "https://github.com/tracel-ai/burn"
 version.workspace = true
-rust-version = "1.79"
+rust-version = "1.80"
 
 [features]
 default = ["burn-core/default", "burn-train?/default", "std"]