clippy

tracel-ai · Apr 29, 2024 · 24d8c31 · 24d8c31
1 parent a238be1
commit 24d8c31
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/crates/burn-jit/src/kernel/matmul/base.rs b/crates/burn-jit/src/kernel/matmul/base.rs
@@ -85,6 +85,7 @@ impl Default for Tiling2dConfig {
 }
 
 /// The strategy to be used when launching a matmul kernel.
+#[derive(Default)]
 pub enum MatmulStrategy {
     /// A simple kernel will be used with memory coalescing optimization.
     Simple {
@@ -99,15 +100,12 @@ pub enum MatmulStrategy {
     Tiling2dPadded(Tiling2dConfig),
     #[cfg(feature = "autotune")]
     /// Using autotune to chose the best kernel based on runtime information.
+    #[default]
     Autotune,
 }
 
 #[cfg(feature = "autotune")]
-impl Default for MatmulStrategy {
-    fn default() -> Self {
-        MatmulStrategy::Autotune
-    }
-}
+
 
 #[cfg(not(feature = "autotune"))]
 impl Default for MatmulStrategy {

diff --git a/crates/burn-jit/src/kernel/reduce/base.rs b/crates/burn-jit/src/kernel/reduce/base.rs
@@ -101,19 +101,17 @@ pub fn init_reduce_output<R: Runtime, EI: JitElement, EO: JitElement, const D: u
 
 #[derive(Copy, Clone, Debug)]
 #[allow(missing_docs)]
+#[derive(Default)]
 pub enum ReduceStrategy {
     Naive,
     SharedMemory,
     #[cfg(feature = "autotune")]
+    #[default]
     Autotune,
 }
 
 #[cfg(feature = "autotune")]
-impl Default for ReduceStrategy {
-    fn default() -> Self {
-        ReduceStrategy::Autotune
-    }
-}
+
 
 #[cfg(not(feature = "autotune"))]
 impl Default for ReduceStrategy {