Skip to main content

emit_mma_tn_k32swz_kernel

logicaffeine_compile::codegen_cuda::gemm_tile

Function emit_mma_tn_k32swz_kernel 

Source 
pub fn emit_mma_tn_k32swz_kernel(
    bm: usize,
    bn: usize,
    warps_m: usize,
    warps_n: usize,
    stages: usize,
) -> String
Expand description

Emit JUST the swizzled-k32 kernel (no host main) — so the compiler can compile it to a cubin and analyze its SASS (register pressure, occupancy, tensor-core utilisation) via codegen_sass::SassReport.