xref: /openbmc/linux/kernel/sched/sched.h (revision a06c488d)
1 
2 #include <linux/sched.h>
3 #include <linux/sched/sysctl.h>
4 #include <linux/sched/rt.h>
5 #include <linux/sched/deadline.h>
6 #include <linux/mutex.h>
7 #include <linux/spinlock.h>
8 #include <linux/stop_machine.h>
9 #include <linux/irq_work.h>
10 #include <linux/tick.h>
11 #include <linux/slab.h>
12 
13 #include "cpupri.h"
14 #include "cpudeadline.h"
15 #include "cpuacct.h"
16 
17 struct rq;
18 struct cpuidle_state;
19 
20 /* task_struct::on_rq states: */
21 #define TASK_ON_RQ_QUEUED	1
22 #define TASK_ON_RQ_MIGRATING	2
23 
24 extern __read_mostly int scheduler_running;
25 
26 extern unsigned long calc_load_update;
27 extern atomic_long_t calc_load_tasks;
28 
29 extern void calc_global_load_tick(struct rq *this_rq);
30 extern long calc_load_fold_active(struct rq *this_rq);
31 
32 #ifdef CONFIG_SMP
33 extern void update_cpu_load_active(struct rq *this_rq);
34 #else
35 static inline void update_cpu_load_active(struct rq *this_rq) { }
36 #endif
37 
38 /*
39  * Helpers for converting nanosecond timing to jiffy resolution
40  */
41 #define NS_TO_JIFFIES(TIME)	((unsigned long)(TIME) / (NSEC_PER_SEC / HZ))
42 
43 /*
44  * Increase resolution of nice-level calculations for 64-bit architectures.
45  * The extra resolution improves shares distribution and load balancing of
46  * low-weight task groups (eg. nice +19 on an autogroup), deeper taskgroup
47  * hierarchies, especially on larger systems. This is not a user-visible change
48  * and does not change the user-interface for setting shares/weights.
49  *
50  * We increase resolution only if we have enough bits to allow this increased
51  * resolution (i.e. BITS_PER_LONG > 32). The costs for increasing resolution
52  * when BITS_PER_LONG <= 32 are pretty high and the returns do not justify the
53  * increased costs.
54  */
55 #if 0 /* BITS_PER_LONG > 32 -- currently broken: it increases power usage under light load  */
56 # define SCHED_LOAD_RESOLUTION	10
57 # define scale_load(w)		((w) << SCHED_LOAD_RESOLUTION)
58 # define scale_load_down(w)	((w) >> SCHED_LOAD_RESOLUTION)
59 #else
60 # define SCHED_LOAD_RESOLUTION	0
61 # define scale_load(w)		(w)
62 # define scale_load_down(w)	(w)
63 #endif
64 
65 #define SCHED_LOAD_SHIFT	(10 + SCHED_LOAD_RESOLUTION)
66 #define SCHED_LOAD_SCALE	(1L << SCHED_LOAD_SHIFT)
67 
68 #define NICE_0_LOAD		SCHED_LOAD_SCALE
69 #define NICE_0_SHIFT		SCHED_LOAD_SHIFT
70 
71 /*
72  * Single value that decides SCHED_DEADLINE internal math precision.
73  * 10 -> just above 1us
74  * 9  -> just above 0.5us
75  */
76 #define DL_SCALE (10)
77 
78 /*
79  * These are the 'tuning knobs' of the scheduler:
80  */
81 
82 /*
83  * single value that denotes runtime == period, ie unlimited time.
84  */
85 #define RUNTIME_INF	((u64)~0ULL)
86 
87 static inline int idle_policy(int policy)
88 {
89 	return policy == SCHED_IDLE;
90 }
91 static inline int fair_policy(int policy)
92 {
93 	return policy == SCHED_NORMAL || policy == SCHED_BATCH;
94 }
95 
96 static inline int rt_policy(int policy)
97 {
98 	return policy == SCHED_FIFO || policy == SCHED_RR;
99 }
100 
101 static inline int dl_policy(int policy)
102 {
103 	return policy == SCHED_DEADLINE;
104 }
105 static inline bool valid_policy(int policy)
106 {
107 	return idle_policy(policy) || fair_policy(policy) ||
108 		rt_policy(policy) || dl_policy(policy);
109 }
110 
111 static inline int task_has_rt_policy(struct task_struct *p)
112 {
113 	return rt_policy(p->policy);
114 }
115 
116 static inline int task_has_dl_policy(struct task_struct *p)
117 {
118 	return dl_policy(p->policy);
119 }
120 
121 /*
122  * Tells if entity @a should preempt entity @b.
123  */
124 static inline bool
125 dl_entity_preempt(struct sched_dl_entity *a, struct sched_dl_entity *b)
126 {
127 	return dl_time_before(a->deadline, b->deadline);
128 }
129 
130 /*
131  * This is the priority-queue data structure of the RT scheduling class:
132  */
133 struct rt_prio_array {
134 	DECLARE_BITMAP(bitmap, MAX_RT_PRIO+1); /* include 1 bit for delimiter */
135 	struct list_head queue[MAX_RT_PRIO];
136 };
137 
138 struct rt_bandwidth {
139 	/* nests inside the rq lock: */
140 	raw_spinlock_t		rt_runtime_lock;
141 	ktime_t			rt_period;
142 	u64			rt_runtime;
143 	struct hrtimer		rt_period_timer;
144 	unsigned int		rt_period_active;
145 };
146 
147 void __dl_clear_params(struct task_struct *p);
148 
149 /*
150  * To keep the bandwidth of -deadline tasks and groups under control
151  * we need some place where:
152  *  - store the maximum -deadline bandwidth of the system (the group);
153  *  - cache the fraction of that bandwidth that is currently allocated.
154  *
155  * This is all done in the data structure below. It is similar to the
156  * one used for RT-throttling (rt_bandwidth), with the main difference
157  * that, since here we are only interested in admission control, we
158  * do not decrease any runtime while the group "executes", neither we
159  * need a timer to replenish it.
160  *
161  * With respect to SMP, the bandwidth is given on a per-CPU basis,
162  * meaning that:
163  *  - dl_bw (< 100%) is the bandwidth of the system (group) on each CPU;
164  *  - dl_total_bw array contains, in the i-eth element, the currently
165  *    allocated bandwidth on the i-eth CPU.
166  * Moreover, groups consume bandwidth on each CPU, while tasks only
167  * consume bandwidth on the CPU they're running on.
168  * Finally, dl_total_bw_cpu is used to cache the index of dl_total_bw
169  * that will be shown the next time the proc or cgroup controls will
170  * be red. It on its turn can be changed by writing on its own
171  * control.
172  */
173 struct dl_bandwidth {
174 	raw_spinlock_t dl_runtime_lock;
175 	u64 dl_runtime;
176 	u64 dl_period;
177 };
178 
179 static inline int dl_bandwidth_enabled(void)
180 {
181 	return sysctl_sched_rt_runtime >= 0;
182 }
183 
184 extern struct dl_bw *dl_bw_of(int i);
185 
186 struct dl_bw {
187 	raw_spinlock_t lock;
188 	u64 bw, total_bw;
189 };
190 
191 static inline
192 void __dl_clear(struct dl_bw *dl_b, u64 tsk_bw)
193 {
194 	dl_b->total_bw -= tsk_bw;
195 }
196 
197 static inline
198 void __dl_add(struct dl_bw *dl_b, u64 tsk_bw)
199 {
200 	dl_b->total_bw += tsk_bw;
201 }
202 
203 static inline
204 bool __dl_overflow(struct dl_bw *dl_b, int cpus, u64 old_bw, u64 new_bw)
205 {
206 	return dl_b->bw != -1 &&
207 	       dl_b->bw * cpus < dl_b->total_bw - old_bw + new_bw;
208 }
209 
210 extern struct mutex sched_domains_mutex;
211 
212 #ifdef CONFIG_CGROUP_SCHED
213 
214 #include <linux/cgroup.h>
215 
216 struct cfs_rq;
217 struct rt_rq;
218 
219 extern struct list_head task_groups;
220 
221 struct cfs_bandwidth {
222 #ifdef CONFIG_CFS_BANDWIDTH
223 	raw_spinlock_t lock;
224 	ktime_t period;
225 	u64 quota, runtime;
226 	s64 hierarchical_quota;
227 	u64 runtime_expires;
228 
229 	int idle, period_active;
230 	struct hrtimer period_timer, slack_timer;
231 	struct list_head throttled_cfs_rq;
232 
233 	/* statistics */
234 	int nr_periods, nr_throttled;
235 	u64 throttled_time;
236 #endif
237 };
238 
239 /* task group related information */
240 struct task_group {
241 	struct cgroup_subsys_state css;
242 
243 #ifdef CONFIG_FAIR_GROUP_SCHED
244 	/* schedulable entities of this group on each cpu */
245 	struct sched_entity **se;
246 	/* runqueue "owned" by this group on each cpu */
247 	struct cfs_rq **cfs_rq;
248 	unsigned long shares;
249 
250 #ifdef	CONFIG_SMP
251 	/*
252 	 * load_avg can be heavily contended at clock tick time, so put
253 	 * it in its own cacheline separated from the fields above which
254 	 * will also be accessed at each tick.
255 	 */
256 	atomic_long_t load_avg ____cacheline_aligned;
257 #endif
258 #endif
259 
260 #ifdef CONFIG_RT_GROUP_SCHED
261 	struct sched_rt_entity **rt_se;
262 	struct rt_rq **rt_rq;
263 
264 	struct rt_bandwidth rt_bandwidth;
265 #endif
266 
267 	struct rcu_head rcu;
268 	struct list_head list;
269 
270 	struct task_group *parent;
271 	struct list_head siblings;
272 	struct list_head children;
273 
274 #ifdef CONFIG_SCHED_AUTOGROUP
275 	struct autogroup *autogroup;
276 #endif
277 
278 	struct cfs_bandwidth cfs_bandwidth;
279 };
280 
281 #ifdef CONFIG_FAIR_GROUP_SCHED
282 #define ROOT_TASK_GROUP_LOAD	NICE_0_LOAD
283 
284 /*
285  * A weight of 0 or 1 can cause arithmetics problems.
286  * A weight of a cfs_rq is the sum of weights of which entities
287  * are queued on this cfs_rq, so a weight of a entity should not be
288  * too large, so as the shares value of a task group.
289  * (The default weight is 1024 - so there's no practical
290  *  limitation from this.)
291  */
292 #define MIN_SHARES	(1UL <<  1)
293 #define MAX_SHARES	(1UL << 18)
294 #endif
295 
296 typedef int (*tg_visitor)(struct task_group *, void *);
297 
298 extern int walk_tg_tree_from(struct task_group *from,
299 			     tg_visitor down, tg_visitor up, void *data);
300 
301 /*
302  * Iterate the full tree, calling @down when first entering a node and @up when
303  * leaving it for the final time.
304  *
305  * Caller must hold rcu_lock or sufficient equivalent.
306  */
307 static inline int walk_tg_tree(tg_visitor down, tg_visitor up, void *data)
308 {
309 	return walk_tg_tree_from(&root_task_group, down, up, data);
310 }
311 
312 extern int tg_nop(struct task_group *tg, void *data);
313 
314 extern void free_fair_sched_group(struct task_group *tg);
315 extern int alloc_fair_sched_group(struct task_group *tg, struct task_group *parent);
316 extern void unregister_fair_sched_group(struct task_group *tg, int cpu);
317 extern void init_tg_cfs_entry(struct task_group *tg, struct cfs_rq *cfs_rq,
318 			struct sched_entity *se, int cpu,
319 			struct sched_entity *parent);
320 extern void init_cfs_bandwidth(struct cfs_bandwidth *cfs_b);
321 extern int sched_group_set_shares(struct task_group *tg, unsigned long shares);
322 
323 extern void __refill_cfs_bandwidth_runtime(struct cfs_bandwidth *cfs_b);
324 extern void start_cfs_bandwidth(struct cfs_bandwidth *cfs_b);
325 extern void unthrottle_cfs_rq(struct cfs_rq *cfs_rq);
326 
327 extern void free_rt_sched_group(struct task_group *tg);
328 extern int alloc_rt_sched_group(struct task_group *tg, struct task_group *parent);
329 extern void init_tg_rt_entry(struct task_group *tg, struct rt_rq *rt_rq,
330 		struct sched_rt_entity *rt_se, int cpu,
331 		struct sched_rt_entity *parent);
332 
333 extern struct task_group *sched_create_group(struct task_group *parent);
334 extern void sched_online_group(struct task_group *tg,
335 			       struct task_group *parent);
336 extern void sched_destroy_group(struct task_group *tg);
337 extern void sched_offline_group(struct task_group *tg);
338 
339 extern void sched_move_task(struct task_struct *tsk);
340 
341 #ifdef CONFIG_FAIR_GROUP_SCHED
342 extern int sched_group_set_shares(struct task_group *tg, unsigned long shares);
343 
344 #ifdef CONFIG_SMP
345 extern void set_task_rq_fair(struct sched_entity *se,
346 			     struct cfs_rq *prev, struct cfs_rq *next);
347 #else /* !CONFIG_SMP */
348 static inline void set_task_rq_fair(struct sched_entity *se,
349 			     struct cfs_rq *prev, struct cfs_rq *next) { }
350 #endif /* CONFIG_SMP */
351 #endif /* CONFIG_FAIR_GROUP_SCHED */
352 
353 #else /* CONFIG_CGROUP_SCHED */
354 
355 struct cfs_bandwidth { };
356 
357 #endif	/* CONFIG_CGROUP_SCHED */
358 
359 /* CFS-related fields in a runqueue */
360 struct cfs_rq {
361 	struct load_weight load;
362 	unsigned int nr_running, h_nr_running;
363 
364 	u64 exec_clock;
365 	u64 min_vruntime;
366 #ifndef CONFIG_64BIT
367 	u64 min_vruntime_copy;
368 #endif
369 
370 	struct rb_root tasks_timeline;
371 	struct rb_node *rb_leftmost;
372 
373 	/*
374 	 * 'curr' points to currently running entity on this cfs_rq.
375 	 * It is set to NULL otherwise (i.e when none are currently running).
376 	 */
377 	struct sched_entity *curr, *next, *last, *skip;
378 
379 #ifdef	CONFIG_SCHED_DEBUG
380 	unsigned int nr_spread_over;
381 #endif
382 
383 #ifdef CONFIG_SMP
384 	/*
385 	 * CFS load tracking
386 	 */
387 	struct sched_avg avg;
388 	u64 runnable_load_sum;
389 	unsigned long runnable_load_avg;
390 #ifdef CONFIG_FAIR_GROUP_SCHED
391 	unsigned long tg_load_avg_contrib;
392 #endif
393 	atomic_long_t removed_load_avg, removed_util_avg;
394 #ifndef CONFIG_64BIT
395 	u64 load_last_update_time_copy;
396 #endif
397 
398 #ifdef CONFIG_FAIR_GROUP_SCHED
399 	/*
400 	 *   h_load = weight * f(tg)
401 	 *
402 	 * Where f(tg) is the recursive weight fraction assigned to
403 	 * this group.
404 	 */
405 	unsigned long h_load;
406 	u64 last_h_load_update;
407 	struct sched_entity *h_load_next;
408 #endif /* CONFIG_FAIR_GROUP_SCHED */
409 #endif /* CONFIG_SMP */
410 
411 #ifdef CONFIG_FAIR_GROUP_SCHED
412 	struct rq *rq;	/* cpu runqueue to which this cfs_rq is attached */
413 
414 	/*
415 	 * leaf cfs_rqs are those that hold tasks (lowest schedulable entity in
416 	 * a hierarchy). Non-leaf lrqs hold other higher schedulable entities
417 	 * (like users, containers etc.)
418 	 *
419 	 * leaf_cfs_rq_list ties together list of leaf cfs_rq's in a cpu. This
420 	 * list is used during load balance.
421 	 */
422 	int on_list;
423 	struct list_head leaf_cfs_rq_list;
424 	struct task_group *tg;	/* group that "owns" this runqueue */
425 
426 #ifdef CONFIG_CFS_BANDWIDTH
427 	int runtime_enabled;
428 	u64 runtime_expires;
429 	s64 runtime_remaining;
430 
431 	u64 throttled_clock, throttled_clock_task;
432 	u64 throttled_clock_task_time;
433 	int throttled, throttle_count;
434 	struct list_head throttled_list;
435 #endif /* CONFIG_CFS_BANDWIDTH */
436 #endif /* CONFIG_FAIR_GROUP_SCHED */
437 };
438 
439 static inline int rt_bandwidth_enabled(void)
440 {
441 	return sysctl_sched_rt_runtime >= 0;
442 }
443 
444 /* RT IPI pull logic requires IRQ_WORK */
445 #ifdef CONFIG_IRQ_WORK
446 # define HAVE_RT_PUSH_IPI
447 #endif
448 
449 /* Real-Time classes' related field in a runqueue: */
450 struct rt_rq {
451 	struct rt_prio_array active;
452 	unsigned int rt_nr_running;
453 #if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
454 	struct {
455 		int curr; /* highest queued rt task prio */
456 #ifdef CONFIG_SMP
457 		int next; /* next highest */
458 #endif
459 	} highest_prio;
460 #endif
461 #ifdef CONFIG_SMP
462 	unsigned long rt_nr_migratory;
463 	unsigned long rt_nr_total;
464 	int overloaded;
465 	struct plist_head pushable_tasks;
466 #ifdef HAVE_RT_PUSH_IPI
467 	int push_flags;
468 	int push_cpu;
469 	struct irq_work push_work;
470 	raw_spinlock_t push_lock;
471 #endif
472 #endif /* CONFIG_SMP */
473 	int rt_queued;
474 
475 	int rt_throttled;
476 	u64 rt_time;
477 	u64 rt_runtime;
478 	/* Nests inside the rq lock: */
479 	raw_spinlock_t rt_runtime_lock;
480 
481 #ifdef CONFIG_RT_GROUP_SCHED
482 	unsigned long rt_nr_boosted;
483 
484 	struct rq *rq;
485 	struct task_group *tg;
486 #endif
487 };
488 
489 /* Deadline class' related fields in a runqueue */
490 struct dl_rq {
491 	/* runqueue is an rbtree, ordered by deadline */
492 	struct rb_root rb_root;
493 	struct rb_node *rb_leftmost;
494 
495 	unsigned long dl_nr_running;
496 
497 #ifdef CONFIG_SMP
498 	/*
499 	 * Deadline values of the currently executing and the
500 	 * earliest ready task on this rq. Caching these facilitates
501 	 * the decision wether or not a ready but not running task
502 	 * should migrate somewhere else.
503 	 */
504 	struct {
505 		u64 curr;
506 		u64 next;
507 	} earliest_dl;
508 
509 	unsigned long dl_nr_migratory;
510 	int overloaded;
511 
512 	/*
513 	 * Tasks on this rq that can be pushed away. They are kept in
514 	 * an rb-tree, ordered by tasks' deadlines, with caching
515 	 * of the leftmost (earliest deadline) element.
516 	 */
517 	struct rb_root pushable_dl_tasks_root;
518 	struct rb_node *pushable_dl_tasks_leftmost;
519 #else
520 	struct dl_bw dl_bw;
521 #endif
522 };
523 
524 #ifdef CONFIG_SMP
525 
526 /*
527  * We add the notion of a root-domain which will be used to define per-domain
528  * variables. Each exclusive cpuset essentially defines an island domain by
529  * fully partitioning the member cpus from any other cpuset. Whenever a new
530  * exclusive cpuset is created, we also create and attach a new root-domain
531  * object.
532  *
533  */
534 struct root_domain {
535 	atomic_t refcount;
536 	atomic_t rto_count;
537 	struct rcu_head rcu;
538 	cpumask_var_t span;
539 	cpumask_var_t online;
540 
541 	/* Indicate more than one runnable task for any CPU */
542 	bool overload;
543 
544 	/*
545 	 * The bit corresponding to a CPU gets set here if such CPU has more
546 	 * than one runnable -deadline task (as it is below for RT tasks).
547 	 */
548 	cpumask_var_t dlo_mask;
549 	atomic_t dlo_count;
550 	struct dl_bw dl_bw;
551 	struct cpudl cpudl;
552 
553 	/*
554 	 * The "RT overload" flag: it gets set if a CPU has more than
555 	 * one runnable RT task.
556 	 */
557 	cpumask_var_t rto_mask;
558 	struct cpupri cpupri;
559 };
560 
561 extern struct root_domain def_root_domain;
562 
563 #endif /* CONFIG_SMP */
564 
565 /*
566  * This is the main, per-CPU runqueue data structure.
567  *
568  * Locking rule: those places that want to lock multiple runqueues
569  * (such as the load balancing or the thread migration code), lock
570  * acquire operations must be ordered by ascending &runqueue.
571  */
572 struct rq {
573 	/* runqueue lock: */
574 	raw_spinlock_t lock;
575 
576 	/*
577 	 * nr_running and cpu_load should be in the same cacheline because
578 	 * remote CPUs use both these fields when doing load calculation.
579 	 */
580 	unsigned int nr_running;
581 #ifdef CONFIG_NUMA_BALANCING
582 	unsigned int nr_numa_running;
583 	unsigned int nr_preferred_running;
584 #endif
585 	#define CPU_LOAD_IDX_MAX 5
586 	unsigned long cpu_load[CPU_LOAD_IDX_MAX];
587 	unsigned long last_load_update_tick;
588 #ifdef CONFIG_NO_HZ_COMMON
589 	u64 nohz_stamp;
590 	unsigned long nohz_flags;
591 #endif
592 #ifdef CONFIG_NO_HZ_FULL
593 	unsigned long last_sched_tick;
594 #endif
595 	/* capture load from *all* tasks on this cpu: */
596 	struct load_weight load;
597 	unsigned long nr_load_updates;
598 	u64 nr_switches;
599 
600 	struct cfs_rq cfs;
601 	struct rt_rq rt;
602 	struct dl_rq dl;
603 
604 #ifdef CONFIG_FAIR_GROUP_SCHED
605 	/* list of leaf cfs_rq on this cpu: */
606 	struct list_head leaf_cfs_rq_list;
607 #endif /* CONFIG_FAIR_GROUP_SCHED */
608 
609 	/*
610 	 * This is part of a global counter where only the total sum
611 	 * over all CPUs matters. A task can increase this counter on
612 	 * one CPU and if it got migrated afterwards it may decrease
613 	 * it on another CPU. Always updated under the runqueue lock:
614 	 */
615 	unsigned long nr_uninterruptible;
616 
617 	struct task_struct *curr, *idle, *stop;
618 	unsigned long next_balance;
619 	struct mm_struct *prev_mm;
620 
621 	unsigned int clock_skip_update;
622 	u64 clock;
623 	u64 clock_task;
624 
625 	atomic_t nr_iowait;
626 
627 #ifdef CONFIG_SMP
628 	struct root_domain *rd;
629 	struct sched_domain *sd;
630 
631 	unsigned long cpu_capacity;
632 	unsigned long cpu_capacity_orig;
633 
634 	struct callback_head *balance_callback;
635 
636 	unsigned char idle_balance;
637 	/* For active balancing */
638 	int active_balance;
639 	int push_cpu;
640 	struct cpu_stop_work active_balance_work;
641 	/* cpu of this runqueue: */
642 	int cpu;
643 	int online;
644 
645 	struct list_head cfs_tasks;
646 
647 	u64 rt_avg;
648 	u64 age_stamp;
649 	u64 idle_stamp;
650 	u64 avg_idle;
651 
652 	/* This is used to determine avg_idle's max value */
653 	u64 max_idle_balance_cost;
654 #endif
655 
656 #ifdef CONFIG_IRQ_TIME_ACCOUNTING
657 	u64 prev_irq_time;
658 #endif
659 #ifdef CONFIG_PARAVIRT
660 	u64 prev_steal_time;
661 #endif
662 #ifdef CONFIG_PARAVIRT_TIME_ACCOUNTING
663 	u64 prev_steal_time_rq;
664 #endif
665 
666 	/* calc_load related fields */
667 	unsigned long calc_load_update;
668 	long calc_load_active;
669 
670 #ifdef CONFIG_SCHED_HRTICK
671 #ifdef CONFIG_SMP
672 	int hrtick_csd_pending;
673 	struct call_single_data hrtick_csd;
674 #endif
675 	struct hrtimer hrtick_timer;
676 #endif
677 
678 #ifdef CONFIG_SCHEDSTATS
679 	/* latency stats */
680 	struct sched_info rq_sched_info;
681 	unsigned long long rq_cpu_time;
682 	/* could above be rq->cfs_rq.exec_clock + rq->rt_rq.rt_runtime ? */
683 
684 	/* sys_sched_yield() stats */
685 	unsigned int yld_count;
686 
687 	/* schedule() stats */
688 	unsigned int sched_count;
689 	unsigned int sched_goidle;
690 
691 	/* try_to_wake_up() stats */
692 	unsigned int ttwu_count;
693 	unsigned int ttwu_local;
694 #endif
695 
696 #ifdef CONFIG_SMP
697 	struct llist_head wake_list;
698 #endif
699 
700 #ifdef CONFIG_CPU_IDLE
701 	/* Must be inspected within a rcu lock section */
702 	struct cpuidle_state *idle_state;
703 #endif
704 };
705 
706 static inline int cpu_of(struct rq *rq)
707 {
708 #ifdef CONFIG_SMP
709 	return rq->cpu;
710 #else
711 	return 0;
712 #endif
713 }
714 
715 DECLARE_PER_CPU_SHARED_ALIGNED(struct rq, runqueues);
716 
717 #define cpu_rq(cpu)		(&per_cpu(runqueues, (cpu)))
718 #define this_rq()		this_cpu_ptr(&runqueues)
719 #define task_rq(p)		cpu_rq(task_cpu(p))
720 #define cpu_curr(cpu)		(cpu_rq(cpu)->curr)
721 #define raw_rq()		raw_cpu_ptr(&runqueues)
722 
723 static inline u64 __rq_clock_broken(struct rq *rq)
724 {
725 	return READ_ONCE(rq->clock);
726 }
727 
728 static inline u64 rq_clock(struct rq *rq)
729 {
730 	lockdep_assert_held(&rq->lock);
731 	return rq->clock;
732 }
733 
734 static inline u64 rq_clock_task(struct rq *rq)
735 {
736 	lockdep_assert_held(&rq->lock);
737 	return rq->clock_task;
738 }
739 
740 #define RQCF_REQ_SKIP	0x01
741 #define RQCF_ACT_SKIP	0x02
742 
743 static inline void rq_clock_skip_update(struct rq *rq, bool skip)
744 {
745 	lockdep_assert_held(&rq->lock);
746 	if (skip)
747 		rq->clock_skip_update |= RQCF_REQ_SKIP;
748 	else
749 		rq->clock_skip_update &= ~RQCF_REQ_SKIP;
750 }
751 
752 #ifdef CONFIG_NUMA
753 enum numa_topology_type {
754 	NUMA_DIRECT,
755 	NUMA_GLUELESS_MESH,
756 	NUMA_BACKPLANE,
757 };
758 extern enum numa_topology_type sched_numa_topology_type;
759 extern int sched_max_numa_distance;
760 extern bool find_numa_distance(int distance);
761 #endif
762 
763 #ifdef CONFIG_NUMA_BALANCING
764 /* The regions in numa_faults array from task_struct */
765 enum numa_faults_stats {
766 	NUMA_MEM = 0,
767 	NUMA_CPU,
768 	NUMA_MEMBUF,
769 	NUMA_CPUBUF
770 };
771 extern void sched_setnuma(struct task_struct *p, int node);
772 extern int migrate_task_to(struct task_struct *p, int cpu);
773 extern int migrate_swap(struct task_struct *, struct task_struct *);
774 #endif /* CONFIG_NUMA_BALANCING */
775 
776 #ifdef CONFIG_SMP
777 
778 static inline void
779 queue_balance_callback(struct rq *rq,
780 		       struct callback_head *head,
781 		       void (*func)(struct rq *rq))
782 {
783 	lockdep_assert_held(&rq->lock);
784 
785 	if (unlikely(head->next))
786 		return;
787 
788 	head->func = (void (*)(struct callback_head *))func;
789 	head->next = rq->balance_callback;
790 	rq->balance_callback = head;
791 }
792 
793 extern void sched_ttwu_pending(void);
794 
795 #define rcu_dereference_check_sched_domain(p) \
796 	rcu_dereference_check((p), \
797 			      lockdep_is_held(&sched_domains_mutex))
798 
799 /*
800  * The domain tree (rq->sd) is protected by RCU's quiescent state transition.
801  * See detach_destroy_domains: synchronize_sched for details.
802  *
803  * The domain tree of any CPU may only be accessed from within
804  * preempt-disabled sections.
805  */
806 #define for_each_domain(cpu, __sd) \
807 	for (__sd = rcu_dereference_check_sched_domain(cpu_rq(cpu)->sd); \
808 			__sd; __sd = __sd->parent)
809 
810 #define for_each_lower_domain(sd) for (; sd; sd = sd->child)
811 
812 /**
813  * highest_flag_domain - Return highest sched_domain containing flag.
814  * @cpu:	The cpu whose highest level of sched domain is to
815  *		be returned.
816  * @flag:	The flag to check for the highest sched_domain
817  *		for the given cpu.
818  *
819  * Returns the highest sched_domain of a cpu which contains the given flag.
820  */
821 static inline struct sched_domain *highest_flag_domain(int cpu, int flag)
822 {
823 	struct sched_domain *sd, *hsd = NULL;
824 
825 	for_each_domain(cpu, sd) {
826 		if (!(sd->flags & flag))
827 			break;
828 		hsd = sd;
829 	}
830 
831 	return hsd;
832 }
833 
834 static inline struct sched_domain *lowest_flag_domain(int cpu, int flag)
835 {
836 	struct sched_domain *sd;
837 
838 	for_each_domain(cpu, sd) {
839 		if (sd->flags & flag)
840 			break;
841 	}
842 
843 	return sd;
844 }
845 
846 DECLARE_PER_CPU(struct sched_domain *, sd_llc);
847 DECLARE_PER_CPU(int, sd_llc_size);
848 DECLARE_PER_CPU(int, sd_llc_id);
849 DECLARE_PER_CPU(struct sched_domain *, sd_numa);
850 DECLARE_PER_CPU(struct sched_domain *, sd_busy);
851 DECLARE_PER_CPU(struct sched_domain *, sd_asym);
852 
853 struct sched_group_capacity {
854 	atomic_t ref;
855 	/*
856 	 * CPU capacity of this group, SCHED_LOAD_SCALE being max capacity
857 	 * for a single CPU.
858 	 */
859 	unsigned int capacity;
860 	unsigned long next_update;
861 	int imbalance; /* XXX unrelated to capacity but shared group state */
862 	/*
863 	 * Number of busy cpus in this group.
864 	 */
865 	atomic_t nr_busy_cpus;
866 
867 	unsigned long cpumask[0]; /* iteration mask */
868 };
869 
870 struct sched_group {
871 	struct sched_group *next;	/* Must be a circular list */
872 	atomic_t ref;
873 
874 	unsigned int group_weight;
875 	struct sched_group_capacity *sgc;
876 
877 	/*
878 	 * The CPUs this group covers.
879 	 *
880 	 * NOTE: this field is variable length. (Allocated dynamically
881 	 * by attaching extra space to the end of the structure,
882 	 * depending on how many CPUs the kernel has booted up with)
883 	 */
884 	unsigned long cpumask[0];
885 };
886 
887 static inline struct cpumask *sched_group_cpus(struct sched_group *sg)
888 {
889 	return to_cpumask(sg->cpumask);
890 }
891 
892 /*
893  * cpumask masking which cpus in the group are allowed to iterate up the domain
894  * tree.
895  */
896 static inline struct cpumask *sched_group_mask(struct sched_group *sg)
897 {
898 	return to_cpumask(sg->sgc->cpumask);
899 }
900 
901 /**
902  * group_first_cpu - Returns the first cpu in the cpumask of a sched_group.
903  * @group: The group whose first cpu is to be returned.
904  */
905 static inline unsigned int group_first_cpu(struct sched_group *group)
906 {
907 	return cpumask_first(sched_group_cpus(group));
908 }
909 
910 extern int group_balance_cpu(struct sched_group *sg);
911 
912 #else
913 
914 static inline void sched_ttwu_pending(void) { }
915 
916 #endif /* CONFIG_SMP */
917 
918 #include "stats.h"
919 #include "auto_group.h"
920 
921 #ifdef CONFIG_CGROUP_SCHED
922 
923 /*
924  * Return the group to which this tasks belongs.
925  *
926  * We cannot use task_css() and friends because the cgroup subsystem
927  * changes that value before the cgroup_subsys::attach() method is called,
928  * therefore we cannot pin it and might observe the wrong value.
929  *
930  * The same is true for autogroup's p->signal->autogroup->tg, the autogroup
931  * core changes this before calling sched_move_task().
932  *
933  * Instead we use a 'copy' which is updated from sched_move_task() while
934  * holding both task_struct::pi_lock and rq::lock.
935  */
936 static inline struct task_group *task_group(struct task_struct *p)
937 {
938 	return p->sched_task_group;
939 }
940 
941 /* Change a task's cfs_rq and parent entity if it moves across CPUs/groups */
942 static inline void set_task_rq(struct task_struct *p, unsigned int cpu)
943 {
944 #if defined(CONFIG_FAIR_GROUP_SCHED) || defined(CONFIG_RT_GROUP_SCHED)
945 	struct task_group *tg = task_group(p);
946 #endif
947 
948 #ifdef CONFIG_FAIR_GROUP_SCHED
949 	set_task_rq_fair(&p->se, p->se.cfs_rq, tg->cfs_rq[cpu]);
950 	p->se.cfs_rq = tg->cfs_rq[cpu];
951 	p->se.parent = tg->se[cpu];
952 #endif
953 
954 #ifdef CONFIG_RT_GROUP_SCHED
955 	p->rt.rt_rq  = tg->rt_rq[cpu];
956 	p->rt.parent = tg->rt_se[cpu];
957 #endif
958 }
959 
960 #else /* CONFIG_CGROUP_SCHED */
961 
962 static inline void set_task_rq(struct task_struct *p, unsigned int cpu) { }
963 static inline struct task_group *task_group(struct task_struct *p)
964 {
965 	return NULL;
966 }
967 
968 #endif /* CONFIG_CGROUP_SCHED */
969 
970 static inline void __set_task_cpu(struct task_struct *p, unsigned int cpu)
971 {
972 	set_task_rq(p, cpu);
973 #ifdef CONFIG_SMP
974 	/*
975 	 * After ->cpu is set up to a new value, task_rq_lock(p, ...) can be
976 	 * successfuly executed on another CPU. We must ensure that updates of
977 	 * per-task data have been completed by this moment.
978 	 */
979 	smp_wmb();
980 	task_thread_info(p)->cpu = cpu;
981 	p->wake_cpu = cpu;
982 #endif
983 }
984 
985 /*
986  * Tunables that become constants when CONFIG_SCHED_DEBUG is off:
987  */
988 #ifdef CONFIG_SCHED_DEBUG
989 # include <linux/static_key.h>
990 # define const_debug __read_mostly
991 #else
992 # define const_debug const
993 #endif
994 
995 extern const_debug unsigned int sysctl_sched_features;
996 
997 #define SCHED_FEAT(name, enabled)	\
998 	__SCHED_FEAT_##name ,
999 
1000 enum {
1001 #include "features.h"
1002 	__SCHED_FEAT_NR,
1003 };
1004 
1005 #undef SCHED_FEAT
1006 
1007 #if defined(CONFIG_SCHED_DEBUG) && defined(HAVE_JUMP_LABEL)
1008 #define SCHED_FEAT(name, enabled)					\
1009 static __always_inline bool static_branch_##name(struct static_key *key) \
1010 {									\
1011 	return static_key_##enabled(key);				\
1012 }
1013 
1014 #include "features.h"
1015 
1016 #undef SCHED_FEAT
1017 
1018 extern struct static_key sched_feat_keys[__SCHED_FEAT_NR];
1019 #define sched_feat(x) (static_branch_##x(&sched_feat_keys[__SCHED_FEAT_##x]))
1020 #else /* !(SCHED_DEBUG && HAVE_JUMP_LABEL) */
1021 #define sched_feat(x) (sysctl_sched_features & (1UL << __SCHED_FEAT_##x))
1022 #endif /* SCHED_DEBUG && HAVE_JUMP_LABEL */
1023 
1024 extern struct static_key_false sched_numa_balancing;
1025 
1026 static inline u64 global_rt_period(void)
1027 {
1028 	return (u64)sysctl_sched_rt_period * NSEC_PER_USEC;
1029 }
1030 
1031 static inline u64 global_rt_runtime(void)
1032 {
1033 	if (sysctl_sched_rt_runtime < 0)
1034 		return RUNTIME_INF;
1035 
1036 	return (u64)sysctl_sched_rt_runtime * NSEC_PER_USEC;
1037 }
1038 
1039 static inline int task_current(struct rq *rq, struct task_struct *p)
1040 {
1041 	return rq->curr == p;
1042 }
1043 
1044 static inline int task_running(struct rq *rq, struct task_struct *p)
1045 {
1046 #ifdef CONFIG_SMP
1047 	return p->on_cpu;
1048 #else
1049 	return task_current(rq, p);
1050 #endif
1051 }
1052 
1053 static inline int task_on_rq_queued(struct task_struct *p)
1054 {
1055 	return p->on_rq == TASK_ON_RQ_QUEUED;
1056 }
1057 
1058 static inline int task_on_rq_migrating(struct task_struct *p)
1059 {
1060 	return p->on_rq == TASK_ON_RQ_MIGRATING;
1061 }
1062 
1063 #ifndef prepare_arch_switch
1064 # define prepare_arch_switch(next)	do { } while (0)
1065 #endif
1066 #ifndef finish_arch_post_lock_switch
1067 # define finish_arch_post_lock_switch()	do { } while (0)
1068 #endif
1069 
1070 static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
1071 {
1072 #ifdef CONFIG_SMP
1073 	/*
1074 	 * We can optimise this out completely for !SMP, because the
1075 	 * SMP rebalancing from interrupt is the only thing that cares
1076 	 * here.
1077 	 */
1078 	next->on_cpu = 1;
1079 #endif
1080 }
1081 
1082 static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
1083 {
1084 #ifdef CONFIG_SMP
1085 	/*
1086 	 * After ->on_cpu is cleared, the task can be moved to a different CPU.
1087 	 * We must ensure this doesn't happen until the switch is completely
1088 	 * finished.
1089 	 *
1090 	 * In particular, the load of prev->state in finish_task_switch() must
1091 	 * happen before this.
1092 	 *
1093 	 * Pairs with the smp_cond_acquire() in try_to_wake_up().
1094 	 */
1095 	smp_store_release(&prev->on_cpu, 0);
1096 #endif
1097 #ifdef CONFIG_DEBUG_SPINLOCK
1098 	/* this is a valid case when another task releases the spinlock */
1099 	rq->lock.owner = current;
1100 #endif
1101 	/*
1102 	 * If we are tracking spinlock dependencies then we have to
1103 	 * fix up the runqueue lock - which gets 'carried over' from
1104 	 * prev into current:
1105 	 */
1106 	spin_acquire(&rq->lock.dep_map, 0, 0, _THIS_IP_);
1107 
1108 	raw_spin_unlock_irq(&rq->lock);
1109 }
1110 
1111 /*
1112  * wake flags
1113  */
1114 #define WF_SYNC		0x01		/* waker goes to sleep after wakeup */
1115 #define WF_FORK		0x02		/* child wakeup after fork */
1116 #define WF_MIGRATED	0x4		/* internal use, task got migrated */
1117 
1118 /*
1119  * To aid in avoiding the subversion of "niceness" due to uneven distribution
1120  * of tasks with abnormal "nice" values across CPUs the contribution that
1121  * each task makes to its run queue's load is weighted according to its
1122  * scheduling class and "nice" value. For SCHED_NORMAL tasks this is just a
1123  * scaled version of the new time slice allocation that they receive on time
1124  * slice expiry etc.
1125  */
1126 
1127 #define WEIGHT_IDLEPRIO                3
1128 #define WMULT_IDLEPRIO         1431655765
1129 
1130 extern const int sched_prio_to_weight[40];
1131 extern const u32 sched_prio_to_wmult[40];
1132 
1133 #define ENQUEUE_WAKEUP		0x01
1134 #define ENQUEUE_HEAD		0x02
1135 #ifdef CONFIG_SMP
1136 #define ENQUEUE_WAKING		0x04	/* sched_class::task_waking was called */
1137 #else
1138 #define ENQUEUE_WAKING		0x00
1139 #endif
1140 #define ENQUEUE_REPLENISH	0x08
1141 #define ENQUEUE_RESTORE	0x10
1142 
1143 #define DEQUEUE_SLEEP		0x01
1144 #define DEQUEUE_SAVE		0x02
1145 
1146 #define RETRY_TASK		((void *)-1UL)
1147 
1148 struct sched_class {
1149 	const struct sched_class *next;
1150 
1151 	void (*enqueue_task) (struct rq *rq, struct task_struct *p, int flags);
1152 	void (*dequeue_task) (struct rq *rq, struct task_struct *p, int flags);
1153 	void (*yield_task) (struct rq *rq);
1154 	bool (*yield_to_task) (struct rq *rq, struct task_struct *p, bool preempt);
1155 
1156 	void (*check_preempt_curr) (struct rq *rq, struct task_struct *p, int flags);
1157 
1158 	/*
1159 	 * It is the responsibility of the pick_next_task() method that will
1160 	 * return the next task to call put_prev_task() on the @prev task or
1161 	 * something equivalent.
1162 	 *
1163 	 * May return RETRY_TASK when it finds a higher prio class has runnable
1164 	 * tasks.
1165 	 */
1166 	struct task_struct * (*pick_next_task) (struct rq *rq,
1167 						struct task_struct *prev);
1168 	void (*put_prev_task) (struct rq *rq, struct task_struct *p);
1169 
1170 #ifdef CONFIG_SMP
1171 	int  (*select_task_rq)(struct task_struct *p, int task_cpu, int sd_flag, int flags);
1172 	void (*migrate_task_rq)(struct task_struct *p);
1173 
1174 	void (*task_waking) (struct task_struct *task);
1175 	void (*task_woken) (struct rq *this_rq, struct task_struct *task);
1176 
1177 	void (*set_cpus_allowed)(struct task_struct *p,
1178 				 const struct cpumask *newmask);
1179 
1180 	void (*rq_online)(struct rq *rq);
1181 	void (*rq_offline)(struct rq *rq);
1182 #endif
1183 
1184 	void (*set_curr_task) (struct rq *rq);
1185 	void (*task_tick) (struct rq *rq, struct task_struct *p, int queued);
1186 	void (*task_fork) (struct task_struct *p);
1187 	void (*task_dead) (struct task_struct *p);
1188 
1189 	/*
1190 	 * The switched_from() call is allowed to drop rq->lock, therefore we
1191 	 * cannot assume the switched_from/switched_to pair is serliazed by
1192 	 * rq->lock. They are however serialized by p->pi_lock.
1193 	 */
1194 	void (*switched_from) (struct rq *this_rq, struct task_struct *task);
1195 	void (*switched_to) (struct rq *this_rq, struct task_struct *task);
1196 	void (*prio_changed) (struct rq *this_rq, struct task_struct *task,
1197 			     int oldprio);
1198 
1199 	unsigned int (*get_rr_interval) (struct rq *rq,
1200 					 struct task_struct *task);
1201 
1202 	void (*update_curr) (struct rq *rq);
1203 
1204 #ifdef CONFIG_FAIR_GROUP_SCHED
1205 	void (*task_move_group) (struct task_struct *p);
1206 #endif
1207 };
1208 
1209 static inline void put_prev_task(struct rq *rq, struct task_struct *prev)
1210 {
1211 	prev->sched_class->put_prev_task(rq, prev);
1212 }
1213 
1214 #define sched_class_highest (&stop_sched_class)
1215 #define for_each_class(class) \
1216    for (class = sched_class_highest; class; class = class->next)
1217 
1218 extern const struct sched_class stop_sched_class;
1219 extern const struct sched_class dl_sched_class;
1220 extern const struct sched_class rt_sched_class;
1221 extern const struct sched_class fair_sched_class;
1222 extern const struct sched_class idle_sched_class;
1223 
1224 
1225 #ifdef CONFIG_SMP
1226 
1227 extern void update_group_capacity(struct sched_domain *sd, int cpu);
1228 
1229 extern void trigger_load_balance(struct rq *rq);
1230 
1231 extern void set_cpus_allowed_common(struct task_struct *p, const struct cpumask *new_mask);
1232 
1233 #endif
1234 
1235 #ifdef CONFIG_CPU_IDLE
1236 static inline void idle_set_state(struct rq *rq,
1237 				  struct cpuidle_state *idle_state)
1238 {
1239 	rq->idle_state = idle_state;
1240 }
1241 
1242 static inline struct cpuidle_state *idle_get_state(struct rq *rq)
1243 {
1244 	WARN_ON(!rcu_read_lock_held());
1245 	return rq->idle_state;
1246 }
1247 #else
1248 static inline void idle_set_state(struct rq *rq,
1249 				  struct cpuidle_state *idle_state)
1250 {
1251 }
1252 
1253 static inline struct cpuidle_state *idle_get_state(struct rq *rq)
1254 {
1255 	return NULL;
1256 }
1257 #endif
1258 
1259 extern void sysrq_sched_debug_show(void);
1260 extern void sched_init_granularity(void);
1261 extern void update_max_interval(void);
1262 
1263 extern void init_sched_dl_class(void);
1264 extern void init_sched_rt_class(void);
1265 extern void init_sched_fair_class(void);
1266 
1267 extern void resched_curr(struct rq *rq);
1268 extern void resched_cpu(int cpu);
1269 
1270 extern struct rt_bandwidth def_rt_bandwidth;
1271 extern void init_rt_bandwidth(struct rt_bandwidth *rt_b, u64 period, u64 runtime);
1272 
1273 extern struct dl_bandwidth def_dl_bandwidth;
1274 extern void init_dl_bandwidth(struct dl_bandwidth *dl_b, u64 period, u64 runtime);
1275 extern void init_dl_task_timer(struct sched_dl_entity *dl_se);
1276 
1277 unsigned long to_ratio(u64 period, u64 runtime);
1278 
1279 extern void init_entity_runnable_average(struct sched_entity *se);
1280 
1281 static inline void add_nr_running(struct rq *rq, unsigned count)
1282 {
1283 	unsigned prev_nr = rq->nr_running;
1284 
1285 	rq->nr_running = prev_nr + count;
1286 
1287 	if (prev_nr < 2 && rq->nr_running >= 2) {
1288 #ifdef CONFIG_SMP
1289 		if (!rq->rd->overload)
1290 			rq->rd->overload = true;
1291 #endif
1292 
1293 #ifdef CONFIG_NO_HZ_FULL
1294 		if (tick_nohz_full_cpu(rq->cpu)) {
1295 			/*
1296 			 * Tick is needed if more than one task runs on a CPU.
1297 			 * Send the target an IPI to kick it out of nohz mode.
1298 			 *
1299 			 * We assume that IPI implies full memory barrier and the
1300 			 * new value of rq->nr_running is visible on reception
1301 			 * from the target.
1302 			 */
1303 			tick_nohz_full_kick_cpu(rq->cpu);
1304 		}
1305 #endif
1306 	}
1307 }
1308 
1309 static inline void sub_nr_running(struct rq *rq, unsigned count)
1310 {
1311 	rq->nr_running -= count;
1312 }
1313 
1314 static inline void rq_last_tick_reset(struct rq *rq)
1315 {
1316 #ifdef CONFIG_NO_HZ_FULL
1317 	rq->last_sched_tick = jiffies;
1318 #endif
1319 }
1320 
1321 extern void update_rq_clock(struct rq *rq);
1322 
1323 extern void activate_task(struct rq *rq, struct task_struct *p, int flags);
1324 extern void deactivate_task(struct rq *rq, struct task_struct *p, int flags);
1325 
1326 extern void check_preempt_curr(struct rq *rq, struct task_struct *p, int flags);
1327 
1328 extern const_debug unsigned int sysctl_sched_time_avg;
1329 extern const_debug unsigned int sysctl_sched_nr_migrate;
1330 extern const_debug unsigned int sysctl_sched_migration_cost;
1331 
1332 static inline u64 sched_avg_period(void)
1333 {
1334 	return (u64)sysctl_sched_time_avg * NSEC_PER_MSEC / 2;
1335 }
1336 
1337 #ifdef CONFIG_SCHED_HRTICK
1338 
1339 /*
1340  * Use hrtick when:
1341  *  - enabled by features
1342  *  - hrtimer is actually high res
1343  */
1344 static inline int hrtick_enabled(struct rq *rq)
1345 {
1346 	if (!sched_feat(HRTICK))
1347 		return 0;
1348 	if (!cpu_active(cpu_of(rq)))
1349 		return 0;
1350 	return hrtimer_is_hres_active(&rq->hrtick_timer);
1351 }
1352 
1353 void hrtick_start(struct rq *rq, u64 delay);
1354 
1355 #else
1356 
1357 static inline int hrtick_enabled(struct rq *rq)
1358 {
1359 	return 0;
1360 }
1361 
1362 #endif /* CONFIG_SCHED_HRTICK */
1363 
1364 #ifdef CONFIG_SMP
1365 extern void sched_avg_update(struct rq *rq);
1366 
1367 #ifndef arch_scale_freq_capacity
1368 static __always_inline
1369 unsigned long arch_scale_freq_capacity(struct sched_domain *sd, int cpu)
1370 {
1371 	return SCHED_CAPACITY_SCALE;
1372 }
1373 #endif
1374 
1375 #ifndef arch_scale_cpu_capacity
1376 static __always_inline
1377 unsigned long arch_scale_cpu_capacity(struct sched_domain *sd, int cpu)
1378 {
1379 	if (sd && (sd->flags & SD_SHARE_CPUCAPACITY) && (sd->span_weight > 1))
1380 		return sd->smt_gain / sd->span_weight;
1381 
1382 	return SCHED_CAPACITY_SCALE;
1383 }
1384 #endif
1385 
1386 static inline void sched_rt_avg_update(struct rq *rq, u64 rt_delta)
1387 {
1388 	rq->rt_avg += rt_delta * arch_scale_freq_capacity(NULL, cpu_of(rq));
1389 	sched_avg_update(rq);
1390 }
1391 #else
1392 static inline void sched_rt_avg_update(struct rq *rq, u64 rt_delta) { }
1393 static inline void sched_avg_update(struct rq *rq) { }
1394 #endif
1395 
1396 /*
1397  * __task_rq_lock - lock the rq @p resides on.
1398  */
1399 static inline struct rq *__task_rq_lock(struct task_struct *p)
1400 	__acquires(rq->lock)
1401 {
1402 	struct rq *rq;
1403 
1404 	lockdep_assert_held(&p->pi_lock);
1405 
1406 	for (;;) {
1407 		rq = task_rq(p);
1408 		raw_spin_lock(&rq->lock);
1409 		if (likely(rq == task_rq(p) && !task_on_rq_migrating(p))) {
1410 			lockdep_pin_lock(&rq->lock);
1411 			return rq;
1412 		}
1413 		raw_spin_unlock(&rq->lock);
1414 
1415 		while (unlikely(task_on_rq_migrating(p)))
1416 			cpu_relax();
1417 	}
1418 }
1419 
1420 /*
1421  * task_rq_lock - lock p->pi_lock and lock the rq @p resides on.
1422  */
1423 static inline struct rq *task_rq_lock(struct task_struct *p, unsigned long *flags)
1424 	__acquires(p->pi_lock)
1425 	__acquires(rq->lock)
1426 {
1427 	struct rq *rq;
1428 
1429 	for (;;) {
1430 		raw_spin_lock_irqsave(&p->pi_lock, *flags);
1431 		rq = task_rq(p);
1432 		raw_spin_lock(&rq->lock);
1433 		/*
1434 		 *	move_queued_task()		task_rq_lock()
1435 		 *
1436 		 *	ACQUIRE (rq->lock)
1437 		 *	[S] ->on_rq = MIGRATING		[L] rq = task_rq()
1438 		 *	WMB (__set_task_cpu())		ACQUIRE (rq->lock);
1439 		 *	[S] ->cpu = new_cpu		[L] task_rq()
1440 		 *					[L] ->on_rq
1441 		 *	RELEASE (rq->lock)
1442 		 *
1443 		 * If we observe the old cpu in task_rq_lock, the acquire of
1444 		 * the old rq->lock will fully serialize against the stores.
1445 		 *
1446 		 * If we observe the new cpu in task_rq_lock, the acquire will
1447 		 * pair with the WMB to ensure we must then also see migrating.
1448 		 */
1449 		if (likely(rq == task_rq(p) && !task_on_rq_migrating(p))) {
1450 			lockdep_pin_lock(&rq->lock);
1451 			return rq;
1452 		}
1453 		raw_spin_unlock(&rq->lock);
1454 		raw_spin_unlock_irqrestore(&p->pi_lock, *flags);
1455 
1456 		while (unlikely(task_on_rq_migrating(p)))
1457 			cpu_relax();
1458 	}
1459 }
1460 
1461 static inline void __task_rq_unlock(struct rq *rq)
1462 	__releases(rq->lock)
1463 {
1464 	lockdep_unpin_lock(&rq->lock);
1465 	raw_spin_unlock(&rq->lock);
1466 }
1467 
1468 static inline void
1469 task_rq_unlock(struct rq *rq, struct task_struct *p, unsigned long *flags)
1470 	__releases(rq->lock)
1471 	__releases(p->pi_lock)
1472 {
1473 	lockdep_unpin_lock(&rq->lock);
1474 	raw_spin_unlock(&rq->lock);
1475 	raw_spin_unlock_irqrestore(&p->pi_lock, *flags);
1476 }
1477 
1478 #ifdef CONFIG_SMP
1479 #ifdef CONFIG_PREEMPT
1480 
1481 static inline void double_rq_lock(struct rq *rq1, struct rq *rq2);
1482 
1483 /*
1484  * fair double_lock_balance: Safely acquires both rq->locks in a fair
1485  * way at the expense of forcing extra atomic operations in all
1486  * invocations.  This assures that the double_lock is acquired using the
1487  * same underlying policy as the spinlock_t on this architecture, which
1488  * reduces latency compared to the unfair variant below.  However, it
1489  * also adds more overhead and therefore may reduce throughput.
1490  */
1491 static inline int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
1492 	__releases(this_rq->lock)
1493 	__acquires(busiest->lock)
1494 	__acquires(this_rq->lock)
1495 {
1496 	raw_spin_unlock(&this_rq->lock);
1497 	double_rq_lock(this_rq, busiest);
1498 
1499 	return 1;
1500 }
1501 
1502 #else
1503 /*
1504  * Unfair double_lock_balance: Optimizes throughput at the expense of
1505  * latency by eliminating extra atomic operations when the locks are
1506  * already in proper order on entry.  This favors lower cpu-ids and will
1507  * grant the double lock to lower cpus over higher ids under contention,
1508  * regardless of entry order into the function.
1509  */
1510 static inline int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
1511 	__releases(this_rq->lock)
1512 	__acquires(busiest->lock)
1513 	__acquires(this_rq->lock)
1514 {
1515 	int ret = 0;
1516 
1517 	if (unlikely(!raw_spin_trylock(&busiest->lock))) {
1518 		if (busiest < this_rq) {
1519 			raw_spin_unlock(&this_rq->lock);
1520 			raw_spin_lock(&busiest->lock);
1521 			raw_spin_lock_nested(&this_rq->lock,
1522 					      SINGLE_DEPTH_NESTING);
1523 			ret = 1;
1524 		} else
1525 			raw_spin_lock_nested(&busiest->lock,
1526 					      SINGLE_DEPTH_NESTING);
1527 	}
1528 	return ret;
1529 }
1530 
1531 #endif /* CONFIG_PREEMPT */
1532 
1533 /*
1534  * double_lock_balance - lock the busiest runqueue, this_rq is locked already.
1535  */
1536 static inline int double_lock_balance(struct rq *this_rq, struct rq *busiest)
1537 {
1538 	if (unlikely(!irqs_disabled())) {
1539 		/* printk() doesn't work good under rq->lock */
1540 		raw_spin_unlock(&this_rq->lock);
1541 		BUG_ON(1);
1542 	}
1543 
1544 	return _double_lock_balance(this_rq, busiest);
1545 }
1546 
1547 static inline void double_unlock_balance(struct rq *this_rq, struct rq *busiest)
1548 	__releases(busiest->lock)
1549 {
1550 	raw_spin_unlock(&busiest->lock);
1551 	lock_set_subclass(&this_rq->lock.dep_map, 0, _RET_IP_);
1552 }
1553 
1554 static inline void double_lock(spinlock_t *l1, spinlock_t *l2)
1555 {
1556 	if (l1 > l2)
1557 		swap(l1, l2);
1558 
1559 	spin_lock(l1);
1560 	spin_lock_nested(l2, SINGLE_DEPTH_NESTING);
1561 }
1562 
1563 static inline void double_lock_irq(spinlock_t *l1, spinlock_t *l2)
1564 {
1565 	if (l1 > l2)
1566 		swap(l1, l2);
1567 
1568 	spin_lock_irq(l1);
1569 	spin_lock_nested(l2, SINGLE_DEPTH_NESTING);
1570 }
1571 
1572 static inline void double_raw_lock(raw_spinlock_t *l1, raw_spinlock_t *l2)
1573 {
1574 	if (l1 > l2)
1575 		swap(l1, l2);
1576 
1577 	raw_spin_lock(l1);
1578 	raw_spin_lock_nested(l2, SINGLE_DEPTH_NESTING);
1579 }
1580 
1581 /*
1582  * double_rq_lock - safely lock two runqueues
1583  *
1584  * Note this does not disable interrupts like task_rq_lock,
1585  * you need to do so manually before calling.
1586  */
1587 static inline void double_rq_lock(struct rq *rq1, struct rq *rq2)
1588 	__acquires(rq1->lock)
1589 	__acquires(rq2->lock)
1590 {
1591 	BUG_ON(!irqs_disabled());
1592 	if (rq1 == rq2) {
1593 		raw_spin_lock(&rq1->lock);
1594 		__acquire(rq2->lock);	/* Fake it out ;) */
1595 	} else {
1596 		if (rq1 < rq2) {
1597 			raw_spin_lock(&rq1->lock);
1598 			raw_spin_lock_nested(&rq2->lock, SINGLE_DEPTH_NESTING);
1599 		} else {
1600 			raw_spin_lock(&rq2->lock);
1601 			raw_spin_lock_nested(&rq1->lock, SINGLE_DEPTH_NESTING);
1602 		}
1603 	}
1604 }
1605 
1606 /*
1607  * double_rq_unlock - safely unlock two runqueues
1608  *
1609  * Note this does not restore interrupts like task_rq_unlock,
1610  * you need to do so manually after calling.
1611  */
1612 static inline void double_rq_unlock(struct rq *rq1, struct rq *rq2)
1613 	__releases(rq1->lock)
1614 	__releases(rq2->lock)
1615 {
1616 	raw_spin_unlock(&rq1->lock);
1617 	if (rq1 != rq2)
1618 		raw_spin_unlock(&rq2->lock);
1619 	else
1620 		__release(rq2->lock);
1621 }
1622 
1623 #else /* CONFIG_SMP */
1624 
1625 /*
1626  * double_rq_lock - safely lock two runqueues
1627  *
1628  * Note this does not disable interrupts like task_rq_lock,
1629  * you need to do so manually before calling.
1630  */
1631 static inline void double_rq_lock(struct rq *rq1, struct rq *rq2)
1632 	__acquires(rq1->lock)
1633 	__acquires(rq2->lock)
1634 {
1635 	BUG_ON(!irqs_disabled());
1636 	BUG_ON(rq1 != rq2);
1637 	raw_spin_lock(&rq1->lock);
1638 	__acquire(rq2->lock);	/* Fake it out ;) */
1639 }
1640 
1641 /*
1642  * double_rq_unlock - safely unlock two runqueues
1643  *
1644  * Note this does not restore interrupts like task_rq_unlock,
1645  * you need to do so manually after calling.
1646  */
1647 static inline void double_rq_unlock(struct rq *rq1, struct rq *rq2)
1648 	__releases(rq1->lock)
1649 	__releases(rq2->lock)
1650 {
1651 	BUG_ON(rq1 != rq2);
1652 	raw_spin_unlock(&rq1->lock);
1653 	__release(rq2->lock);
1654 }
1655 
1656 #endif
1657 
1658 extern struct sched_entity *__pick_first_entity(struct cfs_rq *cfs_rq);
1659 extern struct sched_entity *__pick_last_entity(struct cfs_rq *cfs_rq);
1660 
1661 #ifdef	CONFIG_SCHED_DEBUG
1662 extern void print_cfs_stats(struct seq_file *m, int cpu);
1663 extern void print_rt_stats(struct seq_file *m, int cpu);
1664 extern void print_dl_stats(struct seq_file *m, int cpu);
1665 extern void
1666 print_cfs_rq(struct seq_file *m, int cpu, struct cfs_rq *cfs_rq);
1667 
1668 #ifdef CONFIG_NUMA_BALANCING
1669 extern void
1670 show_numa_stats(struct task_struct *p, struct seq_file *m);
1671 extern void
1672 print_numa_stats(struct seq_file *m, int node, unsigned long tsf,
1673 	unsigned long tpf, unsigned long gsf, unsigned long gpf);
1674 #endif /* CONFIG_NUMA_BALANCING */
1675 #endif /* CONFIG_SCHED_DEBUG */
1676 
1677 extern void init_cfs_rq(struct cfs_rq *cfs_rq);
1678 extern void init_rt_rq(struct rt_rq *rt_rq);
1679 extern void init_dl_rq(struct dl_rq *dl_rq);
1680 
1681 extern void cfs_bandwidth_usage_inc(void);
1682 extern void cfs_bandwidth_usage_dec(void);
1683 
1684 #ifdef CONFIG_NO_HZ_COMMON
1685 enum rq_nohz_flag_bits {
1686 	NOHZ_TICK_STOPPED,
1687 	NOHZ_BALANCE_KICK,
1688 };
1689 
1690 #define nohz_flags(cpu)	(&cpu_rq(cpu)->nohz_flags)
1691 #endif
1692 
1693 #ifdef CONFIG_IRQ_TIME_ACCOUNTING
1694 
1695 DECLARE_PER_CPU(u64, cpu_hardirq_time);
1696 DECLARE_PER_CPU(u64, cpu_softirq_time);
1697 
1698 #ifndef CONFIG_64BIT
1699 DECLARE_PER_CPU(seqcount_t, irq_time_seq);
1700 
1701 static inline void irq_time_write_begin(void)
1702 {
1703 	__this_cpu_inc(irq_time_seq.sequence);
1704 	smp_wmb();
1705 }
1706 
1707 static inline void irq_time_write_end(void)
1708 {
1709 	smp_wmb();
1710 	__this_cpu_inc(irq_time_seq.sequence);
1711 }
1712 
1713 static inline u64 irq_time_read(int cpu)
1714 {
1715 	u64 irq_time;
1716 	unsigned seq;
1717 
1718 	do {
1719 		seq = read_seqcount_begin(&per_cpu(irq_time_seq, cpu));
1720 		irq_time = per_cpu(cpu_softirq_time, cpu) +
1721 			   per_cpu(cpu_hardirq_time, cpu);
1722 	} while (read_seqcount_retry(&per_cpu(irq_time_seq, cpu), seq));
1723 
1724 	return irq_time;
1725 }
1726 #else /* CONFIG_64BIT */
1727 static inline void irq_time_write_begin(void)
1728 {
1729 }
1730 
1731 static inline void irq_time_write_end(void)
1732 {
1733 }
1734 
1735 static inline u64 irq_time_read(int cpu)
1736 {
1737 	return per_cpu(cpu_softirq_time, cpu) + per_cpu(cpu_hardirq_time, cpu);
1738 }
1739 #endif /* CONFIG_64BIT */
1740 #endif /* CONFIG_IRQ_TIME_ACCOUNTING */
1741