Skip to main content

emit_wmma

Function emit_wmma 

Source
pub fn emit_wmma(kind: MmaKind) -> String
Expand description

Emit a WMMA tensor-core GEMM kernel for a tensor-core MmaKind — the SAME 16×16×16 fragment loop as the int8 IMMA path, over the operand/accumulator types the precision fixes (bf16/f16 → f32 accumulate). Row-major A(M×K)·B(K×N). (Not for Dp4aI8, which is a CUDA-core kernel; use emit_tiled_dp4a.)