mm: print out the zonelists on request for manual verification
[sfrench/cifs-2.6.git] / kernel / sched_fair.c
1 /*
2  * Completely Fair Scheduling (CFS) Class (SCHED_NORMAL/SCHED_BATCH)
3  *
4  *  Copyright (C) 2007 Red Hat, Inc., Ingo Molnar <mingo@redhat.com>
5  *
6  *  Interactivity improvements by Mike Galbraith
7  *  (C) 2007 Mike Galbraith <efault@gmx.de>
8  *
9  *  Various enhancements by Dmitry Adamushko.
10  *  (C) 2007 Dmitry Adamushko <dmitry.adamushko@gmail.com>
11  *
12  *  Group scheduling enhancements by Srivatsa Vaddagiri
13  *  Copyright IBM Corporation, 2007
14  *  Author: Srivatsa Vaddagiri <vatsa@linux.vnet.ibm.com>
15  *
16  *  Scaled math optimizations by Thomas Gleixner
17  *  Copyright (C) 2007, Thomas Gleixner <tglx@linutronix.de>
18  *
19  *  Adaptive scheduling granularity, math enhancements by Peter Zijlstra
20  *  Copyright (C) 2007 Red Hat, Inc., Peter Zijlstra <pzijlstr@redhat.com>
21  */
22
23 #include <linux/latencytop.h>
24
25 /*
26  * Targeted preemption latency for CPU-bound tasks:
27  * (default: 20ms * (1 + ilog(ncpus)), units: nanoseconds)
28  *
29  * NOTE: this latency value is not the same as the concept of
30  * 'timeslice length' - timeslices in CFS are of variable length
31  * and have no persistent notion like in traditional, time-slice
32  * based scheduling concepts.
33  *
34  * (to see the precise effective timeslice length of your workload,
35  *  run vmstat and monitor the context-switches (cs) field)
36  */
37 unsigned int sysctl_sched_latency = 20000000ULL;
38
39 /*
40  * Minimal preemption granularity for CPU-bound tasks:
41  * (default: 4 msec * (1 + ilog(ncpus)), units: nanoseconds)
42  */
43 unsigned int sysctl_sched_min_granularity = 4000000ULL;
44
45 /*
46  * is kept at sysctl_sched_latency / sysctl_sched_min_granularity
47  */
48 static unsigned int sched_nr_latency = 5;
49
50 /*
51  * After fork, child runs first. (default) If set to 0 then
52  * parent will (try to) run first.
53  */
54 const_debug unsigned int sysctl_sched_child_runs_first = 1;
55
56 /*
57  * sys_sched_yield() compat mode
58  *
59  * This option switches the agressive yield implementation of the
60  * old scheduler back on.
61  */
62 unsigned int __read_mostly sysctl_sched_compat_yield;
63
64 /*
65  * SCHED_OTHER wake-up granularity.
66  * (default: 5 msec * (1 + ilog(ncpus)), units: nanoseconds)
67  *
68  * This option delays the preemption effects of decoupled workloads
69  * and reduces their over-scheduling. Synchronous workloads will still
70  * have immediate wakeup/sleep latencies.
71  */
72 unsigned int sysctl_sched_wakeup_granularity = 5000000UL;
73
74 const_debug unsigned int sysctl_sched_migration_cost = 500000UL;
75
76 /**************************************************************
77  * CFS operations on generic schedulable entities:
78  */
79
80 static inline struct task_struct *task_of(struct sched_entity *se)
81 {
82         return container_of(se, struct task_struct, se);
83 }
84
85 #ifdef CONFIG_FAIR_GROUP_SCHED
86
87 /* cpu runqueue to which this cfs_rq is attached */
88 static inline struct rq *rq_of(struct cfs_rq *cfs_rq)
89 {
90         return cfs_rq->rq;
91 }
92
93 /* An entity is a task if it doesn't "own" a runqueue */
94 #define entity_is_task(se)      (!se->my_q)
95
96 /* Walk up scheduling entities hierarchy */
97 #define for_each_sched_entity(se) \
98                 for (; se; se = se->parent)
99
100 static inline struct cfs_rq *task_cfs_rq(struct task_struct *p)
101 {
102         return p->se.cfs_rq;
103 }
104
105 /* runqueue on which this entity is (to be) queued */
106 static inline struct cfs_rq *cfs_rq_of(struct sched_entity *se)
107 {
108         return se->cfs_rq;
109 }
110
111 /* runqueue "owned" by this group */
112 static inline struct cfs_rq *group_cfs_rq(struct sched_entity *grp)
113 {
114         return grp->my_q;
115 }
116
117 /* Given a group's cfs_rq on one cpu, return its corresponding cfs_rq on
118  * another cpu ('this_cpu')
119  */
120 static inline struct cfs_rq *cpu_cfs_rq(struct cfs_rq *cfs_rq, int this_cpu)
121 {
122         return cfs_rq->tg->cfs_rq[this_cpu];
123 }
124
125 /* Iterate thr' all leaf cfs_rq's on a runqueue */
126 #define for_each_leaf_cfs_rq(rq, cfs_rq) \
127         list_for_each_entry_rcu(cfs_rq, &rq->leaf_cfs_rq_list, leaf_cfs_rq_list)
128
129 /* Do the two (enqueued) entities belong to the same group ? */
130 static inline int
131 is_same_group(struct sched_entity *se, struct sched_entity *pse)
132 {
133         if (se->cfs_rq == pse->cfs_rq)
134                 return 1;
135
136         return 0;
137 }
138
139 static inline struct sched_entity *parent_entity(struct sched_entity *se)
140 {
141         return se->parent;
142 }
143
144 #else   /* CONFIG_FAIR_GROUP_SCHED */
145
146 static inline struct rq *rq_of(struct cfs_rq *cfs_rq)
147 {
148         return container_of(cfs_rq, struct rq, cfs);
149 }
150
151 #define entity_is_task(se)      1
152
153 #define for_each_sched_entity(se) \
154                 for (; se; se = NULL)
155
156 static inline struct cfs_rq *task_cfs_rq(struct task_struct *p)
157 {
158         return &task_rq(p)->cfs;
159 }
160
161 static inline struct cfs_rq *cfs_rq_of(struct sched_entity *se)
162 {
163         struct task_struct *p = task_of(se);
164         struct rq *rq = task_rq(p);
165
166         return &rq->cfs;
167 }
168
169 /* runqueue "owned" by this group */
170 static inline struct cfs_rq *group_cfs_rq(struct sched_entity *grp)
171 {
172         return NULL;
173 }
174
175 static inline struct cfs_rq *cpu_cfs_rq(struct cfs_rq *cfs_rq, int this_cpu)
176 {
177         return &cpu_rq(this_cpu)->cfs;
178 }
179
180 #define for_each_leaf_cfs_rq(rq, cfs_rq) \
181                 for (cfs_rq = &rq->cfs; cfs_rq; cfs_rq = NULL)
182
183 static inline int
184 is_same_group(struct sched_entity *se, struct sched_entity *pse)
185 {
186         return 1;
187 }
188
189 static inline struct sched_entity *parent_entity(struct sched_entity *se)
190 {
191         return NULL;
192 }
193
194 #endif  /* CONFIG_FAIR_GROUP_SCHED */
195
196
197 /**************************************************************
198  * Scheduling class tree data structure manipulation methods:
199  */
200
201 static inline u64 max_vruntime(u64 min_vruntime, u64 vruntime)
202 {
203         s64 delta = (s64)(vruntime - min_vruntime);
204         if (delta > 0)
205                 min_vruntime = vruntime;
206
207         return min_vruntime;
208 }
209
210 static inline u64 min_vruntime(u64 min_vruntime, u64 vruntime)
211 {
212         s64 delta = (s64)(vruntime - min_vruntime);
213         if (delta < 0)
214                 min_vruntime = vruntime;
215
216         return min_vruntime;
217 }
218
219 static inline s64 entity_key(struct cfs_rq *cfs_rq, struct sched_entity *se)
220 {
221         return se->vruntime - cfs_rq->min_vruntime;
222 }
223
224 /*
225  * Enqueue an entity into the rb-tree:
226  */
227 static void __enqueue_entity(struct cfs_rq *cfs_rq, struct sched_entity *se)
228 {
229         struct rb_node **link = &cfs_rq->tasks_timeline.rb_node;
230         struct rb_node *parent = NULL;
231         struct sched_entity *entry;
232         s64 key = entity_key(cfs_rq, se);
233         int leftmost = 1;
234
235         /*
236          * Find the right place in the rbtree:
237          */
238         while (*link) {
239                 parent = *link;
240                 entry = rb_entry(parent, struct sched_entity, run_node);
241                 /*
242                  * We dont care about collisions. Nodes with
243                  * the same key stay together.
244                  */
245                 if (key < entity_key(cfs_rq, entry)) {
246                         link = &parent->rb_left;
247                 } else {
248                         link = &parent->rb_right;
249                         leftmost = 0;
250                 }
251         }
252
253         /*
254          * Maintain a cache of leftmost tree entries (it is frequently
255          * used):
256          */
257         if (leftmost) {
258                 cfs_rq->rb_leftmost = &se->run_node;
259                 /*
260                  * maintain cfs_rq->min_vruntime to be a monotonic increasing
261                  * value tracking the leftmost vruntime in the tree.
262                  */
263                 cfs_rq->min_vruntime =
264                         max_vruntime(cfs_rq->min_vruntime, se->vruntime);
265         }
266
267         rb_link_node(&se->run_node, parent, link);
268         rb_insert_color(&se->run_node, &cfs_rq->tasks_timeline);
269 }
270
271 static void __dequeue_entity(struct cfs_rq *cfs_rq, struct sched_entity *se)
272 {
273         if (cfs_rq->rb_leftmost == &se->run_node) {
274                 struct rb_node *next_node;
275                 struct sched_entity *next;
276
277                 next_node = rb_next(&se->run_node);
278                 cfs_rq->rb_leftmost = next_node;
279
280                 if (next_node) {
281                         next = rb_entry(next_node,
282                                         struct sched_entity, run_node);
283                         cfs_rq->min_vruntime =
284                                 max_vruntime(cfs_rq->min_vruntime,
285                                              next->vruntime);
286                 }
287         }
288
289         if (cfs_rq->next == se)
290                 cfs_rq->next = NULL;
291
292         rb_erase(&se->run_node, &cfs_rq->tasks_timeline);
293 }
294
295 static inline struct rb_node *first_fair(struct cfs_rq *cfs_rq)
296 {
297         return cfs_rq->rb_leftmost;
298 }
299
300 static struct sched_entity *__pick_next_entity(struct cfs_rq *cfs_rq)
301 {
302         return rb_entry(first_fair(cfs_rq), struct sched_entity, run_node);
303 }
304
305 static inline struct sched_entity *__pick_last_entity(struct cfs_rq *cfs_rq)
306 {
307         struct rb_node *last = rb_last(&cfs_rq->tasks_timeline);
308
309         if (!last)
310                 return NULL;
311
312         return rb_entry(last, struct sched_entity, run_node);
313 }
314
315 /**************************************************************
316  * Scheduling class statistics methods:
317  */
318
319 #ifdef CONFIG_SCHED_DEBUG
320 int sched_nr_latency_handler(struct ctl_table *table, int write,
321                 struct file *filp, void __user *buffer, size_t *lenp,
322                 loff_t *ppos)
323 {
324         int ret = proc_dointvec_minmax(table, write, filp, buffer, lenp, ppos);
325
326         if (ret || !write)
327                 return ret;
328
329         sched_nr_latency = DIV_ROUND_UP(sysctl_sched_latency,
330                                         sysctl_sched_min_granularity);
331
332         return 0;
333 }
334 #endif
335
336 /*
337  * delta *= w / rw
338  */
339 static inline unsigned long
340 calc_delta_weight(unsigned long delta, struct sched_entity *se)
341 {
342         for_each_sched_entity(se) {
343                 delta = calc_delta_mine(delta,
344                                 se->load.weight, &cfs_rq_of(se)->load);
345         }
346
347         return delta;
348 }
349
350 /*
351  * delta *= rw / w
352  */
353 static inline unsigned long
354 calc_delta_fair(unsigned long delta, struct sched_entity *se)
355 {
356         for_each_sched_entity(se) {
357                 delta = calc_delta_mine(delta,
358                                 cfs_rq_of(se)->load.weight, &se->load);
359         }
360
361         return delta;
362 }
363
364 /*
365  * The idea is to set a period in which each task runs once.
366  *
367  * When there are too many tasks (sysctl_sched_nr_latency) we have to stretch
368  * this period because otherwise the slices get too small.
369  *
370  * p = (nr <= nl) ? l : l*nr/nl
371  */
372 static u64 __sched_period(unsigned long nr_running)
373 {
374         u64 period = sysctl_sched_latency;
375         unsigned long nr_latency = sched_nr_latency;
376
377         if (unlikely(nr_running > nr_latency)) {
378                 period = sysctl_sched_min_granularity;
379                 period *= nr_running;
380         }
381
382         return period;
383 }
384
385 /*
386  * We calculate the wall-time slice from the period by taking a part
387  * proportional to the weight.
388  *
389  * s = p*w/rw
390  */
391 static u64 sched_slice(struct cfs_rq *cfs_rq, struct sched_entity *se)
392 {
393         return calc_delta_weight(__sched_period(cfs_rq->nr_running), se);
394 }
395
396 /*
397  * We calculate the vruntime slice of a to be inserted task
398  *
399  * vs = s*rw/w = p
400  */
401 static u64 sched_vslice_add(struct cfs_rq *cfs_rq, struct sched_entity *se)
402 {
403         unsigned long nr_running = cfs_rq->nr_running;
404
405         if (!se->on_rq)
406                 nr_running++;
407
408         return __sched_period(nr_running);
409 }
410
411 /*
412  * The goal of calc_delta_asym() is to be asymmetrically around NICE_0_LOAD, in
413  * that it favours >=0 over <0.
414  *
415  *   -20         |
416  *               |
417  *     0 --------+-------
418  *             .'
419  *    19     .'
420  *
421  */
422 static unsigned long
423 calc_delta_asym(unsigned long delta, struct sched_entity *se)
424 {
425         struct load_weight lw = {
426                 .weight = NICE_0_LOAD,
427                 .inv_weight = 1UL << (WMULT_SHIFT-NICE_0_SHIFT)
428         };
429
430         for_each_sched_entity(se) {
431                 struct load_weight *se_lw = &se->load;
432                 unsigned long rw = cfs_rq_of(se)->load.weight;
433
434 #ifdef CONFIG_FAIR_SCHED_GROUP
435                 struct cfs_rq *cfs_rq = se->my_q;
436                 struct task_group *tg = NULL
437
438                 if (cfs_rq)
439                         tg = cfs_rq->tg;
440
441                 if (tg && tg->shares < NICE_0_LOAD) {
442                         /*
443                          * scale shares to what it would have been had
444                          * tg->weight been NICE_0_LOAD:
445                          *
446                          *   weight = 1024 * shares / tg->weight
447                          */
448                         lw.weight *= se->load.weight;
449                         lw.weight /= tg->shares;
450
451                         lw.inv_weight = 0;
452
453                         se_lw = &lw;
454                         rw += lw.weight - se->load.weight;
455                 } else
456 #endif
457
458                 if (se->load.weight < NICE_0_LOAD) {
459                         se_lw = &lw;
460                         rw += NICE_0_LOAD - se->load.weight;
461                 }
462
463                 delta = calc_delta_mine(delta, rw, se_lw);
464         }
465
466         return delta;
467 }
468
469 /*
470  * Update the current task's runtime statistics. Skip current tasks that
471  * are not in our scheduling class.
472  */
473 static inline void
474 __update_curr(struct cfs_rq *cfs_rq, struct sched_entity *curr,
475               unsigned long delta_exec)
476 {
477         unsigned long delta_exec_weighted;
478
479         schedstat_set(curr->exec_max, max((u64)delta_exec, curr->exec_max));
480
481         curr->sum_exec_runtime += delta_exec;
482         schedstat_add(cfs_rq, exec_clock, delta_exec);
483         delta_exec_weighted = calc_delta_fair(delta_exec, curr);
484         curr->vruntime += delta_exec_weighted;
485 }
486
487 static void update_curr(struct cfs_rq *cfs_rq)
488 {
489         struct sched_entity *curr = cfs_rq->curr;
490         u64 now = rq_of(cfs_rq)->clock;
491         unsigned long delta_exec;
492
493         if (unlikely(!curr))
494                 return;
495
496         /*
497          * Get the amount of time the current task was running
498          * since the last time we changed load (this cannot
499          * overflow on 32 bits):
500          */
501         delta_exec = (unsigned long)(now - curr->exec_start);
502
503         __update_curr(cfs_rq, curr, delta_exec);
504         curr->exec_start = now;
505
506         if (entity_is_task(curr)) {
507                 struct task_struct *curtask = task_of(curr);
508
509                 cpuacct_charge(curtask, delta_exec);
510         }
511 }
512
513 static inline void
514 update_stats_wait_start(struct cfs_rq *cfs_rq, struct sched_entity *se)
515 {
516         schedstat_set(se->wait_start, rq_of(cfs_rq)->clock);
517 }
518
519 /*
520  * Task is being enqueued - update stats:
521  */
522 static void update_stats_enqueue(struct cfs_rq *cfs_rq, struct sched_entity *se)
523 {
524         /*
525          * Are we enqueueing a waiting task? (for current tasks
526          * a dequeue/enqueue event is a NOP)
527          */
528         if (se != cfs_rq->curr)
529                 update_stats_wait_start(cfs_rq, se);
530 }
531
532 static void
533 update_stats_wait_end(struct cfs_rq *cfs_rq, struct sched_entity *se)
534 {
535         schedstat_set(se->wait_max, max(se->wait_max,
536                         rq_of(cfs_rq)->clock - se->wait_start));
537         schedstat_set(se->wait_count, se->wait_count + 1);
538         schedstat_set(se->wait_sum, se->wait_sum +
539                         rq_of(cfs_rq)->clock - se->wait_start);
540         schedstat_set(se->wait_start, 0);
541 }
542
543 static inline void
544 update_stats_dequeue(struct cfs_rq *cfs_rq, struct sched_entity *se)
545 {
546         /*
547          * Mark the end of the wait period if dequeueing a
548          * waiting task:
549          */
550         if (se != cfs_rq->curr)
551                 update_stats_wait_end(cfs_rq, se);
552 }
553
554 /*
555  * We are picking a new current task - update its stats:
556  */
557 static inline void
558 update_stats_curr_start(struct cfs_rq *cfs_rq, struct sched_entity *se)
559 {
560         /*
561          * We are starting a new run period:
562          */
563         se->exec_start = rq_of(cfs_rq)->clock;
564 }
565
566 /**************************************************
567  * Scheduling class queueing methods:
568  */
569
570 #if defined CONFIG_SMP && defined CONFIG_FAIR_GROUP_SCHED
571 static void
572 add_cfs_task_weight(struct cfs_rq *cfs_rq, unsigned long weight)
573 {
574         cfs_rq->task_weight += weight;
575 }
576 #else
577 static inline void
578 add_cfs_task_weight(struct cfs_rq *cfs_rq, unsigned long weight)
579 {
580 }
581 #endif
582
583 static void
584 account_entity_enqueue(struct cfs_rq *cfs_rq, struct sched_entity *se)
585 {
586         update_load_add(&cfs_rq->load, se->load.weight);
587         if (!parent_entity(se))
588                 inc_cpu_load(rq_of(cfs_rq), se->load.weight);
589         if (entity_is_task(se))
590                 add_cfs_task_weight(cfs_rq, se->load.weight);
591         cfs_rq->nr_running++;
592         se->on_rq = 1;
593         list_add(&se->group_node, &cfs_rq->tasks);
594 }
595
596 static void
597 account_entity_dequeue(struct cfs_rq *cfs_rq, struct sched_entity *se)
598 {
599         update_load_sub(&cfs_rq->load, se->load.weight);
600         if (!parent_entity(se))
601                 dec_cpu_load(rq_of(cfs_rq), se->load.weight);
602         if (entity_is_task(se))
603                 add_cfs_task_weight(cfs_rq, -se->load.weight);
604         cfs_rq->nr_running--;
605         se->on_rq = 0;
606         list_del_init(&se->group_node);
607 }
608
609 static void enqueue_sleeper(struct cfs_rq *cfs_rq, struct sched_entity *se)
610 {
611 #ifdef CONFIG_SCHEDSTATS
612         if (se->sleep_start) {
613                 u64 delta = rq_of(cfs_rq)->clock - se->sleep_start;
614                 struct task_struct *tsk = task_of(se);
615
616                 if ((s64)delta < 0)
617                         delta = 0;
618
619                 if (unlikely(delta > se->sleep_max))
620                         se->sleep_max = delta;
621
622                 se->sleep_start = 0;
623                 se->sum_sleep_runtime += delta;
624
625                 account_scheduler_latency(tsk, delta >> 10, 1);
626         }
627         if (se->block_start) {
628                 u64 delta = rq_of(cfs_rq)->clock - se->block_start;
629                 struct task_struct *tsk = task_of(se);
630
631                 if ((s64)delta < 0)
632                         delta = 0;
633
634                 if (unlikely(delta > se->block_max))
635                         se->block_max = delta;
636
637                 se->block_start = 0;
638                 se->sum_sleep_runtime += delta;
639
640                 /*
641                  * Blocking time is in units of nanosecs, so shift by 20 to
642                  * get a milliseconds-range estimation of the amount of
643                  * time that the task spent sleeping:
644                  */
645                 if (unlikely(prof_on == SLEEP_PROFILING)) {
646
647                         profile_hits(SLEEP_PROFILING, (void *)get_wchan(tsk),
648                                      delta >> 20);
649                 }
650                 account_scheduler_latency(tsk, delta >> 10, 0);
651         }
652 #endif
653 }
654
655 static void check_spread(struct cfs_rq *cfs_rq, struct sched_entity *se)
656 {
657 #ifdef CONFIG_SCHED_DEBUG
658         s64 d = se->vruntime - cfs_rq->min_vruntime;
659
660         if (d < 0)
661                 d = -d;
662
663         if (d > 3*sysctl_sched_latency)
664                 schedstat_inc(cfs_rq, nr_spread_over);
665 #endif
666 }
667
668 static void
669 place_entity(struct cfs_rq *cfs_rq, struct sched_entity *se, int initial)
670 {
671         u64 vruntime;
672
673         if (first_fair(cfs_rq)) {
674                 vruntime = min_vruntime(cfs_rq->min_vruntime,
675                                 __pick_next_entity(cfs_rq)->vruntime);
676         } else
677                 vruntime = cfs_rq->min_vruntime;
678
679         /*
680          * The 'current' period is already promised to the current tasks,
681          * however the extra weight of the new task will slow them down a
682          * little, place the new task so that it fits in the slot that
683          * stays open at the end.
684          */
685         if (initial && sched_feat(START_DEBIT))
686                 vruntime += sched_vslice_add(cfs_rq, se);
687
688         if (!initial) {
689                 /* sleeps upto a single latency don't count. */
690                 if (sched_feat(NEW_FAIR_SLEEPERS)) {
691                         unsigned long thresh = sysctl_sched_latency;
692
693                         /*
694                          * convert the sleeper threshold into virtual time
695                          */
696                         if (sched_feat(NORMALIZED_SLEEPER))
697                                 thresh = calc_delta_fair(thresh, se);
698
699                         vruntime -= thresh;
700                 }
701
702                 /* ensure we never gain time by being placed backwards. */
703                 vruntime = max_vruntime(se->vruntime, vruntime);
704         }
705
706         se->vruntime = vruntime;
707 }
708
709 static void
710 enqueue_entity(struct cfs_rq *cfs_rq, struct sched_entity *se, int wakeup)
711 {
712         /*
713          * Update run-time statistics of the 'current'.
714          */
715         update_curr(cfs_rq);
716         account_entity_enqueue(cfs_rq, se);
717
718         if (wakeup) {
719                 place_entity(cfs_rq, se, 0);
720                 enqueue_sleeper(cfs_rq, se);
721         }
722
723         update_stats_enqueue(cfs_rq, se);
724         check_spread(cfs_rq, se);
725         if (se != cfs_rq->curr)
726                 __enqueue_entity(cfs_rq, se);
727 }
728
729 static void
730 dequeue_entity(struct cfs_rq *cfs_rq, struct sched_entity *se, int sleep)
731 {
732         /*
733          * Update run-time statistics of the 'current'.
734          */
735         update_curr(cfs_rq);
736
737         update_stats_dequeue(cfs_rq, se);
738         if (sleep) {
739 #ifdef CONFIG_SCHEDSTATS
740                 if (entity_is_task(se)) {
741                         struct task_struct *tsk = task_of(se);
742
743                         if (tsk->state & TASK_INTERRUPTIBLE)
744                                 se->sleep_start = rq_of(cfs_rq)->clock;
745                         if (tsk->state & TASK_UNINTERRUPTIBLE)
746                                 se->block_start = rq_of(cfs_rq)->clock;
747                 }
748 #endif
749         }
750
751         if (se != cfs_rq->curr)
752                 __dequeue_entity(cfs_rq, se);
753         account_entity_dequeue(cfs_rq, se);
754 }
755
756 /*
757  * Preempt the current task with a newly woken task if needed:
758  */
759 static void
760 check_preempt_tick(struct cfs_rq *cfs_rq, struct sched_entity *curr)
761 {
762         unsigned long ideal_runtime, delta_exec;
763
764         ideal_runtime = sched_slice(cfs_rq, curr);
765         delta_exec = curr->sum_exec_runtime - curr->prev_sum_exec_runtime;
766         if (delta_exec > ideal_runtime)
767                 resched_task(rq_of(cfs_rq)->curr);
768 }
769
770 static void
771 set_next_entity(struct cfs_rq *cfs_rq, struct sched_entity *se)
772 {
773         /* 'current' is not kept within the tree. */
774         if (se->on_rq) {
775                 /*
776                  * Any task has to be enqueued before it get to execute on
777                  * a CPU. So account for the time it spent waiting on the
778                  * runqueue.
779                  */
780                 update_stats_wait_end(cfs_rq, se);
781                 __dequeue_entity(cfs_rq, se);
782         }
783
784         update_stats_curr_start(cfs_rq, se);
785         cfs_rq->curr = se;
786 #ifdef CONFIG_SCHEDSTATS
787         /*
788          * Track our maximum slice length, if the CPU's load is at
789          * least twice that of our own weight (i.e. dont track it
790          * when there are only lesser-weight tasks around):
791          */
792         if (rq_of(cfs_rq)->load.weight >= 2*se->load.weight) {
793                 se->slice_max = max(se->slice_max,
794                         se->sum_exec_runtime - se->prev_sum_exec_runtime);
795         }
796 #endif
797         se->prev_sum_exec_runtime = se->sum_exec_runtime;
798 }
799
800 static struct sched_entity *
801 pick_next(struct cfs_rq *cfs_rq, struct sched_entity *se)
802 {
803         struct rq *rq = rq_of(cfs_rq);
804         u64 pair_slice = rq->clock - cfs_rq->pair_start;
805
806         if (!cfs_rq->next || pair_slice > sched_slice(cfs_rq, cfs_rq->next)) {
807                 cfs_rq->pair_start = rq->clock;
808                 return se;
809         }
810
811         return cfs_rq->next;
812 }
813
814 static struct sched_entity *pick_next_entity(struct cfs_rq *cfs_rq)
815 {
816         struct sched_entity *se = NULL;
817
818         if (first_fair(cfs_rq)) {
819                 se = __pick_next_entity(cfs_rq);
820                 se = pick_next(cfs_rq, se);
821                 set_next_entity(cfs_rq, se);
822         }
823
824         return se;
825 }
826
827 static void put_prev_entity(struct cfs_rq *cfs_rq, struct sched_entity *prev)
828 {
829         /*
830          * If still on the runqueue then deactivate_task()
831          * was not called and update_curr() has to be done:
832          */
833         if (prev->on_rq)
834                 update_curr(cfs_rq);
835
836         check_spread(cfs_rq, prev);
837         if (prev->on_rq) {
838                 update_stats_wait_start(cfs_rq, prev);
839                 /* Put 'current' back into the tree. */
840                 __enqueue_entity(cfs_rq, prev);
841         }
842         cfs_rq->curr = NULL;
843 }
844
845 static void
846 entity_tick(struct cfs_rq *cfs_rq, struct sched_entity *curr, int queued)
847 {
848         /*
849          * Update run-time statistics of the 'current'.
850          */
851         update_curr(cfs_rq);
852
853 #ifdef CONFIG_SCHED_HRTICK
854         /*
855          * queued ticks are scheduled to match the slice, so don't bother
856          * validating it and just reschedule.
857          */
858         if (queued) {
859                 resched_task(rq_of(cfs_rq)->curr);
860                 return;
861         }
862         /*
863          * don't let the period tick interfere with the hrtick preemption
864          */
865         if (!sched_feat(DOUBLE_TICK) &&
866                         hrtimer_active(&rq_of(cfs_rq)->hrtick_timer))
867                 return;
868 #endif
869
870         if (cfs_rq->nr_running > 1 || !sched_feat(WAKEUP_PREEMPT))
871                 check_preempt_tick(cfs_rq, curr);
872 }
873
874 /**************************************************
875  * CFS operations on tasks:
876  */
877
878 #ifdef CONFIG_SCHED_HRTICK
879 static void hrtick_start_fair(struct rq *rq, struct task_struct *p)
880 {
881         struct sched_entity *se = &p->se;
882         struct cfs_rq *cfs_rq = cfs_rq_of(se);
883
884         WARN_ON(task_rq(p) != rq);
885
886         if (hrtick_enabled(rq) && cfs_rq->nr_running > 1) {
887                 u64 slice = sched_slice(cfs_rq, se);
888                 u64 ran = se->sum_exec_runtime - se->prev_sum_exec_runtime;
889                 s64 delta = slice - ran;
890
891                 if (delta < 0) {
892                         if (rq->curr == p)
893                                 resched_task(p);
894                         return;
895                 }
896
897                 /*
898                  * Don't schedule slices shorter than 10000ns, that just
899                  * doesn't make sense. Rely on vruntime for fairness.
900                  */
901                 if (rq->curr != p)
902                         delta = max(10000LL, delta);
903
904                 hrtick_start(rq, delta);
905         }
906 }
907 #else /* !CONFIG_SCHED_HRTICK */
908 static inline void
909 hrtick_start_fair(struct rq *rq, struct task_struct *p)
910 {
911 }
912 #endif
913
914 /*
915  * The enqueue_task method is called before nr_running is
916  * increased. Here we update the fair scheduling stats and
917  * then put the task into the rbtree:
918  */
919 static void enqueue_task_fair(struct rq *rq, struct task_struct *p, int wakeup)
920 {
921         struct cfs_rq *cfs_rq;
922         struct sched_entity *se = &p->se;
923
924         for_each_sched_entity(se) {
925                 if (se->on_rq)
926                         break;
927                 cfs_rq = cfs_rq_of(se);
928                 enqueue_entity(cfs_rq, se, wakeup);
929                 wakeup = 1;
930         }
931
932         hrtick_start_fair(rq, rq->curr);
933 }
934
935 /*
936  * The dequeue_task method is called before nr_running is
937  * decreased. We remove the task from the rbtree and
938  * update the fair scheduling stats:
939  */
940 static void dequeue_task_fair(struct rq *rq, struct task_struct *p, int sleep)
941 {
942         struct cfs_rq *cfs_rq;
943         struct sched_entity *se = &p->se;
944
945         for_each_sched_entity(se) {
946                 cfs_rq = cfs_rq_of(se);
947                 dequeue_entity(cfs_rq, se, sleep);
948                 /* Don't dequeue parent if it has other entities besides us */
949                 if (cfs_rq->load.weight)
950                         break;
951                 sleep = 1;
952         }
953
954         hrtick_start_fair(rq, rq->curr);
955 }
956
957 /*
958  * sched_yield() support is very simple - we dequeue and enqueue.
959  *
960  * If compat_yield is turned on then we requeue to the end of the tree.
961  */
962 static void yield_task_fair(struct rq *rq)
963 {
964         struct task_struct *curr = rq->curr;
965         struct cfs_rq *cfs_rq = task_cfs_rq(curr);
966         struct sched_entity *rightmost, *se = &curr->se;
967
968         /*
969          * Are we the only task in the tree?
970          */
971         if (unlikely(cfs_rq->nr_running == 1))
972                 return;
973
974         if (likely(!sysctl_sched_compat_yield) && curr->policy != SCHED_BATCH) {
975                 update_rq_clock(rq);
976                 /*
977                  * Update run-time statistics of the 'current'.
978                  */
979                 update_curr(cfs_rq);
980
981                 return;
982         }
983         /*
984          * Find the rightmost entry in the rbtree:
985          */
986         rightmost = __pick_last_entity(cfs_rq);
987         /*
988          * Already in the rightmost position?
989          */
990         if (unlikely(!rightmost || rightmost->vruntime < se->vruntime))
991                 return;
992
993         /*
994          * Minimally necessary key value to be last in the tree:
995          * Upon rescheduling, sched_class::put_prev_task() will place
996          * 'current' within the tree based on its new key value.
997          */
998         se->vruntime = rightmost->vruntime + 1;
999 }
1000
1001 /*
1002  * wake_idle() will wake a task on an idle cpu if task->cpu is
1003  * not idle and an idle cpu is available.  The span of cpus to
1004  * search starts with cpus closest then further out as needed,
1005  * so we always favor a closer, idle cpu.
1006  * Domains may include CPUs that are not usable for migration,
1007  * hence we need to mask them out (cpu_active_map)
1008  *
1009  * Returns the CPU we should wake onto.
1010  */
1011 #if defined(ARCH_HAS_SCHED_WAKE_IDLE)
1012 static int wake_idle(int cpu, struct task_struct *p)
1013 {
1014         cpumask_t tmp;
1015         struct sched_domain *sd;
1016         int i;
1017
1018         /*
1019          * If it is idle, then it is the best cpu to run this task.
1020          *
1021          * This cpu is also the best, if it has more than one task already.
1022          * Siblings must be also busy(in most cases) as they didn't already
1023          * pickup the extra load from this cpu and hence we need not check
1024          * sibling runqueue info. This will avoid the checks and cache miss
1025          * penalities associated with that.
1026          */
1027         if (idle_cpu(cpu) || cpu_rq(cpu)->cfs.nr_running > 1)
1028                 return cpu;
1029
1030         for_each_domain(cpu, sd) {
1031                 if ((sd->flags & SD_WAKE_IDLE)
1032                     || ((sd->flags & SD_WAKE_IDLE_FAR)
1033                         && !task_hot(p, task_rq(p)->clock, sd))) {
1034                         cpus_and(tmp, sd->span, p->cpus_allowed);
1035                         cpus_and(tmp, tmp, cpu_active_map);
1036                         for_each_cpu_mask_nr(i, tmp) {
1037                                 if (idle_cpu(i)) {
1038                                         if (i != task_cpu(p)) {
1039                                                 schedstat_inc(p,
1040                                                        se.nr_wakeups_idle);
1041                                         }
1042                                         return i;
1043                                 }
1044                         }
1045                 } else {
1046                         break;
1047                 }
1048         }
1049         return cpu;
1050 }
1051 #else /* !ARCH_HAS_SCHED_WAKE_IDLE*/
1052 static inline int wake_idle(int cpu, struct task_struct *p)
1053 {
1054         return cpu;
1055 }
1056 #endif
1057
1058 #ifdef CONFIG_SMP
1059
1060 static const struct sched_class fair_sched_class;
1061
1062 #ifdef CONFIG_FAIR_GROUP_SCHED
1063 /*
1064  * effective_load() calculates the load change as seen from the root_task_group
1065  *
1066  * Adding load to a group doesn't make a group heavier, but can cause movement
1067  * of group shares between cpus. Assuming the shares were perfectly aligned one
1068  * can calculate the shift in shares.
1069  *
1070  * The problem is that perfectly aligning the shares is rather expensive, hence
1071  * we try to avoid doing that too often - see update_shares(), which ratelimits
1072  * this change.
1073  *
1074  * We compensate this by not only taking the current delta into account, but
1075  * also considering the delta between when the shares were last adjusted and
1076  * now.
1077  *
1078  * We still saw a performance dip, some tracing learned us that between
1079  * cgroup:/ and cgroup:/foo balancing the number of affine wakeups increased
1080  * significantly. Therefore try to bias the error in direction of failing
1081  * the affine wakeup.
1082  *
1083  */
1084 static long effective_load(struct task_group *tg, int cpu,
1085                 long wl, long wg)
1086 {
1087         struct sched_entity *se = tg->se[cpu];
1088         long more_w;
1089
1090         if (!tg->parent)
1091                 return wl;
1092
1093         /*
1094          * By not taking the decrease of shares on the other cpu into
1095          * account our error leans towards reducing the affine wakeups.
1096          */
1097         if (!wl && sched_feat(ASYM_EFF_LOAD))
1098                 return wl;
1099
1100         /*
1101          * Instead of using this increment, also add the difference
1102          * between when the shares were last updated and now.
1103          */
1104         more_w = se->my_q->load.weight - se->my_q->rq_weight;
1105         wl += more_w;
1106         wg += more_w;
1107
1108         for_each_sched_entity(se) {
1109 #define D(n) (likely(n) ? (n) : 1)
1110
1111                 long S, rw, s, a, b;
1112
1113                 S = se->my_q->tg->shares;
1114                 s = se->my_q->shares;
1115                 rw = se->my_q->rq_weight;
1116
1117                 a = S*(rw + wl);
1118                 b = S*rw + s*wg;
1119
1120                 wl = s*(a-b)/D(b);
1121                 /*
1122                  * Assume the group is already running and will
1123                  * thus already be accounted for in the weight.
1124                  *
1125                  * That is, moving shares between CPUs, does not
1126                  * alter the group weight.
1127                  */
1128                 wg = 0;
1129 #undef D
1130         }
1131
1132         return wl;
1133 }
1134
1135 #else
1136
1137 static inline unsigned long effective_load(struct task_group *tg, int cpu,
1138                 unsigned long wl, unsigned long wg)
1139 {
1140         return wl;
1141 }
1142
1143 #endif
1144
1145 static int
1146 wake_affine(struct rq *rq, struct sched_domain *this_sd, struct rq *this_rq,
1147             struct task_struct *p, int prev_cpu, int this_cpu, int sync,
1148             int idx, unsigned long load, unsigned long this_load,
1149             unsigned int imbalance)
1150 {
1151         struct task_struct *curr = this_rq->curr;
1152         struct task_group *tg;
1153         unsigned long tl = this_load;
1154         unsigned long tl_per_task;
1155         unsigned long weight;
1156         int balanced;
1157
1158         if (!(this_sd->flags & SD_WAKE_AFFINE) || !sched_feat(AFFINE_WAKEUPS))
1159                 return 0;
1160
1161         /*
1162          * If sync wakeup then subtract the (maximum possible)
1163          * effect of the currently running task from the load
1164          * of the current CPU:
1165          */
1166         if (sync) {
1167                 tg = task_group(current);
1168                 weight = current->se.load.weight;
1169
1170                 tl += effective_load(tg, this_cpu, -weight, -weight);
1171                 load += effective_load(tg, prev_cpu, 0, -weight);
1172         }
1173
1174         tg = task_group(p);
1175         weight = p->se.load.weight;
1176
1177         balanced = 100*(tl + effective_load(tg, this_cpu, weight, weight)) <=
1178                 imbalance*(load + effective_load(tg, prev_cpu, 0, weight));
1179
1180         /*
1181          * If the currently running task will sleep within
1182          * a reasonable amount of time then attract this newly
1183          * woken task:
1184          */
1185         if (sync && balanced) {
1186                 if (curr->se.avg_overlap < sysctl_sched_migration_cost &&
1187                     p->se.avg_overlap < sysctl_sched_migration_cost)
1188                         return 1;
1189         }
1190
1191         schedstat_inc(p, se.nr_wakeups_affine_attempts);
1192         tl_per_task = cpu_avg_load_per_task(this_cpu);
1193
1194         if ((tl <= load && tl + target_load(prev_cpu, idx) <= tl_per_task) ||
1195                         balanced) {
1196                 /*
1197                  * This domain has SD_WAKE_AFFINE and
1198                  * p is cache cold in this domain, and
1199                  * there is no bad imbalance.
1200                  */
1201                 schedstat_inc(this_sd, ttwu_move_affine);
1202                 schedstat_inc(p, se.nr_wakeups_affine);
1203
1204                 return 1;
1205         }
1206         return 0;
1207 }
1208
1209 static int select_task_rq_fair(struct task_struct *p, int sync)
1210 {
1211         struct sched_domain *sd, *this_sd = NULL;
1212         int prev_cpu, this_cpu, new_cpu;
1213         unsigned long load, this_load;
1214         struct rq *rq, *this_rq;
1215         unsigned int imbalance;
1216         int idx;
1217
1218         prev_cpu        = task_cpu(p);
1219         rq              = task_rq(p);
1220         this_cpu        = smp_processor_id();
1221         this_rq         = cpu_rq(this_cpu);
1222         new_cpu         = prev_cpu;
1223
1224         /*
1225          * 'this_sd' is the first domain that both
1226          * this_cpu and prev_cpu are present in:
1227          */
1228         for_each_domain(this_cpu, sd) {
1229                 if (cpu_isset(prev_cpu, sd->span)) {
1230                         this_sd = sd;
1231                         break;
1232                 }
1233         }
1234
1235         if (unlikely(!cpu_isset(this_cpu, p->cpus_allowed)))
1236                 goto out;
1237
1238         /*
1239          * Check for affine wakeup and passive balancing possibilities.
1240          */
1241         if (!this_sd)
1242                 goto out;
1243
1244         idx = this_sd->wake_idx;
1245
1246         imbalance = 100 + (this_sd->imbalance_pct - 100) / 2;
1247
1248         load = source_load(prev_cpu, idx);
1249         this_load = target_load(this_cpu, idx);
1250
1251         if (wake_affine(rq, this_sd, this_rq, p, prev_cpu, this_cpu, sync, idx,
1252                                      load, this_load, imbalance))
1253                 return this_cpu;
1254
1255         if (prev_cpu == this_cpu)
1256                 goto out;
1257
1258         /*
1259          * Start passive balancing when half the imbalance_pct
1260          * limit is reached.
1261          */
1262         if (this_sd->flags & SD_WAKE_BALANCE) {
1263                 if (imbalance*this_load <= 100*load) {
1264                         schedstat_inc(this_sd, ttwu_move_balance);
1265                         schedstat_inc(p, se.nr_wakeups_passive);
1266                         return this_cpu;
1267                 }
1268         }
1269
1270 out:
1271         return wake_idle(new_cpu, p);
1272 }
1273 #endif /* CONFIG_SMP */
1274
1275 static unsigned long wakeup_gran(struct sched_entity *se)
1276 {
1277         unsigned long gran = sysctl_sched_wakeup_granularity;
1278
1279         /*
1280          * More easily preempt - nice tasks, while not making it harder for
1281          * + nice tasks.
1282          */
1283         if (sched_feat(ASYM_GRAN))
1284                 gran = calc_delta_asym(sysctl_sched_wakeup_granularity, se);
1285         else
1286                 gran = calc_delta_fair(sysctl_sched_wakeup_granularity, se);
1287
1288         return gran;
1289 }
1290
1291 /*
1292  * Should 'se' preempt 'curr'.
1293  *
1294  *             |s1
1295  *        |s2
1296  *   |s3
1297  *         g
1298  *      |<--->|c
1299  *
1300  *  w(c, s1) = -1
1301  *  w(c, s2) =  0
1302  *  w(c, s3) =  1
1303  *
1304  */
1305 static int
1306 wakeup_preempt_entity(struct sched_entity *curr, struct sched_entity *se)
1307 {
1308         s64 gran, vdiff = curr->vruntime - se->vruntime;
1309
1310         if (vdiff < 0)
1311                 return -1;
1312
1313         gran = wakeup_gran(curr);
1314         if (vdiff > gran)
1315                 return 1;
1316
1317         return 0;
1318 }
1319
1320 /* return depth at which a sched entity is present in the hierarchy */
1321 static inline int depth_se(struct sched_entity *se)
1322 {
1323         int depth = 0;
1324
1325         for_each_sched_entity(se)
1326                 depth++;
1327
1328         return depth;
1329 }
1330
1331 /*
1332  * Preempt the current task with a newly woken task if needed:
1333  */
1334 static void check_preempt_wakeup(struct rq *rq, struct task_struct *p)
1335 {
1336         struct task_struct *curr = rq->curr;
1337         struct cfs_rq *cfs_rq = task_cfs_rq(curr);
1338         struct sched_entity *se = &curr->se, *pse = &p->se;
1339         int se_depth, pse_depth;
1340
1341         if (unlikely(rt_prio(p->prio))) {
1342                 update_rq_clock(rq);
1343                 update_curr(cfs_rq);
1344                 resched_task(curr);
1345                 return;
1346         }
1347
1348         if (unlikely(se == pse))
1349                 return;
1350
1351         cfs_rq_of(pse)->next = pse;
1352
1353         /*
1354          * Batch tasks do not preempt (their preemption is driven by
1355          * the tick):
1356          */
1357         if (unlikely(p->policy == SCHED_BATCH))
1358                 return;
1359
1360         if (!sched_feat(WAKEUP_PREEMPT))
1361                 return;
1362
1363         /*
1364          * preemption test can be made between sibling entities who are in the
1365          * same cfs_rq i.e who have a common parent. Walk up the hierarchy of
1366          * both tasks until we find their ancestors who are siblings of common
1367          * parent.
1368          */
1369
1370         /* First walk up until both entities are at same depth */
1371         se_depth = depth_se(se);
1372         pse_depth = depth_se(pse);
1373
1374         while (se_depth > pse_depth) {
1375                 se_depth--;
1376                 se = parent_entity(se);
1377         }
1378
1379         while (pse_depth > se_depth) {
1380                 pse_depth--;
1381                 pse = parent_entity(pse);
1382         }
1383
1384         while (!is_same_group(se, pse)) {
1385                 se = parent_entity(se);
1386                 pse = parent_entity(pse);
1387         }
1388
1389         if (wakeup_preempt_entity(se, pse) == 1)
1390                 resched_task(curr);
1391 }
1392
1393 static struct task_struct *pick_next_task_fair(struct rq *rq)
1394 {
1395         struct task_struct *p;
1396         struct cfs_rq *cfs_rq = &rq->cfs;
1397         struct sched_entity *se;
1398
1399         if (unlikely(!cfs_rq->nr_running))
1400                 return NULL;
1401
1402         do {
1403                 se = pick_next_entity(cfs_rq);
1404                 cfs_rq = group_cfs_rq(se);
1405         } while (cfs_rq);
1406
1407         p = task_of(se);
1408         hrtick_start_fair(rq, p);
1409
1410         return p;
1411 }
1412
1413 /*
1414  * Account for a descheduled task:
1415  */
1416 static void put_prev_task_fair(struct rq *rq, struct task_struct *prev)
1417 {
1418         struct sched_entity *se = &prev->se;
1419         struct cfs_rq *cfs_rq;
1420
1421         for_each_sched_entity(se) {
1422                 cfs_rq = cfs_rq_of(se);
1423                 put_prev_entity(cfs_rq, se);
1424         }
1425 }
1426
1427 #ifdef CONFIG_SMP
1428 /**************************************************
1429  * Fair scheduling class load-balancing methods:
1430  */
1431
1432 /*
1433  * Load-balancing iterator. Note: while the runqueue stays locked
1434  * during the whole iteration, the current task might be
1435  * dequeued so the iterator has to be dequeue-safe. Here we
1436  * achieve that by always pre-iterating before returning
1437  * the current task:
1438  */
1439 static struct task_struct *
1440 __load_balance_iterator(struct cfs_rq *cfs_rq, struct list_head *next)
1441 {
1442         struct task_struct *p = NULL;
1443         struct sched_entity *se;
1444
1445         while (next != &cfs_rq->tasks) {
1446                 se = list_entry(next, struct sched_entity, group_node);
1447                 next = next->next;
1448
1449                 /* Skip over entities that are not tasks */
1450                 if (entity_is_task(se)) {
1451                         p = task_of(se);
1452                         break;
1453                 }
1454         }
1455
1456         cfs_rq->balance_iterator = next;
1457         return p;
1458 }
1459
1460 static struct task_struct *load_balance_start_fair(void *arg)
1461 {
1462         struct cfs_rq *cfs_rq = arg;
1463
1464         return __load_balance_iterator(cfs_rq, cfs_rq->tasks.next);
1465 }
1466
1467 static struct task_struct *load_balance_next_fair(void *arg)
1468 {
1469         struct cfs_rq *cfs_rq = arg;
1470
1471         return __load_balance_iterator(cfs_rq, cfs_rq->balance_iterator);
1472 }
1473
1474 static unsigned long
1475 __load_balance_fair(struct rq *this_rq, int this_cpu, struct rq *busiest,
1476                 unsigned long max_load_move, struct sched_domain *sd,
1477                 enum cpu_idle_type idle, int *all_pinned, int *this_best_prio,
1478                 struct cfs_rq *cfs_rq)
1479 {
1480         struct rq_iterator cfs_rq_iterator;
1481
1482         cfs_rq_iterator.start = load_balance_start_fair;
1483         cfs_rq_iterator.next = load_balance_next_fair;
1484         cfs_rq_iterator.arg = cfs_rq;
1485
1486         return balance_tasks(this_rq, this_cpu, busiest,
1487                         max_load_move, sd, idle, all_pinned,
1488                         this_best_prio, &cfs_rq_iterator);
1489 }
1490
1491 #ifdef CONFIG_FAIR_GROUP_SCHED
1492 static unsigned long
1493 load_balance_fair(struct rq *this_rq, int this_cpu, struct rq *busiest,
1494                   unsigned long max_load_move,
1495                   struct sched_domain *sd, enum cpu_idle_type idle,
1496                   int *all_pinned, int *this_best_prio)
1497 {
1498         long rem_load_move = max_load_move;
1499         int busiest_cpu = cpu_of(busiest);
1500         struct task_group *tg;
1501
1502         rcu_read_lock();
1503         update_h_load(busiest_cpu);
1504
1505         list_for_each_entry(tg, &task_groups, list) {
1506                 struct cfs_rq *busiest_cfs_rq = tg->cfs_rq[busiest_cpu];
1507                 unsigned long busiest_h_load = busiest_cfs_rq->h_load;
1508                 unsigned long busiest_weight = busiest_cfs_rq->load.weight;
1509                 u64 rem_load, moved_load;
1510
1511                 /*
1512                  * empty group
1513                  */
1514                 if (!busiest_cfs_rq->task_weight)
1515                         continue;
1516
1517                 rem_load = (u64)rem_load_move * busiest_weight;
1518                 rem_load = div_u64(rem_load, busiest_h_load + 1);
1519
1520                 moved_load = __load_balance_fair(this_rq, this_cpu, busiest,
1521                                 rem_load, sd, idle, all_pinned, this_best_prio,
1522                                 tg->cfs_rq[busiest_cpu]);
1523
1524                 if (!moved_load)
1525                         continue;
1526
1527                 moved_load *= busiest_h_load;
1528                 moved_load = div_u64(moved_load, busiest_weight + 1);
1529
1530                 rem_load_move -= moved_load;
1531                 if (rem_load_move < 0)
1532                         break;
1533         }
1534         rcu_read_unlock();
1535
1536         return max_load_move - rem_load_move;
1537 }
1538 #else
1539 static unsigned long
1540 load_balance_fair(struct rq *this_rq, int this_cpu, struct rq *busiest,
1541                   unsigned long max_load_move,
1542                   struct sched_domain *sd, enum cpu_idle_type idle,
1543                   int *all_pinned, int *this_best_prio)
1544 {
1545         return __load_balance_fair(this_rq, this_cpu, busiest,
1546                         max_load_move, sd, idle, all_pinned,
1547                         this_best_prio, &busiest->cfs);
1548 }
1549 #endif
1550
1551 static int
1552 move_one_task_fair(struct rq *this_rq, int this_cpu, struct rq *busiest,
1553                    struct sched_domain *sd, enum cpu_idle_type idle)
1554 {
1555         struct cfs_rq *busy_cfs_rq;
1556         struct rq_iterator cfs_rq_iterator;
1557
1558         cfs_rq_iterator.start = load_balance_start_fair;
1559         cfs_rq_iterator.next = load_balance_next_fair;
1560
1561         for_each_leaf_cfs_rq(busiest, busy_cfs_rq) {
1562                 /*
1563                  * pass busy_cfs_rq argument into
1564                  * load_balance_[start|next]_fair iterators
1565                  */
1566                 cfs_rq_iterator.arg = busy_cfs_rq;
1567                 if (iter_move_one_task(this_rq, this_cpu, busiest, sd, idle,
1568                                        &cfs_rq_iterator))
1569                     return 1;
1570         }
1571
1572         return 0;
1573 }
1574 #endif /* CONFIG_SMP */
1575
1576 /*
1577  * scheduler tick hitting a task of our scheduling class:
1578  */
1579 static void task_tick_fair(struct rq *rq, struct task_struct *curr, int queued)
1580 {
1581         struct cfs_rq *cfs_rq;
1582         struct sched_entity *se = &curr->se;
1583
1584         for_each_sched_entity(se) {
1585                 cfs_rq = cfs_rq_of(se);
1586                 entity_tick(cfs_rq, se, queued);
1587         }
1588 }
1589
1590 #define swap(a, b) do { typeof(a) tmp = (a); (a) = (b); (b) = tmp; } while (0)
1591
1592 /*
1593  * Share the fairness runtime between parent and child, thus the
1594  * total amount of pressure for CPU stays equal - new tasks
1595  * get a chance to run but frequent forkers are not allowed to
1596  * monopolize the CPU. Note: the parent runqueue is locked,
1597  * the child is not running yet.
1598  */
1599 static void task_new_fair(struct rq *rq, struct task_struct *p)
1600 {
1601         struct cfs_rq *cfs_rq = task_cfs_rq(p);
1602         struct sched_entity *se = &p->se, *curr = cfs_rq->curr;
1603         int this_cpu = smp_processor_id();
1604
1605         sched_info_queued(p);
1606
1607         update_curr(cfs_rq);
1608         place_entity(cfs_rq, se, 1);
1609
1610         /* 'curr' will be NULL if the child belongs to a different group */
1611         if (sysctl_sched_child_runs_first && this_cpu == task_cpu(p) &&
1612                         curr && curr->vruntime < se->vruntime) {
1613                 /*
1614                  * Upon rescheduling, sched_class::put_prev_task() will place
1615                  * 'current' within the tree based on its new key value.
1616                  */
1617                 swap(curr->vruntime, se->vruntime);
1618         }
1619
1620         enqueue_task_fair(rq, p, 0);
1621         resched_task(rq->curr);
1622 }
1623
1624 /*
1625  * Priority of the task has changed. Check to see if we preempt
1626  * the current task.
1627  */
1628 static void prio_changed_fair(struct rq *rq, struct task_struct *p,
1629                               int oldprio, int running)
1630 {
1631         /*
1632          * Reschedule if we are currently running on this runqueue and
1633          * our priority decreased, or if we are not currently running on
1634          * this runqueue and our priority is higher than the current's
1635          */
1636         if (running) {
1637                 if (p->prio > oldprio)
1638                         resched_task(rq->curr);
1639         } else
1640                 check_preempt_curr(rq, p);
1641 }
1642
1643 /*
1644  * We switched to the sched_fair class.
1645  */
1646 static void switched_to_fair(struct rq *rq, struct task_struct *p,
1647                              int running)
1648 {
1649         /*
1650          * We were most likely switched from sched_rt, so
1651          * kick off the schedule if running, otherwise just see
1652          * if we can still preempt the current task.
1653          */
1654         if (running)
1655                 resched_task(rq->curr);
1656         else
1657                 check_preempt_curr(rq, p);
1658 }
1659
1660 /* Account for a task changing its policy or group.
1661  *
1662  * This routine is mostly called to set cfs_rq->curr field when a task
1663  * migrates between groups/classes.
1664  */
1665 static void set_curr_task_fair(struct rq *rq)
1666 {
1667         struct sched_entity *se = &rq->curr->se;
1668
1669         for_each_sched_entity(se)
1670                 set_next_entity(cfs_rq_of(se), se);
1671 }
1672
1673 #ifdef CONFIG_FAIR_GROUP_SCHED
1674 static void moved_group_fair(struct task_struct *p)
1675 {
1676         struct cfs_rq *cfs_rq = task_cfs_rq(p);
1677
1678         update_curr(cfs_rq);
1679         place_entity(cfs_rq, &p->se, 1);
1680 }
1681 #endif
1682
1683 /*
1684  * All the scheduling class methods:
1685  */
1686 static const struct sched_class fair_sched_class = {
1687         .next                   = &idle_sched_class,
1688         .enqueue_task           = enqueue_task_fair,
1689         .dequeue_task           = dequeue_task_fair,
1690         .yield_task             = yield_task_fair,
1691 #ifdef CONFIG_SMP
1692         .select_task_rq         = select_task_rq_fair,
1693 #endif /* CONFIG_SMP */
1694
1695         .check_preempt_curr     = check_preempt_wakeup,
1696
1697         .pick_next_task         = pick_next_task_fair,
1698         .put_prev_task          = put_prev_task_fair,
1699
1700 #ifdef CONFIG_SMP
1701         .load_balance           = load_balance_fair,
1702         .move_one_task          = move_one_task_fair,
1703 #endif
1704
1705         .set_curr_task          = set_curr_task_fair,
1706         .task_tick              = task_tick_fair,
1707         .task_new               = task_new_fair,
1708
1709         .prio_changed           = prio_changed_fair,
1710         .switched_to            = switched_to_fair,
1711
1712 #ifdef CONFIG_FAIR_GROUP_SCHED
1713         .moved_group            = moved_group_fair,
1714 #endif
1715 };
1716
1717 #ifdef CONFIG_SCHED_DEBUG
1718 static void print_cfs_stats(struct seq_file *m, int cpu)
1719 {
1720         struct cfs_rq *cfs_rq;
1721
1722         rcu_read_lock();
1723         for_each_leaf_cfs_rq(cpu_rq(cpu), cfs_rq)
1724                 print_cfs_rq(m, cpu, cfs_rq);
1725         rcu_read_unlock();
1726 }
1727 #endif