Function matmul_workgroups 

pub const fn matmul_workgroups(
    m: u32,
    n: u32,
    batch: u32,
    tile: u32,
) -> [u32; 3]

Workgroup counts of a tiled MATMUL over m rows, n columns, and batch batches.