pub fn wmma_legal(m: usize, k: usize, n: usize) -> bool
A shape is WMMA-legal iff every dimension is a multiple of the int8 fragment size (16).