Skip to main content

select_mma

logicaffeine_compile::codegen_cuda::gemm_tile

Function select_mma 

Source 
pub fn select_mma(
    prec: Precision,
    m: usize,
    k: usize,
    n: usize,
    sm_arch: u32,
) -> MmaKind
Expand description

Select the MMA kind for a (precision, shape, sm_arch). Tensor-core WMMA when the shape is 16-aligned and the arch provides the precision’s HMMA/IMMA; otherwise the always-legal DP4A CUDA-core fallback — we never emit a tensor-core kernel on a shape or GPU that cannot run it.