UPSTREAM: sched/preempt: Fix preempt_count manipulations