sched: Implement smarter wake-affine logic