cubecl-matmul 0.8.1

CubeCL Matrix Multiplication Kernels Engine
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

//! Solves full reductions by loading blocks into shared memory.
//! Handles memory movement, bound checks, plane specialization.

pub mod args;
pub mod memory;
pub mod multi_stage;
pub mod read;
pub mod single_stage;

mod base;
mod copy_mechanism;
mod shared;
mod specialization;
mod write;

pub use base::*;
pub use copy_mechanism::*;
pub use shared::*;
pub use specialization::*;
pub use write::*;