Fix clamp min/max line size > 1 (#3078)

Author: laggui

crates/burn-cubecl/src/kernel/clamp.rs

@@ -25,10 +25,11 @@ pub(crate) fn clamp<R: CubeRuntime, E: CubeElement>(
         type Options = Options<N>;
 
         fn execute(input: Line<N>, options: &Self::Options) -> Line<N> {
+            let line_size = input.size();
             Line::clamp(
                 input,
-                Line::new(options.min_value),
-                Line::new(options.max_value),
+                Line::empty(line_size).fill(options.min_value),
+                Line::empty(line_size).fill(options.max_value),
             )
         }
     }

crates/burn-tensor/src/tests/ops/clamp.rs

@@ -70,4 +70,15 @@ mod tests {
             .into_data()
             .assert_eq(&TensorData::from([[1, 1, 2], [3, 4, 4]]), false);
     }
+
+    #[test]
+    fn clamp_min_max_vec_should_compile() {
+        let input = TestTensor::<2>::ones([2, 4], &Default::default());
+        let output = input.clamp(0., 0.5);
+
+        output.into_data().assert_eq(
+            &TensorData::from([[0.5, 0.5, 0.5, 0.5], [0.5, 0.5, 0.5, 0.5]]),
+            false,
+        );
+    }
 }