Don't check const arguments for SquareType

tracel-ai · Aug 22, 2024 · 4f53194 · 4f53194
1 parent 0a784af
commit 4f53194
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/cubecl-macros-2/src/generate/kernel.rs b/crates/cubecl-macros-2/src/generate/kernel.rs
@@ -22,6 +22,9 @@ impl ToTokens for Kernel {
         let input_checks = self
             .parameters
             .iter()
+            // Const can be anything as long as the accessed fields are cube types, since the access
+            // gets resolved at expansion time and collapsed into a literal in the kernel
+            .filter(|(_, _, is_const)| !is_const)
             .map(|(_, ty, _)| {
                 let span = ty.span();
                 let check = prefix_ir(format_ident!("assert_valid_type"));