sched: update memory compute to support flexiblity