Skip to main content

emit_wmma_program

Function emit_wmma_program 

Source
pub fn emit_wmma_program(kind: MmaKind, m: usize, k: usize, n: usize) -> String
Expand description

Full self-contained WMMA program for M×K×N at MmaKind kind — fills A/B with small integers (exact in bf16/f16/int8), runs one warp per 16×16 tile, and prints Σ C (a double). Because the operands are exact small integers and the accumulate is wide (f32/int32), Σ C is an exact integer, so a scalar integer GEMM is the ≡ oracle regardless of precision.