Skip to main content

default_tile_for

oxicuda_backend

Function default_tile_for 

Source 
pub fn default_tile_for(
    m: usize,
    n: usize,
    k: usize,
    caps: &Capabilities,
) -> TileShape
Expand description

Default tile heuristic shared by the trait’s recommended_tile_for: small problems get a small tile (less wasted compute on the fringe), large problems get the canonical 128×128×8 tile, mid-size gets 64×64×16.

caps lets the heuristic snap to a Tensor-Core-friendly tile when the backend reports matrix units.