sched: Extract the task putting code from pick_next_task()

	schedstat_inc(this_rq()->sched_count);

}

static void put_prev_task_balance(struct rq *rq, struct task_struct *prev,

				  struct rq_flags *rf)

{

#ifdef CONFIG_SMP

	const struct sched_class *class;

	/*

	 * We must do the balancing pass before put_prev_task(), such

	 * that when we release the rq->lock the task is in the same

	 * state as before we took rq->lock.

	 *

	 * We can terminate the balance pass as soon as we know there is

	 * a runnable task of @class priority or higher.

	 */

	for_class_range(class, prev->sched_class, &idle_sched_class) {

		if (class->balance(rq, prev, rf))

			break;

	}

#endif

	put_prev_task(rq, prev);

}

/*

 * Pick up the highest-prio task:

 */

	}

restart:

#ifdef CONFIG_SMP

	/*

	 * We must do the balancing pass before put_next_task(), such

	 * that when we release the rq->lock the task is in the same

	 * state as before we took rq->lock.

	 *

	 * We can terminate the balance pass as soon as we know there is

	 * a runnable task of @class priority or higher.

	 */

	for_class_range(class, prev->sched_class, &idle_sched_class) {

		if (class->balance(rq, prev, rf))

			break;

	}

#endif

	put_prev_task(rq, prev);

	put_prev_task_balance(rq, prev, rf);

	for_each_class(class) {

		p = class->pick_next_task(rq);