Function matrix_working_set 

pub fn matrix_working_set(
    m: usize,
    n: usize,
    k: usize,
    element_size: usize,
) -> usize

Calculate working set size for matrix operations