sched: Micro-optimize the smart wake-affine logic