[PATCH 6.1 088/183] sched/psi: Rearrange polling code in preparation