Skip to main content

emit_wmma_tiled_program

logicaffeine_compile::codegen_cuda::gemm_tile

Function emit_wmma_tiled_program 

Source 
pub fn emit_wmma_tiled_program(
    kind: MmaKind,
    tile: &WmmaTile,
    m: usize,
    k: usize,
    n: usize,
) -> String
Expand description

Checksum-matching pipelined-WMMA GEMM program for a tensor-core kind (bf16/f16) — the shared-memory, warp-tiled kernel with the matmul emitters’ fill (a[p]=(p%7)-3, b[p]=(p%5)-2) and the Σ C integer checksum, so on int8-range operands (exact in bf16/f16) it equals the scalar oracle byte-for-byte.