Skip to main content

default_wmma_tile

Function default_wmma_tile 

Source
pub fn default_wmma_tile() -> WmmaTile
Expand description

A sensible default rung-3 config: 128×128 block, BK=32, 8 warps (2×4), so each warp owns a 64×32 warp-tile = 4×2 = 8 fragments. This is the classic WMMA GEMM shape.