cubecl_attention/
lib.rs

1#![allow(clippy::explicit_counter_loop)]
2#![allow(clippy::manual_is_multiple_of)]
3
4mod base;
5/// Components for matrix multiplication
6pub mod components;
7/// Contains matmul kernels
8pub mod kernels;
9/// Tests for matmul kernels
10#[cfg(feature = "export_tests")]
11pub mod tests;
12
13pub use base::*;