sched.h 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155
  1. #include <linux/sched.h>
  2. #include <linux/mutex.h>
  3. #include <linux/spinlock.h>
  4. #include <linux/stop_machine.h>
  5. #include "cpupri.h"
  6. extern __read_mostly int scheduler_running;
  7. /*
  8. * Convert user-nice values [ -20 ... 0 ... 19 ]
  9. * to static priority [ MAX_RT_PRIO..MAX_PRIO-1 ],
  10. * and back.
  11. */
  12. #define NICE_TO_PRIO(nice) (MAX_RT_PRIO + (nice) + 20)
  13. #define PRIO_TO_NICE(prio) ((prio) - MAX_RT_PRIO - 20)
  14. #define TASK_NICE(p) PRIO_TO_NICE((p)->static_prio)
  15. /*
  16. * 'User priority' is the nice value converted to something we
  17. * can work with better when scaling various scheduler parameters,
  18. * it's a [ 0 ... 39 ] range.
  19. */
  20. #define USER_PRIO(p) ((p)-MAX_RT_PRIO)
  21. #define TASK_USER_PRIO(p) USER_PRIO((p)->static_prio)
  22. #define MAX_USER_PRIO (USER_PRIO(MAX_PRIO))
  23. /*
  24. * Helpers for converting nanosecond timing to jiffy resolution
  25. */
  26. #define NS_TO_JIFFIES(TIME) ((unsigned long)(TIME) / (NSEC_PER_SEC / HZ))
  27. #define NICE_0_LOAD SCHED_LOAD_SCALE
  28. #define NICE_0_SHIFT SCHED_LOAD_SHIFT
  29. /*
  30. * These are the 'tuning knobs' of the scheduler:
  31. */
  32. /*
  33. * single value that denotes runtime == period, ie unlimited time.
  34. */
  35. #define RUNTIME_INF ((u64)~0ULL)
  36. static inline int rt_policy(int policy)
  37. {
  38. if (policy == SCHED_FIFO || policy == SCHED_RR)
  39. return 1;
  40. return 0;
  41. }
  42. static inline int task_has_rt_policy(struct task_struct *p)
  43. {
  44. return rt_policy(p->policy);
  45. }
  46. /*
  47. * This is the priority-queue data structure of the RT scheduling class:
  48. */
  49. struct rt_prio_array {
  50. DECLARE_BITMAP(bitmap, MAX_RT_PRIO+1); /* include 1 bit for delimiter */
  51. struct list_head queue[MAX_RT_PRIO];
  52. };
  53. struct rt_bandwidth {
  54. /* nests inside the rq lock: */
  55. raw_spinlock_t rt_runtime_lock;
  56. ktime_t rt_period;
  57. u64 rt_runtime;
  58. struct hrtimer rt_period_timer;
  59. };
  60. extern struct mutex sched_domains_mutex;
  61. #ifdef CONFIG_CGROUP_SCHED
  62. #include <linux/cgroup.h>
  63. struct cfs_rq;
  64. struct rt_rq;
  65. static LIST_HEAD(task_groups);
  66. struct cfs_bandwidth {
  67. #ifdef CONFIG_CFS_BANDWIDTH
  68. raw_spinlock_t lock;
  69. ktime_t period;
  70. u64 quota, runtime;
  71. s64 hierarchal_quota;
  72. u64 runtime_expires;
  73. int idle, timer_active;
  74. struct hrtimer period_timer, slack_timer;
  75. struct list_head throttled_cfs_rq;
  76. /* statistics */
  77. int nr_periods, nr_throttled;
  78. u64 throttled_time;
  79. #endif
  80. };
  81. /* task group related information */
  82. struct task_group {
  83. struct cgroup_subsys_state css;
  84. #ifdef CONFIG_FAIR_GROUP_SCHED
  85. /* schedulable entities of this group on each cpu */
  86. struct sched_entity **se;
  87. /* runqueue "owned" by this group on each cpu */
  88. struct cfs_rq **cfs_rq;
  89. unsigned long shares;
  90. atomic_t load_weight;
  91. #endif
  92. #ifdef CONFIG_RT_GROUP_SCHED
  93. struct sched_rt_entity **rt_se;
  94. struct rt_rq **rt_rq;
  95. struct rt_bandwidth rt_bandwidth;
  96. #endif
  97. struct rcu_head rcu;
  98. struct list_head list;
  99. struct task_group *parent;
  100. struct list_head siblings;
  101. struct list_head children;
  102. #ifdef CONFIG_SCHED_AUTOGROUP
  103. struct autogroup *autogroup;
  104. #endif
  105. struct cfs_bandwidth cfs_bandwidth;
  106. };
  107. #ifdef CONFIG_FAIR_GROUP_SCHED
  108. #define ROOT_TASK_GROUP_LOAD NICE_0_LOAD
  109. /*
  110. * A weight of 0 or 1 can cause arithmetics problems.
  111. * A weight of a cfs_rq is the sum of weights of which entities
  112. * are queued on this cfs_rq, so a weight of a entity should not be
  113. * too large, so as the shares value of a task group.
  114. * (The default weight is 1024 - so there's no practical
  115. * limitation from this.)
  116. */
  117. #define MIN_SHARES (1UL << 1)
  118. #define MAX_SHARES (1UL << 18)
  119. #endif
  120. /* Default task group.
  121. * Every task in system belong to this group at bootup.
  122. */
  123. extern struct task_group root_task_group;
  124. typedef int (*tg_visitor)(struct task_group *, void *);
  125. extern int walk_tg_tree_from(struct task_group *from,
  126. tg_visitor down, tg_visitor up, void *data);
  127. /*
  128. * Iterate the full tree, calling @down when first entering a node and @up when
  129. * leaving it for the final time.
  130. *
  131. * Caller must hold rcu_lock or sufficient equivalent.
  132. */
  133. static inline int walk_tg_tree(tg_visitor down, tg_visitor up, void *data)
  134. {
  135. return walk_tg_tree_from(&root_task_group, down, up, data);
  136. }
  137. extern int tg_nop(struct task_group *tg, void *data);
  138. extern void free_fair_sched_group(struct task_group *tg);
  139. extern int alloc_fair_sched_group(struct task_group *tg, struct task_group *parent);
  140. extern void unregister_fair_sched_group(struct task_group *tg, int cpu);
  141. extern void init_tg_cfs_entry(struct task_group *tg, struct cfs_rq *cfs_rq,
  142. struct sched_entity *se, int cpu,
  143. struct sched_entity *parent);
  144. extern void init_cfs_bandwidth(struct cfs_bandwidth *cfs_b);
  145. extern int sched_group_set_shares(struct task_group *tg, unsigned long shares);
  146. extern void __refill_cfs_bandwidth_runtime(struct cfs_bandwidth *cfs_b);
  147. extern void __start_cfs_bandwidth(struct cfs_bandwidth *cfs_b);
  148. extern void unthrottle_cfs_rq(struct cfs_rq *cfs_rq);
  149. extern void free_rt_sched_group(struct task_group *tg);
  150. extern int alloc_rt_sched_group(struct task_group *tg, struct task_group *parent);
  151. extern void init_tg_rt_entry(struct task_group *tg, struct rt_rq *rt_rq,
  152. struct sched_rt_entity *rt_se, int cpu,
  153. struct sched_rt_entity *parent);
  154. #else /* CONFIG_CGROUP_SCHED */
  155. struct cfs_bandwidth { };
  156. #endif /* CONFIG_CGROUP_SCHED */
  157. /* CFS-related fields in a runqueue */
  158. struct cfs_rq {
  159. struct load_weight load;
  160. unsigned int nr_running, h_nr_running;
  161. u64 exec_clock;
  162. u64 min_vruntime;
  163. #ifndef CONFIG_64BIT
  164. u64 min_vruntime_copy;
  165. #endif
  166. struct rb_root tasks_timeline;
  167. struct rb_node *rb_leftmost;
  168. /*
  169. * 'curr' points to currently running entity on this cfs_rq.
  170. * It is set to NULL otherwise (i.e when none are currently running).
  171. */
  172. struct sched_entity *curr, *next, *last, *skip;
  173. #ifdef CONFIG_SCHED_DEBUG
  174. unsigned int nr_spread_over;
  175. #endif
  176. #ifdef CONFIG_FAIR_GROUP_SCHED
  177. struct rq *rq; /* cpu runqueue to which this cfs_rq is attached */
  178. /*
  179. * leaf cfs_rqs are those that hold tasks (lowest schedulable entity in
  180. * a hierarchy). Non-leaf lrqs hold other higher schedulable entities
  181. * (like users, containers etc.)
  182. *
  183. * leaf_cfs_rq_list ties together list of leaf cfs_rq's in a cpu. This
  184. * list is used during load balance.
  185. */
  186. int on_list;
  187. struct list_head leaf_cfs_rq_list;
  188. struct task_group *tg; /* group that "owns" this runqueue */
  189. #ifdef CONFIG_SMP
  190. /*
  191. * h_load = weight * f(tg)
  192. *
  193. * Where f(tg) is the recursive weight fraction assigned to
  194. * this group.
  195. */
  196. unsigned long h_load;
  197. /*
  198. * Maintaining per-cpu shares distribution for group scheduling
  199. *
  200. * load_stamp is the last time we updated the load average
  201. * load_last is the last time we updated the load average and saw load
  202. * load_unacc_exec_time is currently unaccounted execution time
  203. */
  204. u64 load_avg;
  205. u64 load_period;
  206. u64 load_stamp, load_last, load_unacc_exec_time;
  207. unsigned long load_contribution;
  208. #endif /* CONFIG_SMP */
  209. #ifdef CONFIG_CFS_BANDWIDTH
  210. int runtime_enabled;
  211. u64 runtime_expires;
  212. s64 runtime_remaining;
  213. u64 throttled_timestamp;
  214. int throttled, throttle_count;
  215. struct list_head throttled_list;
  216. #endif /* CONFIG_CFS_BANDWIDTH */
  217. #endif /* CONFIG_FAIR_GROUP_SCHED */
  218. };
  219. static inline int rt_bandwidth_enabled(void)
  220. {
  221. return sysctl_sched_rt_runtime >= 0;
  222. }
  223. /* Real-Time classes' related field in a runqueue: */
  224. struct rt_rq {
  225. struct rt_prio_array active;
  226. unsigned int rt_nr_running;
  227. #if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
  228. struct {
  229. int curr; /* highest queued rt task prio */
  230. #ifdef CONFIG_SMP
  231. int next; /* next highest */
  232. #endif
  233. } highest_prio;
  234. #endif
  235. #ifdef CONFIG_SMP
  236. unsigned long rt_nr_migratory;
  237. unsigned long rt_nr_total;
  238. int overloaded;
  239. struct plist_head pushable_tasks;
  240. #endif
  241. int rt_throttled;
  242. u64 rt_time;
  243. u64 rt_runtime;
  244. /* Nests inside the rq lock: */
  245. raw_spinlock_t rt_runtime_lock;
  246. #ifdef CONFIG_RT_GROUP_SCHED
  247. unsigned long rt_nr_boosted;
  248. struct rq *rq;
  249. struct list_head leaf_rt_rq_list;
  250. struct task_group *tg;
  251. #endif
  252. };
  253. #ifdef CONFIG_SMP
  254. /*
  255. * We add the notion of a root-domain which will be used to define per-domain
  256. * variables. Each exclusive cpuset essentially defines an island domain by
  257. * fully partitioning the member cpus from any other cpuset. Whenever a new
  258. * exclusive cpuset is created, we also create and attach a new root-domain
  259. * object.
  260. *
  261. */
  262. struct root_domain {
  263. atomic_t refcount;
  264. atomic_t rto_count;
  265. struct rcu_head rcu;
  266. cpumask_var_t span;
  267. cpumask_var_t online;
  268. /*
  269. * The "RT overload" flag: it gets set if a CPU has more than
  270. * one runnable RT task.
  271. */
  272. cpumask_var_t rto_mask;
  273. struct cpupri cpupri;
  274. };
  275. extern struct root_domain def_root_domain;
  276. #endif /* CONFIG_SMP */
  277. /*
  278. * This is the main, per-CPU runqueue data structure.
  279. *
  280. * Locking rule: those places that want to lock multiple runqueues
  281. * (such as the load balancing or the thread migration code), lock
  282. * acquire operations must be ordered by ascending &runqueue.
  283. */
  284. struct rq {
  285. /* runqueue lock: */
  286. raw_spinlock_t lock;
  287. /*
  288. * nr_running and cpu_load should be in the same cacheline because
  289. * remote CPUs use both these fields when doing load calculation.
  290. */
  291. unsigned int nr_running;
  292. #define CPU_LOAD_IDX_MAX 5
  293. unsigned long cpu_load[CPU_LOAD_IDX_MAX];
  294. unsigned long last_load_update_tick;
  295. #ifdef CONFIG_NO_HZ
  296. u64 nohz_stamp;
  297. unsigned long nohz_flags;
  298. #endif
  299. int skip_clock_update;
  300. /* capture load from *all* tasks on this cpu: */
  301. struct load_weight load;
  302. unsigned long nr_load_updates;
  303. u64 nr_switches;
  304. struct cfs_rq cfs;
  305. struct rt_rq rt;
  306. #ifdef CONFIG_FAIR_GROUP_SCHED
  307. /* list of leaf cfs_rq on this cpu: */
  308. struct list_head leaf_cfs_rq_list;
  309. #endif
  310. #ifdef CONFIG_RT_GROUP_SCHED
  311. struct list_head leaf_rt_rq_list;
  312. #endif
  313. /*
  314. * This is part of a global counter where only the total sum
  315. * over all CPUs matters. A task can increase this counter on
  316. * one CPU and if it got migrated afterwards it may decrease
  317. * it on another CPU. Always updated under the runqueue lock:
  318. */
  319. unsigned long nr_uninterruptible;
  320. struct task_struct *curr, *idle, *stop;
  321. unsigned long next_balance;
  322. struct mm_struct *prev_mm;
  323. u64 clock;
  324. u64 clock_task;
  325. atomic_t nr_iowait;
  326. #ifdef CONFIG_SMP
  327. struct root_domain *rd;
  328. struct sched_domain *sd;
  329. unsigned long cpu_power;
  330. unsigned char idle_balance;
  331. /* For active balancing */
  332. int post_schedule;
  333. int active_balance;
  334. int push_cpu;
  335. struct cpu_stop_work active_balance_work;
  336. /* cpu of this runqueue: */
  337. int cpu;
  338. int online;
  339. struct list_head cfs_tasks;
  340. u64 rt_avg;
  341. u64 age_stamp;
  342. u64 idle_stamp;
  343. u64 avg_idle;
  344. #endif
  345. #ifdef CONFIG_IRQ_TIME_ACCOUNTING
  346. u64 prev_irq_time;
  347. #endif
  348. #ifdef CONFIG_PARAVIRT
  349. u64 prev_steal_time;
  350. #endif
  351. #ifdef CONFIG_PARAVIRT_TIME_ACCOUNTING
  352. u64 prev_steal_time_rq;
  353. #endif
  354. /* calc_load related fields */
  355. unsigned long calc_load_update;
  356. long calc_load_active;
  357. #ifdef CONFIG_SCHED_HRTICK
  358. #ifdef CONFIG_SMP
  359. int hrtick_csd_pending;
  360. struct call_single_data hrtick_csd;
  361. #endif
  362. struct hrtimer hrtick_timer;
  363. #endif
  364. #ifdef CONFIG_SCHEDSTATS
  365. /* latency stats */
  366. struct sched_info rq_sched_info;
  367. unsigned long long rq_cpu_time;
  368. /* could above be rq->cfs_rq.exec_clock + rq->rt_rq.rt_runtime ? */
  369. /* sys_sched_yield() stats */
  370. unsigned int yld_count;
  371. /* schedule() stats */
  372. unsigned int sched_count;
  373. unsigned int sched_goidle;
  374. /* try_to_wake_up() stats */
  375. unsigned int ttwu_count;
  376. unsigned int ttwu_local;
  377. #endif
  378. #ifdef CONFIG_SMP
  379. struct llist_head wake_list;
  380. #endif
  381. };
  382. static inline int cpu_of(struct rq *rq)
  383. {
  384. #ifdef CONFIG_SMP
  385. return rq->cpu;
  386. #else
  387. return 0;
  388. #endif
  389. }
  390. DECLARE_PER_CPU(struct rq, runqueues);
  391. #define cpu_rq(cpu) (&per_cpu(runqueues, (cpu)))
  392. #define this_rq() (&__get_cpu_var(runqueues))
  393. #define task_rq(p) cpu_rq(task_cpu(p))
  394. #define cpu_curr(cpu) (cpu_rq(cpu)->curr)
  395. #define raw_rq() (&__raw_get_cpu_var(runqueues))
  396. #ifdef CONFIG_SMP
  397. #define rcu_dereference_check_sched_domain(p) \
  398. rcu_dereference_check((p), \
  399. lockdep_is_held(&sched_domains_mutex))
  400. /*
  401. * The domain tree (rq->sd) is protected by RCU's quiescent state transition.
  402. * See detach_destroy_domains: synchronize_sched for details.
  403. *
  404. * The domain tree of any CPU may only be accessed from within
  405. * preempt-disabled sections.
  406. */
  407. #define for_each_domain(cpu, __sd) \
  408. for (__sd = rcu_dereference_check_sched_domain(cpu_rq(cpu)->sd); \
  409. __sd; __sd = __sd->parent)
  410. #define for_each_lower_domain(sd) for (; sd; sd = sd->child)
  411. /**
  412. * highest_flag_domain - Return highest sched_domain containing flag.
  413. * @cpu: The cpu whose highest level of sched domain is to
  414. * be returned.
  415. * @flag: The flag to check for the highest sched_domain
  416. * for the given cpu.
  417. *
  418. * Returns the highest sched_domain of a cpu which contains the given flag.
  419. */
  420. static inline struct sched_domain *highest_flag_domain(int cpu, int flag)
  421. {
  422. struct sched_domain *sd, *hsd = NULL;
  423. for_each_domain(cpu, sd) {
  424. if (!(sd->flags & flag))
  425. break;
  426. hsd = sd;
  427. }
  428. return hsd;
  429. }
  430. DECLARE_PER_CPU(struct sched_domain *, sd_llc);
  431. DECLARE_PER_CPU(int, sd_llc_id);
  432. extern int group_balance_cpu(struct sched_group *sg);
  433. #endif /* CONFIG_SMP */
  434. #include "stats.h"
  435. #include "auto_group.h"
  436. #ifdef CONFIG_CGROUP_SCHED
  437. /*
  438. * Return the group to which this tasks belongs.
  439. *
  440. * We cannot use task_subsys_state() and friends because the cgroup
  441. * subsystem changes that value before the cgroup_subsys::attach() method
  442. * is called, therefore we cannot pin it and might observe the wrong value.
  443. *
  444. * The same is true for autogroup's p->signal->autogroup->tg, the autogroup
  445. * core changes this before calling sched_move_task().
  446. *
  447. * Instead we use a 'copy' which is updated from sched_move_task() while
  448. * holding both task_struct::pi_lock and rq::lock.
  449. */
  450. static inline struct task_group *task_group(struct task_struct *p)
  451. {
  452. return p->sched_task_group;
  453. }
  454. /* Change a task's cfs_rq and parent entity if it moves across CPUs/groups */
  455. static inline void set_task_rq(struct task_struct *p, unsigned int cpu)
  456. {
  457. #if defined(CONFIG_FAIR_GROUP_SCHED) || defined(CONFIG_RT_GROUP_SCHED)
  458. struct task_group *tg = task_group(p);
  459. #endif
  460. #ifdef CONFIG_FAIR_GROUP_SCHED
  461. p->se.cfs_rq = tg->cfs_rq[cpu];
  462. p->se.parent = tg->se[cpu];
  463. #endif
  464. #ifdef CONFIG_RT_GROUP_SCHED
  465. p->rt.rt_rq = tg->rt_rq[cpu];
  466. p->rt.parent = tg->rt_se[cpu];
  467. #endif
  468. }
  469. #else /* CONFIG_CGROUP_SCHED */
  470. static inline void set_task_rq(struct task_struct *p, unsigned int cpu) { }
  471. static inline struct task_group *task_group(struct task_struct *p)
  472. {
  473. return NULL;
  474. }
  475. #endif /* CONFIG_CGROUP_SCHED */
  476. static inline void __set_task_cpu(struct task_struct *p, unsigned int cpu)
  477. {
  478. set_task_rq(p, cpu);
  479. #ifdef CONFIG_SMP
  480. /*
  481. * After ->cpu is set up to a new value, task_rq_lock(p, ...) can be
  482. * successfuly executed on another CPU. We must ensure that updates of
  483. * per-task data have been completed by this moment.
  484. */
  485. smp_wmb();
  486. task_thread_info(p)->cpu = cpu;
  487. #endif
  488. }
  489. /*
  490. * Tunables that become constants when CONFIG_SCHED_DEBUG is off:
  491. */
  492. #ifdef CONFIG_SCHED_DEBUG
  493. # include <linux/static_key.h>
  494. # define const_debug __read_mostly
  495. #else
  496. # define const_debug const
  497. #endif
  498. extern const_debug unsigned int sysctl_sched_features;
  499. #define SCHED_FEAT(name, enabled) \
  500. __SCHED_FEAT_##name ,
  501. enum {
  502. #include "features.h"
  503. __SCHED_FEAT_NR,
  504. };
  505. #undef SCHED_FEAT
  506. #if defined(CONFIG_SCHED_DEBUG) && defined(HAVE_JUMP_LABEL)
  507. static __always_inline bool static_branch__true(struct static_key *key)
  508. {
  509. return static_key_true(key); /* Not out of line branch. */
  510. }
  511. static __always_inline bool static_branch__false(struct static_key *key)
  512. {
  513. return static_key_false(key); /* Out of line branch. */
  514. }
  515. #define SCHED_FEAT(name, enabled) \
  516. static __always_inline bool static_branch_##name(struct static_key *key) \
  517. { \
  518. return static_branch__##enabled(key); \
  519. }
  520. #include "features.h"
  521. #undef SCHED_FEAT
  522. extern struct static_key sched_feat_keys[__SCHED_FEAT_NR];
  523. #define sched_feat(x) (static_branch_##x(&sched_feat_keys[__SCHED_FEAT_##x]))
  524. #else /* !(SCHED_DEBUG && HAVE_JUMP_LABEL) */
  525. #define sched_feat(x) (sysctl_sched_features & (1UL << __SCHED_FEAT_##x))
  526. #endif /* SCHED_DEBUG && HAVE_JUMP_LABEL */
  527. static inline u64 global_rt_period(void)
  528. {
  529. return (u64)sysctl_sched_rt_period * NSEC_PER_USEC;
  530. }
  531. static inline u64 global_rt_runtime(void)
  532. {
  533. if (sysctl_sched_rt_runtime < 0)
  534. return RUNTIME_INF;
  535. return (u64)sysctl_sched_rt_runtime * NSEC_PER_USEC;
  536. }
  537. static inline int task_current(struct rq *rq, struct task_struct *p)
  538. {
  539. return rq->curr == p;
  540. }
  541. static inline int task_running(struct rq *rq, struct task_struct *p)
  542. {
  543. #ifdef CONFIG_SMP
  544. return p->on_cpu;
  545. #else
  546. return task_current(rq, p);
  547. #endif
  548. }
  549. #ifndef prepare_arch_switch
  550. # define prepare_arch_switch(next) do { } while (0)
  551. #endif
  552. #ifndef finish_arch_switch
  553. # define finish_arch_switch(prev) do { } while (0)
  554. #endif
  555. #ifndef finish_arch_post_lock_switch
  556. # define finish_arch_post_lock_switch() do { } while (0)
  557. #endif
  558. #ifndef __ARCH_WANT_UNLOCKED_CTXSW
  559. static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
  560. {
  561. #ifdef CONFIG_SMP
  562. /*
  563. * We can optimise this out completely for !SMP, because the
  564. * SMP rebalancing from interrupt is the only thing that cares
  565. * here.
  566. */
  567. next->on_cpu = 1;
  568. #endif
  569. }
  570. static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
  571. {
  572. #ifdef CONFIG_SMP
  573. /*
  574. * After ->on_cpu is cleared, the task can be moved to a different CPU.
  575. * We must ensure this doesn't happen until the switch is completely
  576. * finished.
  577. */
  578. smp_wmb();
  579. prev->on_cpu = 0;
  580. #endif
  581. #ifdef CONFIG_DEBUG_SPINLOCK
  582. /* this is a valid case when another task releases the spinlock */
  583. rq->lock.owner = current;
  584. #endif
  585. /*
  586. * If we are tracking spinlock dependencies then we have to
  587. * fix up the runqueue lock - which gets 'carried over' from
  588. * prev into current:
  589. */
  590. spin_acquire(&rq->lock.dep_map, 0, 0, _THIS_IP_);
  591. raw_spin_unlock_irq(&rq->lock);
  592. }
  593. #else /* __ARCH_WANT_UNLOCKED_CTXSW */
  594. static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
  595. {
  596. #ifdef CONFIG_SMP
  597. /*
  598. * We can optimise this out completely for !SMP, because the
  599. * SMP rebalancing from interrupt is the only thing that cares
  600. * here.
  601. */
  602. next->on_cpu = 1;
  603. #endif
  604. #ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW
  605. raw_spin_unlock_irq(&rq->lock);
  606. #else
  607. raw_spin_unlock(&rq->lock);
  608. #endif
  609. }
  610. static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
  611. {
  612. #ifdef CONFIG_SMP
  613. /*
  614. * After ->on_cpu is cleared, the task can be moved to a different CPU.
  615. * We must ensure this doesn't happen until the switch is completely
  616. * finished.
  617. */
  618. smp_wmb();
  619. prev->on_cpu = 0;
  620. #endif
  621. #ifndef __ARCH_WANT_INTERRUPTS_ON_CTXSW
  622. local_irq_enable();
  623. #endif
  624. }
  625. #endif /* __ARCH_WANT_UNLOCKED_CTXSW */
  626. static inline void update_load_add(struct load_weight *lw, unsigned long inc)
  627. {
  628. lw->weight += inc;
  629. lw->inv_weight = 0;
  630. }
  631. static inline void update_load_sub(struct load_weight *lw, unsigned long dec)
  632. {
  633. lw->weight -= dec;
  634. lw->inv_weight = 0;
  635. }
  636. static inline void update_load_set(struct load_weight *lw, unsigned long w)
  637. {
  638. lw->weight = w;
  639. lw->inv_weight = 0;
  640. }
  641. /*
  642. * To aid in avoiding the subversion of "niceness" due to uneven distribution
  643. * of tasks with abnormal "nice" values across CPUs the contribution that
  644. * each task makes to its run queue's load is weighted according to its
  645. * scheduling class and "nice" value. For SCHED_NORMAL tasks this is just a
  646. * scaled version of the new time slice allocation that they receive on time
  647. * slice expiry etc.
  648. */
  649. #define WEIGHT_IDLEPRIO 3
  650. #define WMULT_IDLEPRIO 1431655765
  651. /*
  652. * Nice levels are multiplicative, with a gentle 10% change for every
  653. * nice level changed. I.e. when a CPU-bound task goes from nice 0 to
  654. * nice 1, it will get ~10% less CPU time than another CPU-bound task
  655. * that remained on nice 0.
  656. *
  657. * The "10% effect" is relative and cumulative: from _any_ nice level,
  658. * if you go up 1 level, it's -10% CPU usage, if you go down 1 level
  659. * it's +10% CPU usage. (to achieve that we use a multiplier of 1.25.
  660. * If a task goes up by ~10% and another task goes down by ~10% then
  661. * the relative distance between them is ~25%.)
  662. */
  663. static const int prio_to_weight[40] = {
  664. /* -20 */ 88761, 71755, 56483, 46273, 36291,
  665. /* -15 */ 29154, 23254, 18705, 14949, 11916,
  666. /* -10 */ 9548, 7620, 6100, 4904, 3906,
  667. /* -5 */ 3121, 2501, 1991, 1586, 1277,
  668. /* 0 */ 1024, 820, 655, 526, 423,
  669. /* 5 */ 335, 272, 215, 172, 137,
  670. /* 10 */ 110, 87, 70, 56, 45,
  671. /* 15 */ 36, 29, 23, 18, 15,
  672. };
  673. /*
  674. * Inverse (2^32/x) values of the prio_to_weight[] array, precalculated.
  675. *
  676. * In cases where the weight does not change often, we can use the
  677. * precalculated inverse to speed up arithmetics by turning divisions
  678. * into multiplications:
  679. */
  680. static const u32 prio_to_wmult[40] = {
  681. /* -20 */ 48388, 59856, 76040, 92818, 118348,
  682. /* -15 */ 147320, 184698, 229616, 287308, 360437,
  683. /* -10 */ 449829, 563644, 704093, 875809, 1099582,
  684. /* -5 */ 1376151, 1717300, 2157191, 2708050, 3363326,
  685. /* 0 */ 4194304, 5237765, 6557202, 8165337, 10153587,
  686. /* 5 */ 12820798, 15790321, 19976592, 24970740, 31350126,
  687. /* 10 */ 39045157, 49367440, 61356676, 76695844, 95443717,
  688. /* 15 */ 119304647, 148102320, 186737708, 238609294, 286331153,
  689. };
  690. /* Time spent by the tasks of the cpu accounting group executing in ... */
  691. enum cpuacct_stat_index {
  692. CPUACCT_STAT_USER, /* ... user mode */
  693. CPUACCT_STAT_SYSTEM, /* ... kernel mode */
  694. CPUACCT_STAT_NSTATS,
  695. };
  696. #define sched_class_highest (&stop_sched_class)
  697. #define for_each_class(class) \
  698. for (class = sched_class_highest; class; class = class->next)
  699. extern const struct sched_class stop_sched_class;
  700. extern const struct sched_class rt_sched_class;
  701. extern const struct sched_class fair_sched_class;
  702. extern const struct sched_class idle_sched_class;
  703. #ifdef CONFIG_SMP
  704. extern void trigger_load_balance(struct rq *rq, int cpu);
  705. extern void idle_balance(int this_cpu, struct rq *this_rq);
  706. #else /* CONFIG_SMP */
  707. static inline void idle_balance(int cpu, struct rq *rq)
  708. {
  709. }
  710. #endif
  711. extern void sysrq_sched_debug_show(void);
  712. extern void sched_init_granularity(void);
  713. extern void update_max_interval(void);
  714. extern void update_group_power(struct sched_domain *sd, int cpu);
  715. extern int update_runtime(struct notifier_block *nfb, unsigned long action, void *hcpu);
  716. extern void init_sched_rt_class(void);
  717. extern void init_sched_fair_class(void);
  718. extern void resched_task(struct task_struct *p);
  719. extern void resched_cpu(int cpu);
  720. extern struct rt_bandwidth def_rt_bandwidth;
  721. extern void init_rt_bandwidth(struct rt_bandwidth *rt_b, u64 period, u64 runtime);
  722. extern void update_idle_cpu_load(struct rq *this_rq);
  723. #ifdef CONFIG_CGROUP_CPUACCT
  724. #include <linux/cgroup.h>
  725. /* track cpu usage of a group of tasks and its child groups */
  726. struct cpuacct {
  727. struct cgroup_subsys_state css;
  728. /* cpuusage holds pointer to a u64-type object on every cpu */
  729. u64 __percpu *cpuusage;
  730. struct kernel_cpustat __percpu *cpustat;
  731. };
  732. /* return cpu accounting group corresponding to this container */
  733. static inline struct cpuacct *cgroup_ca(struct cgroup *cgrp)
  734. {
  735. return container_of(cgroup_subsys_state(cgrp, cpuacct_subsys_id),
  736. struct cpuacct, css);
  737. }
  738. /* return cpu accounting group to which this task belongs */
  739. static inline struct cpuacct *task_ca(struct task_struct *tsk)
  740. {
  741. return container_of(task_subsys_state(tsk, cpuacct_subsys_id),
  742. struct cpuacct, css);
  743. }
  744. static inline struct cpuacct *parent_ca(struct cpuacct *ca)
  745. {
  746. if (!ca || !ca->css.cgroup->parent)
  747. return NULL;
  748. return cgroup_ca(ca->css.cgroup->parent);
  749. }
  750. extern void cpuacct_charge(struct task_struct *tsk, u64 cputime);
  751. #else
  752. static inline void cpuacct_charge(struct task_struct *tsk, u64 cputime) {}
  753. #endif
  754. static inline void inc_nr_running(struct rq *rq)
  755. {
  756. rq->nr_running++;
  757. }
  758. static inline void dec_nr_running(struct rq *rq)
  759. {
  760. rq->nr_running--;
  761. }
  762. extern void update_rq_clock(struct rq *rq);
  763. extern void activate_task(struct rq *rq, struct task_struct *p, int flags);
  764. extern void deactivate_task(struct rq *rq, struct task_struct *p, int flags);
  765. extern void check_preempt_curr(struct rq *rq, struct task_struct *p, int flags);
  766. extern const_debug unsigned int sysctl_sched_time_avg;
  767. extern const_debug unsigned int sysctl_sched_nr_migrate;
  768. extern const_debug unsigned int sysctl_sched_migration_cost;
  769. static inline u64 sched_avg_period(void)
  770. {
  771. return (u64)sysctl_sched_time_avg * NSEC_PER_MSEC / 2;
  772. }
  773. #ifdef CONFIG_SCHED_HRTICK
  774. /*
  775. * Use hrtick when:
  776. * - enabled by features
  777. * - hrtimer is actually high res
  778. */
  779. static inline int hrtick_enabled(struct rq *rq)
  780. {
  781. if (!sched_feat(HRTICK))
  782. return 0;
  783. if (!cpu_active(cpu_of(rq)))
  784. return 0;
  785. return hrtimer_is_hres_active(&rq->hrtick_timer);
  786. }
  787. void hrtick_start(struct rq *rq, u64 delay);
  788. #else
  789. static inline int hrtick_enabled(struct rq *rq)
  790. {
  791. return 0;
  792. }
  793. #endif /* CONFIG_SCHED_HRTICK */
  794. #ifdef CONFIG_SMP
  795. extern void sched_avg_update(struct rq *rq);
  796. static inline void sched_rt_avg_update(struct rq *rq, u64 rt_delta)
  797. {
  798. rq->rt_avg += rt_delta;
  799. sched_avg_update(rq);
  800. }
  801. #else
  802. static inline void sched_rt_avg_update(struct rq *rq, u64 rt_delta) { }
  803. static inline void sched_avg_update(struct rq *rq) { }
  804. #endif
  805. extern void start_bandwidth_timer(struct hrtimer *period_timer, ktime_t period);
  806. #ifdef CONFIG_SMP
  807. #ifdef CONFIG_PREEMPT
  808. static inline void double_rq_lock(struct rq *rq1, struct rq *rq2);
  809. /*
  810. * fair double_lock_balance: Safely acquires both rq->locks in a fair
  811. * way at the expense of forcing extra atomic operations in all
  812. * invocations. This assures that the double_lock is acquired using the
  813. * same underlying policy as the spinlock_t on this architecture, which
  814. * reduces latency compared to the unfair variant below. However, it
  815. * also adds more overhead and therefore may reduce throughput.
  816. */
  817. static inline int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
  818. __releases(this_rq->lock)
  819. __acquires(busiest->lock)
  820. __acquires(this_rq->lock)
  821. {
  822. raw_spin_unlock(&this_rq->lock);
  823. double_rq_lock(this_rq, busiest);
  824. return 1;
  825. }
  826. #else
  827. /*
  828. * Unfair double_lock_balance: Optimizes throughput at the expense of
  829. * latency by eliminating extra atomic operations when the locks are
  830. * already in proper order on entry. This favors lower cpu-ids and will
  831. * grant the double lock to lower cpus over higher ids under contention,
  832. * regardless of entry order into the function.
  833. */
  834. static inline int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
  835. __releases(this_rq->lock)
  836. __acquires(busiest->lock)
  837. __acquires(this_rq->lock)
  838. {
  839. int ret = 0;
  840. if (unlikely(!raw_spin_trylock(&busiest->lock))) {
  841. if (busiest < this_rq) {
  842. raw_spin_unlock(&this_rq->lock);
  843. raw_spin_lock(&busiest->lock);
  844. raw_spin_lock_nested(&this_rq->lock,
  845. SINGLE_DEPTH_NESTING);
  846. ret = 1;
  847. } else
  848. raw_spin_lock_nested(&busiest->lock,
  849. SINGLE_DEPTH_NESTING);
  850. }
  851. return ret;
  852. }
  853. #endif /* CONFIG_PREEMPT */
  854. /*
  855. * double_lock_balance - lock the busiest runqueue, this_rq is locked already.
  856. */
  857. static inline int double_lock_balance(struct rq *this_rq, struct rq *busiest)
  858. {
  859. if (unlikely(!irqs_disabled())) {
  860. /* printk() doesn't work good under rq->lock */
  861. raw_spin_unlock(&this_rq->lock);
  862. BUG_ON(1);
  863. }
  864. return _double_lock_balance(this_rq, busiest);
  865. }
  866. static inline void double_unlock_balance(struct rq *this_rq, struct rq *busiest)
  867. __releases(busiest->lock)
  868. {
  869. raw_spin_unlock(&busiest->lock);
  870. lock_set_subclass(&this_rq->lock.dep_map, 0, _RET_IP_);
  871. }
  872. /*
  873. * double_rq_lock - safely lock two runqueues
  874. *
  875. * Note this does not disable interrupts like task_rq_lock,
  876. * you need to do so manually before calling.
  877. */
  878. static inline void double_rq_lock(struct rq *rq1, struct rq *rq2)
  879. __acquires(rq1->lock)
  880. __acquires(rq2->lock)
  881. {
  882. BUG_ON(!irqs_disabled());
  883. if (rq1 == rq2) {
  884. raw_spin_lock(&rq1->lock);
  885. __acquire(rq2->lock); /* Fake it out ;) */
  886. } else {
  887. if (rq1 < rq2) {
  888. raw_spin_lock(&rq1->lock);
  889. raw_spin_lock_nested(&rq2->lock, SINGLE_DEPTH_NESTING);
  890. } else {
  891. raw_spin_lock(&rq2->lock);
  892. raw_spin_lock_nested(&rq1->lock, SINGLE_DEPTH_NESTING);
  893. }
  894. }
  895. }
  896. /*
  897. * double_rq_unlock - safely unlock two runqueues
  898. *
  899. * Note this does not restore interrupts like task_rq_unlock,
  900. * you need to do so manually after calling.
  901. */
  902. static inline void double_rq_unlock(struct rq *rq1, struct rq *rq2)
  903. __releases(rq1->lock)
  904. __releases(rq2->lock)
  905. {
  906. raw_spin_unlock(&rq1->lock);
  907. if (rq1 != rq2)
  908. raw_spin_unlock(&rq2->lock);
  909. else
  910. __release(rq2->lock);
  911. }
  912. #else /* CONFIG_SMP */
  913. /*
  914. * double_rq_lock - safely lock two runqueues
  915. *
  916. * Note this does not disable interrupts like task_rq_lock,
  917. * you need to do so manually before calling.
  918. */
  919. static inline void double_rq_lock(struct rq *rq1, struct rq *rq2)
  920. __acquires(rq1->lock)
  921. __acquires(rq2->lock)
  922. {
  923. BUG_ON(!irqs_disabled());
  924. BUG_ON(rq1 != rq2);
  925. raw_spin_lock(&rq1->lock);
  926. __acquire(rq2->lock); /* Fake it out ;) */
  927. }
  928. /*
  929. * double_rq_unlock - safely unlock two runqueues
  930. *
  931. * Note this does not restore interrupts like task_rq_unlock,
  932. * you need to do so manually after calling.
  933. */
  934. static inline void double_rq_unlock(struct rq *rq1, struct rq *rq2)
  935. __releases(rq1->lock)
  936. __releases(rq2->lock)
  937. {
  938. BUG_ON(rq1 != rq2);
  939. raw_spin_unlock(&rq1->lock);
  940. __release(rq2->lock);
  941. }
  942. #endif
  943. extern struct sched_entity *__pick_first_entity(struct cfs_rq *cfs_rq);
  944. extern struct sched_entity *__pick_last_entity(struct cfs_rq *cfs_rq);
  945. extern void print_cfs_stats(struct seq_file *m, int cpu);
  946. extern void print_rt_stats(struct seq_file *m, int cpu);
  947. extern void init_cfs_rq(struct cfs_rq *cfs_rq);
  948. extern void init_rt_rq(struct rt_rq *rt_rq, struct rq *rq);
  949. extern void unthrottle_offline_cfs_rqs(struct rq *rq);
  950. extern void account_cfs_bandwidth_used(int enabled, int was_enabled);
  951. #ifdef CONFIG_NO_HZ
  952. enum rq_nohz_flag_bits {
  953. NOHZ_TICK_STOPPED,
  954. NOHZ_BALANCE_KICK,
  955. NOHZ_IDLE,
  956. };
  957. #define nohz_flags(cpu) (&cpu_rq(cpu)->nohz_flags)
  958. #endif