pub static SHAPES: Mutex<Vec<((usize, usize, usize), (u64, u64))>>Expand description
(n, k, m) → (calls, ns). A handful of shapes dominate a training step; naming them decides which ones are worth a GPU submit and which are cheaper computed in place.