Skip to main content

emit_wmma_cpasync_bench

Function emit_wmma_cpasync_bench 

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

Timing bench for the cp.async double-buffered WMMA kernel at kind + tile + stages — prints GOPS.