kernel/sched/sched.h

   1
   2 #include <linux/sched.h>
   3 #include <linux/sched/sysctl.h>
   4 #include <linux/sched/rt.h>
   5 #include <linux/mutex.h>
   6 #include <linux/spinlock.h>
   7 #include <linux/stop_machine.h>
   8 #include <linux/tick.h>
   9
  10 #include "cpupri.h"
  11 #include "cpuacct.h"
  12
  13 extern __read_mostly int scheduler_running;
  14
  15 /*
  16  * Convert user-nice values [ -20 ... 0 ... 19 ]
  17  * to static priority [ MAX_RT_PRIO..MAX_PRIO-1 ],
  18  * and back.
  19  */
  20 #define NICE_TO_PRIO(nice)      (MAX_RT_PRIO + (nice) + 20)
  21 #define PRIO_TO_NICE(prio)      ((prio) - MAX_RT_PRIO - 20)
  22 #define TASK_NICE(p)            PRIO_TO_NICE((p)->static_prio)
  23
  24 /*
  25  * 'User priority' is the nice value converted to something we
  26  * can work with better when scaling various scheduler parameters,
  27  * it's a [ 0 ... 39 ] range.
  28  */
  29 #define USER_PRIO(p)            ((p)-MAX_RT_PRIO)
  30 #define TASK_USER_PRIO(p)       USER_PRIO((p)->static_prio)
  31 #define MAX_USER_PRIO           (USER_PRIO(MAX_PRIO))
  32
  33 /*
  34  * Helpers for converting nanosecond timing to jiffy resolution
  35  */
  36 #define NS_TO_JIFFIES(TIME)     ((unsigned long)(TIME) / (NSEC_PER_SEC / HZ))
  37
  38 /*
  39  * Increase resolution of nice-level calculations for 64-bit architectures.
  40  * The extra resolution improves shares distribution and load balancing of
  41  * low-weight task groups (eg. nice +19 on an autogroup), deeper taskgroup
  42  * hierarchies, especially on larger systems. This is not a user-visible change
  43  * and does not change the user-interface for setting shares/weights.
  44  *
  45  * We increase resolution only if we have enough bits to allow this increased
  46  * resolution (i.e. BITS_PER_LONG > 32). The costs for increasing resolution
  47  * when BITS_PER_LONG <= 32 are pretty high and the returns do not justify the
  48  * increased costs.
  49  */
  50 #if 0 /* BITS_PER_LONG > 32 -- currently broken: it increases power usage under light load  */
  51 # define SCHED_LOAD_RESOLUTION  10
  52 # define scale_load(w)          ((w) << SCHED_LOAD_RESOLUTION)
  53 # define scale_load_down(w)     ((w) >> SCHED_LOAD_RESOLUTION)
  54 #else
  55 # define SCHED_LOAD_RESOLUTION  0
  56 # define scale_load(w)          (w)
  57 # define scale_load_down(w)     (w)
  58 #endif
  59
  60 #define SCHED_LOAD_SHIFT        (10 + SCHED_LOAD_RESOLUTION)
  61 #define SCHED_LOAD_SCALE        (1L << SCHED_LOAD_SHIFT)
  62
  63 #define NICE_0_LOAD             SCHED_LOAD_SCALE
  64 #define NICE_0_SHIFT            SCHED_LOAD_SHIFT
  65
  66 /*
  67  * These are the 'tuning knobs' of the scheduler:
  68  */
  69
  70 /*
  71  * single value that denotes runtime == period, ie unlimited time.
  72  */
  73 #define RUNTIME_INF     ((u64)~0ULL)
  74
  75 static inline int rt_policy(int policy)
  76 {
  77         if (policy == SCHED_FIFO || policy == SCHED_RR)
  78                 return 1;
  79         return 0;
  80 }
  81
  82 static inline int task_has_rt_policy(struct task_struct *p)
  83 {
  84         return rt_policy(p->policy);
  85 }
  86
  87 /*
  88  * This is the priority-queue data structure of the RT scheduling class:
  89  */
  90 struct rt_prio_array {
  91         DECLARE_BITMAP(bitmap, MAX_RT_PRIO+1); /* include 1 bit for delimiter */
  92         struct list_head queue[MAX_RT_PRIO];
  93 };
  94
  95 struct rt_bandwidth {
  96         /* nests inside the rq lock: */
  97         raw_spinlock_t          rt_runtime_lock;
  98         ktime_t                 rt_period;
  99         u64                     rt_runtime;
 100         struct hrtimer          rt_period_timer;
 101 };
 102
 103 extern struct mutex sched_domains_mutex;
 104
 105 #ifdef CONFIG_CGROUP_SCHED
 106
 107 #include <linux/cgroup.h>
 108
 109 struct cfs_rq;
 110 struct rt_rq;
 111
 112 extern struct list_head task_groups;
 113
 114 struct cfs_bandwidth {
 115 #ifdef CONFIG_CFS_BANDWIDTH
 116         raw_spinlock_t lock;
 117         ktime_t period;
 118         u64 quota, runtime;
 119         s64 hierarchal_quota;
 120         u64 runtime_expires;
 121
 122         int idle, timer_active;
 123         struct hrtimer period_timer, slack_timer;
 124         struct list_head throttled_cfs_rq;
 125
 126         /* statistics */
 127         int nr_periods, nr_throttled;
 128         u64 throttled_time;
 129 #endif
 130 };
 131
 132 /* task group related information */
 133 struct task_group {
 134         struct cgroup_subsys_state css;
 135
 136 #ifdef CONFIG_FAIR_GROUP_SCHED
 137         /* schedulable entities of this group on each cpu */
 138         struct sched_entity **se;
 139         /* runqueue "owned" by this group on each cpu */
 140         struct cfs_rq **cfs_rq;
 141         unsigned long shares;
 142
 143         atomic_t load_weight;
 144         atomic64_t load_avg;
 145         atomic_t runnable_avg, usage_avg;
 146 #endif
 147
 148 #ifdef CONFIG_RT_GROUP_SCHED
 149         struct sched_rt_entity **rt_se;
 150         struct rt_rq **rt_rq;
 151
 152         struct rt_bandwidth rt_bandwidth;
 153 #endif
 154
 155         struct rcu_head rcu;
 156         struct list_head list;
 157
 158         struct task_group *parent;
 159         struct list_head siblings;
 160         struct list_head children;
 161
 162 #ifdef CONFIG_SCHED_AUTOGROUP
 163         struct autogroup *autogroup;
 164 #endif
 165
 166         struct cfs_bandwidth cfs_bandwidth;
 167 };
 168
 169 #ifdef CONFIG_FAIR_GROUP_SCHED
 170 #define ROOT_TASK_GROUP_LOAD    NICE_0_LOAD
 171
 172 /*
 173  * A weight of 0 or 1 can cause arithmetics problems.
 174  * A weight of a cfs_rq is the sum of weights of which entities
 175  * are queued on this cfs_rq, so a weight of a entity should not be
 176  * too large, so as the shares value of a task group.
 177  * (The default weight is 1024 - so there's no practical
 178  *  limitation from this.)
 179  */
 180 #define MIN_SHARES      (1UL <<  1)
 181 #define MAX_SHARES      (1UL << 18)
 182 #endif
 183
 184 typedef int (*tg_visitor)(struct task_group *, void *);
 185
 186 extern int walk_tg_tree_from(struct task_group *from,
 187                              tg_visitor down, tg_visitor up, void *data);
 188
 189 /*
 190  * Iterate the full tree, calling @down when first entering a node and @up when
 191  * leaving it for the final time.
 192  *
 193  * Caller must hold rcu_lock or sufficient equivalent.
 194  */
 195 static inline int walk_tg_tree(tg_visitor down, tg_visitor up, void *data)
 196 {
 197         return walk_tg_tree_from(&root_task_group, down, up, data);
 198 }
 199
 200 extern int tg_nop(struct task_group *tg, void *data);
 201
 202 extern void free_fair_sched_group(struct task_group *tg);
 203 extern int alloc_fair_sched_group(struct task_group *tg, struct task_group *parent);
 204 extern void unregister_fair_sched_group(struct task_group *tg, int cpu);
 205 extern void init_tg_cfs_entry(struct task_group *tg, struct cfs_rq *cfs_rq,
 206                         struct sched_entity *se, int cpu,
 207                         struct sched_entity *parent);
 208 extern void init_cfs_bandwidth(struct cfs_bandwidth *cfs_b);
 209 extern int sched_group_set_shares(struct task_group *tg, unsigned long shares);
 210
 211 extern void __refill_cfs_bandwidth_runtime(struct cfs_bandwidth *cfs_b);
 212 extern void __start_cfs_bandwidth(struct cfs_bandwidth *cfs_b);
 213 extern void unthrottle_cfs_rq(struct cfs_rq *cfs_rq);
 214
 215 extern void free_rt_sched_group(struct task_group *tg);
 216 extern int alloc_rt_sched_group(struct task_group *tg, struct task_group *parent);
 217 extern void init_tg_rt_entry(struct task_group *tg, struct rt_rq *rt_rq,
 218                 struct sched_rt_entity *rt_se, int cpu,
 219                 struct sched_rt_entity *parent);
 220
 221 extern struct task_group *sched_create_group(struct task_group *parent);
 222 extern void sched_online_group(struct task_group *tg,
 223                                struct task_group *parent);
 224 extern void sched_destroy_group(struct task_group *tg);
 225 extern void sched_offline_group(struct task_group *tg);
 226
 227 extern void sched_move_task(struct task_struct *tsk);
 228
 229 #ifdef CONFIG_FAIR_GROUP_SCHED
 230 extern int sched_group_set_shares(struct task_group *tg, unsigned long shares);
 231 #endif
 232
 233 #else /* CONFIG_CGROUP_SCHED */
 234
 235 struct cfs_bandwidth { };
 236
 237 #endif  /* CONFIG_CGROUP_SCHED */
 238
 239 /* CFS-related fields in a runqueue */
 240 struct cfs_rq {
 241         struct load_weight load;
 242         unsigned int nr_running, h_nr_running;
 243
 244         u64 exec_clock;
 245         u64 min_vruntime;
 246 #ifndef CONFIG_64BIT
 247         u64 min_vruntime_copy;
 248 #endif
 249
 250         struct rb_root tasks_timeline;
 251         struct rb_node *rb_leftmost;
 252
 253         /*
 254          * 'curr' points to currently running entity on this cfs_rq.
 255          * It is set to NULL otherwise (i.e when none are currently running).
 256          */
 257         struct sched_entity *curr, *next, *last, *skip;
 258
 259 #ifdef  CONFIG_SCHED_DEBUG
 260         unsigned int nr_spread_over;
 261 #endif
 262
 263 #ifdef CONFIG_SMP
 264 /*
 265  * Load-tracking only depends on SMP, FAIR_GROUP_SCHED dependency below may be
 266  * removed when useful for applications beyond shares distribution (e.g.
 267  * load-balance).
 268  */
 269 #ifdef CONFIG_FAIR_GROUP_SCHED
 270         /*
 271          * CFS Load tracking
 272          * Under CFS, load is tracked on a per-entity basis and aggregated up.
 273          * This allows for the description of both thread and group usage (in
 274          * the FAIR_GROUP_SCHED case).
 275          */
 276         u64 runnable_load_avg, blocked_load_avg;
 277         atomic64_t decay_counter, removed_load;
 278         u64 last_decay;
 279 #endif /* CONFIG_FAIR_GROUP_SCHED */
 280 /* These always depend on CONFIG_FAIR_GROUP_SCHED */
 281 #ifdef CONFIG_FAIR_GROUP_SCHED
 282         u32 tg_runnable_contrib, tg_usage_contrib;
 283         u64 tg_load_contrib;
 284 #endif /* CONFIG_FAIR_GROUP_SCHED */
 285
 286         /*
 287          *   h_load = weight * f(tg)
 288          *
 289          * Where f(tg) is the recursive weight fraction assigned to
 290          * this group.
 291          */
 292         unsigned long h_load;
 293 #endif /* CONFIG_SMP */
 294
 295 #ifdef CONFIG_FAIR_GROUP_SCHED
 296         struct rq *rq;  /* cpu runqueue to which this cfs_rq is attached */
 297
 298         /*
 299          * leaf cfs_rqs are those that hold tasks (lowest schedulable entity in
 300          * a hierarchy). Non-leaf lrqs hold other higher schedulable entities
 301          * (like users, containers etc.)
 302          *
 303          * leaf_cfs_rq_list ties together list of leaf cfs_rq's in a cpu. This
 304          * list is used during load balance.
 305          */
 306         int on_list;
 307         struct list_head leaf_cfs_rq_list;
 308         struct task_group *tg;  /* group that "owns" this runqueue */
 309
 310 #ifdef CONFIG_CFS_BANDWIDTH
 311         int runtime_enabled;
 312         u64 runtime_expires;
 313         s64 runtime_remaining;
 314
 315         u64 throttled_clock, throttled_clock_task;
 316         u64 throttled_clock_task_time;
 317         int throttled, throttle_count;
 318         struct list_head throttled_list;
 319 #endif /* CONFIG_CFS_BANDWIDTH */
 320 #endif /* CONFIG_FAIR_GROUP_SCHED */
 321 };
 322
 323 static inline int rt_bandwidth_enabled(void)
 324 {
 325         return sysctl_sched_rt_runtime >= 0;
 326 }
 327
 328 /* Real-Time classes' related field in a runqueue: */
 329 struct rt_rq {
 330         struct rt_prio_array active;
 331         unsigned int rt_nr_running;
 332 #if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
 333         struct {
 334                 int curr; /* highest queued rt task prio */
 335 #ifdef CONFIG_SMP
 336                 int next; /* next highest */
 337 #endif
 338         } highest_prio;
 339 #endif
 340 #ifdef CONFIG_SMP
 341         unsigned long rt_nr_migratory;
 342         unsigned long rt_nr_total;
 343         int overloaded;
 344         struct plist_head pushable_tasks;
 345 #endif
 346         int rt_throttled;
 347         u64 rt_time;
 348         u64 rt_runtime;
 349         /* Nests inside the rq lock: */
 350         raw_spinlock_t rt_runtime_lock;
 351
 352 #ifdef CONFIG_RT_GROUP_SCHED
 353         unsigned long rt_nr_boosted;
 354
 355         struct rq *rq;
 356         struct list_head leaf_rt_rq_list;
 357         struct task_group *tg;
 358 #endif
 359 };
 360
 361 #ifdef CONFIG_SMP
 362
 363 /*
 364  * We add the notion of a root-domain which will be used to define per-domain
 365  * variables. Each exclusive cpuset essentially defines an island domain by
 366  * fully partitioning the member cpus from any other cpuset. Whenever a new
 367  * exclusive cpuset is created, we also create and attach a new root-domain
 368  * object.
 369  *
 370  */
 371 struct root_domain {
 372         atomic_t refcount;
 373         atomic_t rto_count;
 374         struct rcu_head rcu;
 375         cpumask_var_t span;
 376         cpumask_var_t online;
 377
 378         /*
 379          * The "RT overload" flag: it gets set if a CPU has more than
 380          * one runnable RT task.
 381          */
 382         cpumask_var_t rto_mask;
 383         struct cpupri cpupri;
 384 };
 385
 386 extern struct root_domain def_root_domain;
 387
 388 #endif /* CONFIG_SMP */
 389
 390 /*
 391  * This is the main, per-CPU runqueue data structure.
 392  *
 393  * Locking rule: those places that want to lock multiple runqueues
 394  * (such as the load balancing or the thread migration code), lock
 395  * acquire operations must be ordered by ascending &runqueue.
 396  */
 397 struct rq {
 398         /* runqueue lock: */
 399         raw_spinlock_t lock;
 400
 401         /*
 402          * nr_running and cpu_load should be in the same cacheline because
 403          * remote CPUs use both these fields when doing load calculation.
 404          */
 405         unsigned int nr_running;
 406         #define CPU_LOAD_IDX_MAX 5
 407         unsigned long cpu_load[CPU_LOAD_IDX_MAX];
 408         unsigned long last_load_update_tick;
 409 #ifdef CONFIG_NO_HZ_COMMON
 410         u64 nohz_stamp;
 411         unsigned long nohz_flags;
 412 #endif
 413 #ifdef CONFIG_NO_HZ_FULL
 414         unsigned long last_sched_tick;
 415 #endif
 416         int skip_clock_update;
 417
 418 #ifdef CONFIG_CPUQUIET_FRAMEWORK
 419         /* time-based average load */
 420         u64 nr_last_stamp;
 421         u64 nr_running_integral;
 422         seqcount_t ave_seqcnt;
 423 #endif
 424
 425         /* capture load from *all* tasks on this cpu: */
 426         struct load_weight load;
 427         unsigned long nr_load_updates;
 428         u64 nr_switches;
 429
 430         struct cfs_rq cfs;
 431         struct rt_rq rt;
 432
 433 #ifdef CONFIG_FAIR_GROUP_SCHED
 434         /* list of leaf cfs_rq on this cpu: */
 435         struct list_head leaf_cfs_rq_list;
 436 #ifdef CONFIG_SMP
 437         unsigned long h_load_throttle;
 438 #endif /* CONFIG_SMP */
 439 #endif /* CONFIG_FAIR_GROUP_SCHED */
 440
 441 #ifdef CONFIG_RT_GROUP_SCHED
 442         struct list_head leaf_rt_rq_list;
 443 #endif
 444
 445         /*
 446          * This is part of a global counter where only the total sum
 447          * over all CPUs matters. A task can increase this counter on
 448          * one CPU and if it got migrated afterwards it may decrease
 449          * it on another CPU. Always updated under the runqueue lock:
 450          */
 451         unsigned long nr_uninterruptible;
 452
 453         struct task_struct *curr, *idle, *stop;
 454         unsigned long next_balance;
 455         struct mm_struct *prev_mm;
 456
 457         u64 clock;
 458         u64 clock_task;
 459
 460         atomic_t nr_iowait;
 461
 462 #ifdef CONFIG_SMP
 463         struct root_domain *rd;
 464         struct sched_domain *sd;
 465
 466         unsigned long cpu_power;
 467
 468         unsigned char idle_balance;
 469         /* For active balancing */
 470         int post_schedule;
 471         int active_balance;
 472         int push_cpu;
 473         struct cpu_stop_work active_balance_work;
 474 #ifdef CONFIG_SCHED_HMP
 475         struct task_struct *migrate_task;
 476         int wake_for_idle_pull;
 477 #endif
 478         /* cpu of this runqueue: */
 479         int cpu;
 480         int online;
 481
 482         struct list_head cfs_tasks;
 483
 484         u64 rt_avg;
 485         u64 age_stamp;
 486         u64 idle_stamp;
 487         u64 avg_idle;
 488 #endif
 489
 490 #ifdef CONFIG_IRQ_TIME_ACCOUNTING
 491         u64 prev_irq_time;
 492 #endif
 493 #ifdef CONFIG_PARAVIRT
 494         u64 prev_steal_time;
 495 #endif
 496 #ifdef CONFIG_PARAVIRT_TIME_ACCOUNTING
 497         u64 prev_steal_time_rq;
 498 #endif
 499
 500         /* calc_load related fields */
 501         unsigned long calc_load_update;
 502         long calc_load_active;
 503
 504 #ifdef CONFIG_SCHED_HRTICK
 505 #ifdef CONFIG_SMP
 506         int hrtick_csd_pending;
 507         struct call_single_data hrtick_csd;
 508 #endif
 509         struct hrtimer hrtick_timer;
 510 #endif
 511
 512 #ifdef CONFIG_SCHEDSTATS
 513         /* latency stats */
 514         struct sched_info rq_sched_info;
 515         unsigned long long rq_cpu_time;
 516         /* could above be rq->cfs_rq.exec_clock + rq->rt_rq.rt_runtime ? */
 517
 518         /* sys_sched_yield() stats */
 519         unsigned int yld_count;
 520
 521         /* schedule() stats */
 522         unsigned int sched_count;
 523         unsigned int sched_goidle;
 524
 525         /* try_to_wake_up() stats */
 526         unsigned int ttwu_count;
 527         unsigned int ttwu_local;
 528 #endif
 529
 530 #ifdef CONFIG_SMP
 531         struct llist_head wake_list;
 532 #endif
 533
 534         struct sched_avg avg;
 535 };
 536
 537 static inline int cpu_of(struct rq *rq)
 538 {
 539 #ifdef CONFIG_SMP
 540         return rq->cpu;
 541 #else
 542         return 0;
 543 #endif
 544 }
 545
 546 DECLARE_PER_CPU(struct rq, runqueues);
 547
 548 #define cpu_rq(cpu)             (&per_cpu(runqueues, (cpu)))
 549 #define this_rq()               (&__get_cpu_var(runqueues))
 550 #define task_rq(p)              cpu_rq(task_cpu(p))
 551 #define cpu_curr(cpu)           (cpu_rq(cpu)->curr)
 552 #define raw_rq()                (&__raw_get_cpu_var(runqueues))
 553
 554 #ifdef CONFIG_SMP
 555
 556 #define rcu_dereference_check_sched_domain(p) \
 557         rcu_dereference_check((p), \
 558                               lockdep_is_held(&sched_domains_mutex))
 559
 560 /*
 561  * The domain tree (rq->sd) is protected by RCU's quiescent state transition.
 562  * See detach_destroy_domains: synchronize_sched for details.
 563  *
 564  * The domain tree of any CPU may only be accessed from within
 565  * preempt-disabled sections.
 566  */
 567 #define for_each_domain(cpu, __sd) \
 568         for (__sd = rcu_dereference_check_sched_domain(cpu_rq(cpu)->sd); \
 569                         __sd; __sd = __sd->parent)
 570
 571 #define for_each_lower_domain(sd) for (; sd; sd = sd->child)
 572
 573 /**
 574  * highest_flag_domain - Return highest sched_domain containing flag.
 575  * @cpu:        The cpu whose highest level of sched domain is to
 576  *              be returned.
 577  * @flag:       The flag to check for the highest sched_domain
 578  *              for the given cpu.
 579  *
 580  * Returns the highest sched_domain of a cpu which contains the given flag.
 581  */
 582 static inline struct sched_domain *highest_flag_domain(int cpu, int flag)
 583 {
 584         struct sched_domain *sd, *hsd = NULL;
 585
 586         for_each_domain(cpu, sd) {
 587                 if (!(sd->flags & flag))
 588                         break;
 589                 hsd = sd;
 590         }
 591
 592         return hsd;
 593 }
 594
 595 DECLARE_PER_CPU(struct sched_domain *, sd_llc);
 596 DECLARE_PER_CPU(int, sd_llc_id);
 597
 598 struct sched_group_power {
 599         atomic_t ref;
 600         /*
 601          * CPU power of this group, SCHED_LOAD_SCALE being max power for a
 602          * single CPU.
 603          */
 604         unsigned int power, power_orig;
 605         unsigned long next_update;
 606         /*
 607          * Number of busy cpus in this group.
 608          */
 609         atomic_t nr_busy_cpus;
 610
 611         unsigned long cpumask[0]; /* iteration mask */
 612 };
 613
 614 struct sched_group {
 615         struct sched_group *next;       /* Must be a circular list */
 616         atomic_t ref;
 617
 618         unsigned int group_weight;
 619         struct sched_group_power *sgp;
 620
 621         /*
 622          * The CPUs this group covers.
 623          *
 624          * NOTE: this field is variable length. (Allocated dynamically
 625          * by attaching extra space to the end of the structure,
 626          * depending on how many CPUs the kernel has booted up with)
 627          */
 628         unsigned long cpumask[0];
 629 };
 630
 631 static inline struct cpumask *sched_group_cpus(struct sched_group *sg)
 632 {
 633         return to_cpumask(sg->cpumask);
 634 }
 635
 636 /*
 637  * cpumask masking which cpus in the group are allowed to iterate up the domain
 638  * tree.
 639  */
 640 static inline struct cpumask *sched_group_mask(struct sched_group *sg)
 641 {
 642         return to_cpumask(sg->sgp->cpumask);
 643 }
 644
 645 /**
 646  * group_first_cpu - Returns the first cpu in the cpumask of a sched_group.
 647  * @group: The group whose first cpu is to be returned.
 648  */
 649 static inline unsigned int group_first_cpu(struct sched_group *group)
 650 {
 651         return cpumask_first(sched_group_cpus(group));
 652 }
 653
 654 extern int group_balance_cpu(struct sched_group *sg);
 655
 656 #ifdef CONFIG_SCHED_HMP
 657 static LIST_HEAD(hmp_domains);
 658 DECLARE_PER_CPU(struct hmp_domain *, hmp_cpu_domain);
 659 #define hmp_cpu_domain(cpu)     (per_cpu(hmp_cpu_domain, (cpu)))
 660 #endif /* CONFIG_SCHED_HMP */
 661
 662 #endif /* CONFIG_SMP */
 663
 664 #include "stats.h"
 665 #include "auto_group.h"
 666
 667 #ifdef CONFIG_CGROUP_SCHED
 668
 669 /*
 670  * Return the group to which this tasks belongs.
 671  *
 672  * We cannot use task_subsys_state() and friends because the cgroup
 673  * subsystem changes that value before the cgroup_subsys::attach() method
 674  * is called, therefore we cannot pin it and might observe the wrong value.
 675  *
 676  * The same is true for autogroup's p->signal->autogroup->tg, the autogroup
 677  * core changes this before calling sched_move_task().
 678  *
 679  * Instead we use a 'copy' which is updated from sched_move_task() while
 680  * holding both task_struct::pi_lock and rq::lock.
 681  */
 682 static inline struct task_group *task_group(struct task_struct *p)
 683 {
 684         return p->sched_task_group;
 685 }
 686
 687 /* Change a task's cfs_rq and parent entity if it moves across CPUs/groups */
 688 static inline void set_task_rq(struct task_struct *p, unsigned int cpu)
 689 {
 690 #if defined(CONFIG_FAIR_GROUP_SCHED) || defined(CONFIG_RT_GROUP_SCHED)
 691         struct task_group *tg = task_group(p);
 692 #endif
 693
 694 #ifdef CONFIG_FAIR_GROUP_SCHED
 695         p->se.cfs_rq = tg->cfs_rq[cpu];
 696         p->se.parent = tg->se[cpu];
 697 #endif
 698
 699 #ifdef CONFIG_RT_GROUP_SCHED
 700         p->rt.rt_rq  = tg->rt_rq[cpu];
 701         p->rt.parent = tg->rt_se[cpu];
 702 #endif
 703 }
 704
 705 #else /* CONFIG_CGROUP_SCHED */
 706
 707 static inline void set_task_rq(struct task_struct *p, unsigned int cpu) { }
 708 static inline struct task_group *task_group(struct task_struct *p)
 709 {
 710         return NULL;
 711 }
 712
 713 #endif /* CONFIG_CGROUP_SCHED */
 714
 715 static inline void __set_task_cpu(struct task_struct *p, unsigned int cpu)
 716 {
 717         set_task_rq(p, cpu);
 718 #ifdef CONFIG_SMP
 719         /*
 720          * After ->cpu is set up to a new value, task_rq_lock(p, ...) can be
 721          * successfuly executed on another CPU. We must ensure that updates of
 722          * per-task data have been completed by this moment.
 723          */
 724         smp_wmb();
 725         task_thread_info(p)->cpu = cpu;
 726 #endif
 727 }
 728
 729 /*
 730  * Tunables that become constants when CONFIG_SCHED_DEBUG is off:
 731  */
 732 #ifdef CONFIG_SCHED_DEBUG
 733 # include <linux/static_key.h>
 734 # define const_debug __read_mostly
 735 #else
 736 # define const_debug const
 737 #endif
 738
 739 extern const_debug unsigned int sysctl_sched_features;
 740
 741 #define SCHED_FEAT(name, enabled)       \
 742         __SCHED_FEAT_##name ,
 743
 744 enum {
 745 #include "features.h"
 746         __SCHED_FEAT_NR,
 747 };
 748
 749 #undef SCHED_FEAT
 750
 751 #if defined(CONFIG_SCHED_DEBUG) && defined(HAVE_JUMP_LABEL)
 752 static __always_inline bool static_branch__true(struct static_key *key)
 753 {
 754         return static_key_true(key); /* Not out of line branch. */
 755 }
 756
 757 static __always_inline bool static_branch__false(struct static_key *key)
 758 {
 759         return static_key_false(key); /* Out of line branch. */
 760 }
 761
 762 #define SCHED_FEAT(name, enabled)                                       \
 763 static __always_inline bool static_branch_##name(struct static_key *key) \
 764 {                                                                       \
 765         return static_branch__##enabled(key);                           \
 766 }
 767
 768 #include "features.h"
 769
 770 #undef SCHED_FEAT
 771
 772 extern struct static_key sched_feat_keys[__SCHED_FEAT_NR];
 773 #define sched_feat(x) (static_branch_##x(&sched_feat_keys[__SCHED_FEAT_##x]))
 774 #else /* !(SCHED_DEBUG && HAVE_JUMP_LABEL) */
 775 #define sched_feat(x) (sysctl_sched_features & (1UL << __SCHED_FEAT_##x))
 776 #endif /* SCHED_DEBUG && HAVE_JUMP_LABEL */
 777
 778 #ifdef CONFIG_NUMA_BALANCING
 779 #define sched_feat_numa(x) sched_feat(x)
 780 #ifdef CONFIG_SCHED_DEBUG
 781 #define numabalancing_enabled sched_feat_numa(NUMA)
 782 #else
 783 extern bool numabalancing_enabled;
 784 #endif /* CONFIG_SCHED_DEBUG */
 785 #else
 786 #define sched_feat_numa(x) (0)
 787 #define numabalancing_enabled (0)
 788 #endif /* CONFIG_NUMA_BALANCING */
 789
 790 static inline u64 global_rt_period(void)
 791 {
 792         return (u64)sysctl_sched_rt_period * NSEC_PER_USEC;
 793 }
 794
 795 static inline u64 global_rt_runtime(void)
 796 {
 797         if (sysctl_sched_rt_runtime < 0)
 798                 return RUNTIME_INF;
 799
 800         return (u64)sysctl_sched_rt_runtime * NSEC_PER_USEC;
 801 }
 802
 803
 804
 805 static inline int task_current(struct rq *rq, struct task_struct *p)
 806 {
 807         return rq->curr == p;
 808 }
 809
 810 static inline int task_running(struct rq *rq, struct task_struct *p)
 811 {
 812 #ifdef CONFIG_SMP
 813         return p->on_cpu;
 814 #else
 815         return task_current(rq, p);
 816 #endif
 817 }
 818
 819
 820 #ifndef prepare_arch_switch
 821 # define prepare_arch_switch(next)      do { } while (0)
 822 #endif
 823 #ifndef finish_arch_switch
 824 # define finish_arch_switch(prev)       do { } while (0)
 825 #endif
 826 #ifndef finish_arch_post_lock_switch
 827 # define finish_arch_post_lock_switch() do { } while (0)
 828 #endif
 829
 830 #ifndef __ARCH_WANT_UNLOCKED_CTXSW
 831 static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
 832 {
 833 #ifdef CONFIG_SMP
 834         /*
 835          * We can optimise this out completely for !SMP, because the
 836          * SMP rebalancing from interrupt is the only thing that cares
 837          * here.
 838          */
 839         next->on_cpu = 1;
 840 #endif
 841 }
 842
 843 static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
 844 {
 845 #ifdef CONFIG_SMP
 846         /*
 847          * After ->on_cpu is cleared, the task can be moved to a different CPU.
 848          * We must ensure this doesn't happen until the switch is completely
 849          * finished.
 850          */
 851         smp_wmb();
 852         prev->on_cpu = 0;
 853 #endif
 854 #ifdef CONFIG_DEBUG_SPINLOCK
 855         /* this is a valid case when another task releases the spinlock */
 856         rq->lock.owner = current;
 857 #endif
 858         /*
 859          * If we are tracking spinlock dependencies then we have to
 860          * fix up the runqueue lock - which gets 'carried over' from
 861          * prev into current:
 862          */
 863         spin_acquire(&rq->lock.dep_map, 0, 0, _THIS_IP_);
 864
 865         raw_spin_unlock_irq(&rq->lock);
 866 }
 867
 868 #else /* __ARCH_WANT_UNLOCKED_CTXSW */
 869 static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
 870 {
 871 #ifdef CONFIG_SMP
 872         /*
 873          * We can optimise this out completely for !SMP, because the
 874          * SMP rebalancing from interrupt is the only thing that cares
 875          * here.
 876          */
 877         next->on_cpu = 1;
 878 #endif
 879         raw_spin_unlock(&rq->lock);
 880 }
 881
 882 static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
 883 {
 884 #ifdef CONFIG_SMP
 885         /*
 886          * After ->on_cpu is cleared, the task can be moved to a different CPU.
 887          * We must ensure this doesn't happen until the switch is completely
 888          * finished.
 889          */
 890         smp_wmb();
 891         prev->on_cpu = 0;
 892 #endif
 893         local_irq_enable();
 894 }
 895 #endif /* __ARCH_WANT_UNLOCKED_CTXSW */
 896
 897 /*
 898  * wake flags
 899  */
 900 #define WF_SYNC         0x01            /* waker goes to sleep after wakeup */
 901 #define WF_FORK         0x02            /* child wakeup after fork */
 902 #define WF_MIGRATED     0x4             /* internal use, task got migrated */
 903
 904 static inline void update_load_add(struct load_weight *lw, unsigned long inc)
 905 {
 906         lw->weight += inc;
 907         lw->inv_weight = 0;
 908 }
 909
 910 static inline void update_load_sub(struct load_weight *lw, unsigned long dec)
 911 {
 912         lw->weight -= dec;
 913         lw->inv_weight = 0;
 914 }
 915
 916 static inline void update_load_set(struct load_weight *lw, unsigned long w)
 917 {
 918         lw->weight = w;
 919         lw->inv_weight = 0;
 920 }
 921
 922 /*
 923  * To aid in avoiding the subversion of "niceness" due to uneven distribution
 924  * of tasks with abnormal "nice" values across CPUs the contribution that
 925  * each task makes to its run queue's load is weighted according to its
 926  * scheduling class and "nice" value. For SCHED_NORMAL tasks this is just a
 927  * scaled version of the new time slice allocation that they receive on time
 928  * slice expiry etc.
 929  */
 930
 931 #define WEIGHT_IDLEPRIO                3
 932 #define WMULT_IDLEPRIO         1431655765
 933
 934 /*
 935  * Nice levels are multiplicative, with a gentle 10% change for every
 936  * nice level changed. I.e. when a CPU-bound task goes from nice 0 to
 937  * nice 1, it will get ~10% less CPU time than another CPU-bound task
 938  * that remained on nice 0.
 939  *
 940  * The "10% effect" is relative and cumulative: from _any_ nice level,
 941  * if you go up 1 level, it's -10% CPU usage, if you go down 1 level
 942  * it's +10% CPU usage. (to achieve that we use a multiplier of 1.25.
 943  * If a task goes up by ~10% and another task goes down by ~10% then
 944  * the relative distance between them is ~25%.)
 945  */
 946 static const int prio_to_weight[40] = {
 947  /* -20 */     88761,     71755,     56483,     46273,     36291,
 948  /* -15 */     29154,     23254,     18705,     14949,     11916,
 949  /* -10 */      9548,      7620,      6100,      4904,      3906,
 950  /*  -5 */      3121,      2501,      1991,      1586,      1277,
 951  /*   0 */      1024,       820,       655,       526,       423,
 952  /*   5 */       335,       272,       215,       172,       137,
 953  /*  10 */       110,        87,        70,        56,        45,
 954  /*  15 */        36,        29,        23,        18,        15,
 955 };
 956
 957 /*
 958  * Inverse (2^32/x) values of the prio_to_weight[] array, precalculated.
 959  *
 960  * In cases where the weight does not change often, we can use the
 961  * precalculated inverse to speed up arithmetics by turning divisions
 962  * into multiplications:
 963  */
 964 static const u32 prio_to_wmult[40] = {
 965  /* -20 */     48388,     59856,     76040,     92818,    118348,
 966  /* -15 */    147320,    184698,    229616,    287308,    360437,
 967  /* -10 */    449829,    563644,    704093,    875809,   1099582,
 968  /*  -5 */   1376151,   1717300,   2157191,   2708050,   3363326,
 969  /*   0 */   4194304,   5237765,   6557202,   8165337,  10153587,
 970  /*   5 */  12820798,  15790321,  19976592,  24970740,  31350126,
 971  /*  10 */  39045157,  49367440,  61356676,  76695844,  95443717,
 972  /*  15 */ 119304647, 148102320, 186737708, 238609294, 286331153,
 973 };
 974
 975 #define ENQUEUE_WAKEUP          1
 976 #define ENQUEUE_HEAD            2
 977 #ifdef CONFIG_SMP
 978 #define ENQUEUE_WAKING          4       /* sched_class::task_waking was called */
 979 #else
 980 #define ENQUEUE_WAKING          0
 981 #endif
 982
 983 #define DEQUEUE_SLEEP           1
 984
 985 struct sched_class {
 986         const struct sched_class *next;
 987
 988         void (*enqueue_task) (struct rq *rq, struct task_struct *p, int flags);
 989         void (*dequeue_task) (struct rq *rq, struct task_struct *p, int flags);
 990         void (*yield_task) (struct rq *rq);
 991         bool (*yield_to_task) (struct rq *rq, struct task_struct *p, bool preempt);
 992
 993         void (*check_preempt_curr) (struct rq *rq, struct task_struct *p, int flags);
 994
 995         struct task_struct * (*pick_next_task) (struct rq *rq);
 996         void (*put_prev_task) (struct rq *rq, struct task_struct *p);
 997
 998 #ifdef CONFIG_SMP
 999         int  (*select_task_rq)(struct task_struct *p, int sd_flag, int flags);
1000         void (*migrate_task_rq)(struct task_struct *p, int next_cpu);
1001
1002         void (*pre_schedule) (struct rq *this_rq, struct task_struct *task);
1003         void (*post_schedule) (struct rq *this_rq);
1004         void (*task_waking) (struct task_struct *task);
1005         void (*task_woken) (struct rq *this_rq, struct task_struct *task);
1006
1007         void (*set_cpus_allowed)(struct task_struct *p,
1008                                  const struct cpumask *newmask);
1009
1010         void (*rq_online)(struct rq *rq);
1011         void (*rq_offline)(struct rq *rq);
1012 #endif
1013
1014         void (*set_curr_task) (struct rq *rq);
1015         void (*task_tick) (struct rq *rq, struct task_struct *p, int queued);
1016         void (*task_fork) (struct task_struct *p);
1017
1018         void (*switched_from) (struct rq *this_rq, struct task_struct *task);
1019         void (*switched_to) (struct rq *this_rq, struct task_struct *task);
1020         void (*prio_changed) (struct rq *this_rq, struct task_struct *task,
1021                              int oldprio);
1022
1023         unsigned int (*get_rr_interval) (struct rq *rq,
1024                                          struct task_struct *task);
1025
1026 #ifdef CONFIG_FAIR_GROUP_SCHED
1027         void (*task_move_group) (struct task_struct *p, int on_rq);
1028 #endif
1029 };
1030
1031 #define sched_class_highest (&stop_sched_class)
1032 #define for_each_class(class) \
1033    for (class = sched_class_highest; class; class = class->next)
1034
1035 extern const struct sched_class stop_sched_class;
1036 extern const struct sched_class rt_sched_class;
1037 extern const struct sched_class fair_sched_class;
1038 extern const struct sched_class idle_sched_class;
1039
1040
1041 #ifdef CONFIG_SMP
1042
1043 extern void update_group_power(struct sched_domain *sd, int cpu);
1044
1045 extern void trigger_load_balance(struct rq *rq, int cpu);
1046 extern void idle_balance(int this_cpu, struct rq *this_rq);
1047
1048 /*
1049  * Only depends on SMP, FAIR_GROUP_SCHED may be removed when runnable_avg
1050  * becomes useful in lb
1051  */
1052 #if defined(CONFIG_FAIR_GROUP_SCHED)
1053 extern void idle_enter_fair(struct rq *this_rq);
1054 extern void idle_exit_fair(struct rq *this_rq);
1055 #else
1056 static inline void idle_enter_fair(struct rq *this_rq) {}
1057 static inline void idle_exit_fair(struct rq *this_rq) {}
1058 #endif
1059
1060 #else   /* CONFIG_SMP */
1061
1062 static inline void idle_balance(int cpu, struct rq *rq)
1063 {
1064 }
1065
1066 #endif
1067
1068 extern void sysrq_sched_debug_show(void);
1069 extern void sched_init_granularity(void);
1070 extern void update_max_interval(void);
1071 extern int update_runtime(struct notifier_block *nfb, unsigned long action, void *hcpu);
1072 extern void init_sched_rt_class(void);
1073 extern void init_sched_fair_class(void);
1074
1075 extern void resched_task(struct task_struct *p);
1076 extern void resched_cpu(int cpu);
1077
1078 extern struct rt_bandwidth def_rt_bandwidth;
1079 extern void init_rt_bandwidth(struct rt_bandwidth *rt_b, u64 period, u64 runtime);
1080
1081 extern void update_idle_cpu_load(struct rq *this_rq);
1082
1083 #ifdef CONFIG_PARAVIRT
1084 static inline u64 steal_ticks(u64 steal)
1085 {
1086         if (unlikely(steal > NSEC_PER_SEC))
1087                 return div_u64(steal, TICK_NSEC);
1088
1089         return __iter_div_u64_rem(steal, TICK_NSEC, &steal);
1090 }
1091 #endif
1092
1093 #ifdef CONFIG_CPUQUIET_FRAMEWORK
1094 /* 27 ~= 134217728ns = 134.2ms
1095  * 26 ~=  67108864ns =  67.1ms
1096  * 25 ~=  33554432ns =  33.5ms
1097  * 24 ~=  16777216ns =  16.8ms
1098  */
1099 #define NR_AVE_SCALE(x)         ((x) << FSHIFT)
1100
1101 static inline u64 do_nr_running_integral(struct rq *rq)
1102 {
1103         s64 nr, deltax;
1104         u64 nr_running_integral = rq->nr_running_integral;
1105
1106         deltax = rq->clock_task - rq->nr_last_stamp;
1107         nr = NR_AVE_SCALE(rq->nr_running);
1108
1109         nr_running_integral += nr * deltax;
1110
1111         return nr_running_integral;
1112 }
1113 #endif
1114
1115 static inline void inc_nr_running(struct rq *rq)
1116 {
1117 #ifdef CONFIG_CPUQUIET_FRAMEWORK
1118         write_seqcount_begin(&rq->ave_seqcnt);
1119         rq->nr_running_integral = do_nr_running_integral(rq);
1120         rq->nr_last_stamp = rq->clock_task;
1121 #endif
1122         rq->nr_running++;
1123 #ifdef CONFIG_CPUQUIET_FRAMEWORK
1124         write_seqcount_end(&rq->ave_seqcnt);
1125 #endif
1126
1127 #ifdef CONFIG_NO_HZ_FULL
1128         if (rq->nr_running == 2) {
1129                 if (tick_nohz_full_cpu(rq->cpu)) {
1130                         /* Order rq->nr_running write against the IPI */
1131                         smp_wmb();
1132                         smp_send_reschedule(rq->cpu);
1133                 }
1134        }
1135 #endif
1136 }
1137
1138 static inline void dec_nr_running(struct rq *rq)
1139 {
1140 #ifdef CONFIG_CPUQUIET_FRAMEWORK
1141         write_seqcount_begin(&rq->ave_seqcnt);
1142         rq->nr_running_integral = do_nr_running_integral(rq);
1143         rq->nr_last_stamp = rq->clock_task;
1144 #endif
1145         rq->nr_running--;
1146 #ifdef CONFIG_CPUQUIET_FRAMEWORK
1147         write_seqcount_end(&rq->ave_seqcnt);
1148 #endif
1149 }
1150
1151 static inline void rq_last_tick_reset(struct rq *rq)
1152 {
1153 #ifdef CONFIG_NO_HZ_FULL
1154         rq->last_sched_tick = jiffies;
1155 #endif
1156 }
1157
1158 extern void update_rq_clock(struct rq *rq);
1159
1160 extern void activate_task(struct rq *rq, struct task_struct *p, int flags);
1161 extern void deactivate_task(struct rq *rq, struct task_struct *p, int flags);
1162
1163 extern void check_preempt_curr(struct rq *rq, struct task_struct *p, int flags);
1164
1165 extern const_debug unsigned int sysctl_sched_time_avg;
1166 extern const_debug unsigned int sysctl_sched_nr_migrate;
1167 extern const_debug unsigned int sysctl_sched_migration_cost;
1168
1169 static inline u64 sched_avg_period(void)
1170 {
1171         return (u64)sysctl_sched_time_avg * NSEC_PER_MSEC / 2;
1172 }
1173
1174 #ifdef CONFIG_SCHED_HRTICK
1175
1176 /*
1177  * Use hrtick when:
1178  *  - enabled by features
1179  *  - hrtimer is actually high res
1180  */
1181 static inline int hrtick_enabled(struct rq *rq)
1182 {
1183         if (!sched_feat(HRTICK))
1184                 return 0;
1185         if (!cpu_active(cpu_of(rq)))
1186                 return 0;
1187         return hrtimer_is_hres_active(&rq->hrtick_timer);
1188 }
1189
1190 void hrtick_start(struct rq *rq, u64 delay);
1191
1192 #else
1193
1194 static inline int hrtick_enabled(struct rq *rq)
1195 {
1196         return 0;
1197 }
1198
1199 #endif /* CONFIG_SCHED_HRTICK */
1200
1201 #ifdef CONFIG_SMP
1202 extern void sched_avg_update(struct rq *rq);
1203 static inline void sched_rt_avg_update(struct rq *rq, u64 rt_delta)
1204 {
1205         rq->rt_avg += rt_delta;
1206         sched_avg_update(rq);
1207 }
1208 #else
1209 static inline void sched_rt_avg_update(struct rq *rq, u64 rt_delta) { }
1210 static inline void sched_avg_update(struct rq *rq) { }
1211 #endif
1212
1213 extern void start_bandwidth_timer(struct hrtimer *period_timer, ktime_t period);
1214
1215 #ifdef CONFIG_SMP
1216 #ifdef CONFIG_PREEMPT
1217
1218 static inline void double_rq_lock(struct rq *rq1, struct rq *rq2);
1219
1220 /*
1221  * fair double_lock_balance: Safely acquires both rq->locks in a fair
1222  * way at the expense of forcing extra atomic operations in all
1223  * invocations.  This assures that the double_lock is acquired using the
1224  * same underlying policy as the spinlock_t on this architecture, which
1225  * reduces latency compared to the unfair variant below.  However, it
1226  * also adds more overhead and therefore may reduce throughput.
1227  */
1228 static inline int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
1229         __releases(this_rq->lock)
1230         __acquires(busiest->lock)
1231         __acquires(this_rq->lock)
1232 {
1233         raw_spin_unlock(&this_rq->lock);
1234         double_rq_lock(this_rq, busiest);
1235
1236         return 1;
1237 }
1238
1239 #else
1240 /*
1241  * Unfair double_lock_balance: Optimizes throughput at the expense of
1242  * latency by eliminating extra atomic operations when the locks are
1243  * already in proper order on entry.  This favors lower cpu-ids and will
1244  * grant the double lock to lower cpus over higher ids under contention,
1245  * regardless of entry order into the function.
1246  */
1247 static inline int _double_lock_balance(struct rq *this_rq, struct rq *busiest)
1248         __releases(this_rq->lock)
1249         __acquires(busiest->lock)
1250         __acquires(this_rq->lock)
1251 {
1252         int ret = 0;
1253
1254         if (unlikely(!raw_spin_trylock(&busiest->lock))) {
1255                 if (busiest < this_rq) {
1256                         raw_spin_unlock(&this_rq->lock);
1257                         raw_spin_lock(&busiest->lock);
1258                         raw_spin_lock_nested(&this_rq->lock,
1259                                               SINGLE_DEPTH_NESTING);
1260                         ret = 1;
1261                 } else
1262                         raw_spin_lock_nested(&busiest->lock,
1263                                               SINGLE_DEPTH_NESTING);
1264         }
1265         return ret;
1266 }
1267
1268 #endif /* CONFIG_PREEMPT */
1269
1270 /*
1271  * double_lock_balance - lock the busiest runqueue, this_rq is locked already.
1272  */
1273 static inline int double_lock_balance(struct rq *this_rq, struct rq *busiest)
1274 {
1275         if (unlikely(!irqs_disabled())) {
1276                 /* printk() doesn't work good under rq->lock */
1277                 raw_spin_unlock(&this_rq->lock);
1278                 BUG_ON(1);
1279         }
1280
1281         return _double_lock_balance(this_rq, busiest);
1282 }
1283
1284 static inline void double_unlock_balance(struct rq *this_rq, struct rq *busiest)
1285         __releases(busiest->lock)
1286 {
1287         raw_spin_unlock(&busiest->lock);
1288         lock_set_subclass(&this_rq->lock.dep_map, 0, _RET_IP_);
1289 }
1290
1291 /*
1292  * double_rq_lock - safely lock two runqueues
1293  *
1294  * Note this does not disable interrupts like task_rq_lock,
1295  * you need to do so manually before calling.
1296  */
1297 static inline void double_rq_lock(struct rq *rq1, struct rq *rq2)
1298         __acquires(rq1->lock)
1299         __acquires(rq2->lock)
1300 {
1301         BUG_ON(!irqs_disabled());
1302         if (rq1 == rq2) {
1303                 raw_spin_lock(&rq1->lock);
1304                 __acquire(rq2->lock);   /* Fake it out ;) */
1305         } else {
1306                 if (rq1 < rq2) {
1307                         raw_spin_lock(&rq1->lock);
1308                         raw_spin_lock_nested(&rq2->lock, SINGLE_DEPTH_NESTING);
1309                 } else {
1310                         raw_spin_lock(&rq2->lock);
1311                         raw_spin_lock_nested(&rq1->lock, SINGLE_DEPTH_NESTING);
1312                 }
1313         }
1314 }
1315
1316 /*
1317  * double_rq_unlock - safely unlock two runqueues
1318  *
1319  * Note this does not restore interrupts like task_rq_unlock,
1320  * you need to do so manually after calling.
1321  */
1322 static inline void double_rq_unlock(struct rq *rq1, struct rq *rq2)
1323         __releases(rq1->lock)
1324         __releases(rq2->lock)
1325 {
1326         raw_spin_unlock(&rq1->lock);
1327         if (rq1 != rq2)
1328                 raw_spin_unlock(&rq2->lock);
1329         else
1330                 __release(rq2->lock);
1331 }
1332
1333 #else /* CONFIG_SMP */
1334
1335 /*
1336  * double_rq_lock - safely lock two runqueues
1337  *
1338  * Note this does not disable interrupts like task_rq_lock,
1339  * you need to do so manually before calling.
1340  */
1341 static inline void double_rq_lock(struct rq *rq1, struct rq *rq2)
1342         __acquires(rq1->lock)
1343         __acquires(rq2->lock)
1344 {
1345         BUG_ON(!irqs_disabled());
1346         BUG_ON(rq1 != rq2);
1347         raw_spin_lock(&rq1->lock);
1348         __acquire(rq2->lock);   /* Fake it out ;) */
1349 }
1350
1351 /*
1352  * double_rq_unlock - safely unlock two runqueues
1353  *
1354  * Note this does not restore interrupts like task_rq_unlock,
1355  * you need to do so manually after calling.
1356  */
1357 static inline void double_rq_unlock(struct rq *rq1, struct rq *rq2)
1358         __releases(rq1->lock)
1359         __releases(rq2->lock)
1360 {
1361         BUG_ON(rq1 != rq2);
1362         raw_spin_unlock(&rq1->lock);
1363         __release(rq2->lock);
1364 }
1365
1366 #endif
1367
1368 extern struct sched_entity *__pick_first_entity(struct cfs_rq *cfs_rq);
1369 extern struct sched_entity *__pick_last_entity(struct cfs_rq *cfs_rq);
1370 extern void print_cfs_stats(struct seq_file *m, int cpu);
1371 extern void print_rt_stats(struct seq_file *m, int cpu);
1372
1373 extern void init_cfs_rq(struct cfs_rq *cfs_rq);
1374 extern void init_rt_rq(struct rt_rq *rt_rq, struct rq *rq);
1375
1376 extern void cfs_bandwidth_usage_inc(void);
1377 extern void cfs_bandwidth_usage_dec(void);
1378
1379 #ifdef CONFIG_NO_HZ_COMMON
1380 enum rq_nohz_flag_bits {
1381         NOHZ_TICK_STOPPED,
1382         NOHZ_BALANCE_KICK,
1383 };
1384
1385 #define nohz_flags(cpu) (&cpu_rq(cpu)->nohz_flags)
1386 #endif
1387
1388 #ifdef CONFIG_IRQ_TIME_ACCOUNTING
1389
1390 DECLARE_PER_CPU(u64, cpu_hardirq_time);
1391 DECLARE_PER_CPU(u64, cpu_softirq_time);
1392
1393 #ifndef CONFIG_64BIT
1394 DECLARE_PER_CPU(seqcount_t, irq_time_seq);
1395
1396 static inline void irq_time_write_begin(void)
1397 {
1398         __this_cpu_inc(irq_time_seq.sequence);
1399         smp_wmb();
1400 }
1401
1402 static inline void irq_time_write_end(void)
1403 {
1404         smp_wmb();
1405         __this_cpu_inc(irq_time_seq.sequence);
1406 }
1407
1408 static inline u64 irq_time_read(int cpu)
1409 {
1410         u64 irq_time;
1411         unsigned seq;
1412
1413         do {
1414                 seq = read_seqcount_begin(&per_cpu(irq_time_seq, cpu));
1415                 irq_time = per_cpu(cpu_softirq_time, cpu) +
1416                            per_cpu(cpu_hardirq_time, cpu);
1417         } while (read_seqcount_retry(&per_cpu(irq_time_seq, cpu), seq));
1418
1419         return irq_time;
1420 }
1421 #else /* CONFIG_64BIT */
1422 static inline void irq_time_write_begin(void)
1423 {
1424 }
1425
1426 static inline void irq_time_write_end(void)
1427 {
1428 }
1429
1430 static inline u64 irq_time_read(int cpu)
1431 {
1432         return per_cpu(cpu_softirq_time, cpu) + per_cpu(cpu_hardirq_time, cpu);
1433 }
1434 #endif /* CONFIG_64BIT */
1435 #endif /* CONFIG_IRQ_TIME_ACCOUNTING */