cubecl-matmul 0.8.1

CubeCL Matrix Multiplication Kernels Engine
Documentation 
1
2
3
4
5
6
7
8
9
10

mod algorithm;
mod base;
mod selector;

pub use algorithm::*;
pub use base::{Selection, launch, launch_ref, launch_with_config, matmul_cmma_tma_ref_no_check};
pub use selector::{
    NUM_SM_APPROX, NUM_TENSOR_CORES_APPROX, TileSizeSelection, find_instruction_size,
    launch_kernel_concrete, launch_kernel_virtual,
};