Skip to main content

gemm_blk_tail

Function gemm_blk_tail 

Source
pub fn gemm_blk_tail(
    tiles: usize,
    workers: usize,
    stages: usize,
) -> (usize, usize)
Expand description

How a dense block GEMM of tiles tiles and stages stages of k fills its last wave of workers blocks: (whole, split) — the first whole tiles a block each, every later tile cut into split parts of its stages, as many as fill the wave. split 1 when the tiles fill whole waves, or a wave’s blocks cannot cut a remaining tile in two.