blob: a3a04085e79407a94a0fa79b8b2da7e77be33185 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * kernel/sched.c
3 *
4 * Kernel scheduler and related syscalls
5 *
6 * Copyright (C) 1991-2002 Linus Torvalds
7 *
8 * 1996-12-23 Modified by Dave Grothe to fix bugs in semaphores and
9 * make semaphores SMP safe
10 * 1998-11-19 Implemented schedule_timeout() and related stuff
11 * by Andrea Arcangeli
12 * 2002-01-04 New ultra-scalable O(1) scheduler by Ingo Molnar:
13 * hybrid priority-list and round-robin design with
14 * an array-switch method of distributing timeslices
15 * and per-CPU runqueues. Cleanups and useful suggestions
16 * by Davide Libenzi, preemptible kernel bits by Robert Love.
17 * 2003-09-03 Interactivity tuning by Con Kolivas.
18 * 2004-04-02 Scheduler domains code by Nick Piggin
19 */
20
21#include <linux/mm.h>
22#include <linux/module.h>
23#include <linux/nmi.h>
24#include <linux/init.h>
25#include <asm/uaccess.h>
26#include <linux/highmem.h>
27#include <linux/smp_lock.h>
28#include <asm/mmu_context.h>
29#include <linux/interrupt.h>
Randy.Dunlapc59ede72006-01-11 12:17:46 -080030#include <linux/capability.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070031#include <linux/completion.h>
32#include <linux/kernel_stat.h>
Ingo Molnar9a11b49a2006-07-03 00:24:33 -070033#include <linux/debug_locks.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070034#include <linux/security.h>
35#include <linux/notifier.h>
36#include <linux/profile.h>
Nigel Cunningham7dfb7102006-12-06 20:34:23 -080037#include <linux/freezer.h>
akpm@osdl.org198e2f12006-01-12 01:05:30 -080038#include <linux/vmalloc.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070039#include <linux/blkdev.h>
40#include <linux/delay.h>
41#include <linux/smp.h>
42#include <linux/threads.h>
43#include <linux/timer.h>
44#include <linux/rcupdate.h>
45#include <linux/cpu.h>
46#include <linux/cpuset.h>
47#include <linux/percpu.h>
48#include <linux/kthread.h>
49#include <linux/seq_file.h>
50#include <linux/syscalls.h>
51#include <linux/times.h>
Jay Lan8f0ab512006-09-30 23:28:59 -070052#include <linux/tsacct_kern.h>
bibo maoc6fd91f2006-03-26 01:38:20 -080053#include <linux/kprobes.h>
Shailabh Nagar0ff92242006-07-14 00:24:37 -070054#include <linux/delayacct.h>
Eric Dumazet5517d862007-05-08 00:32:57 -070055#include <linux/reciprocal_div.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070056
Eric Dumazet5517d862007-05-08 00:32:57 -070057#include <asm/tlb.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070058#include <asm/unistd.h>
59
60/*
Alexey Dobriyanb035b6d2007-02-10 01:45:10 -080061 * Scheduler clock - returns current time in nanosec units.
62 * This is default implementation.
63 * Architectures and sub-architectures can override this.
64 */
65unsigned long long __attribute__((weak)) sched_clock(void)
66{
67 return (unsigned long long)jiffies * (1000000000 / HZ);
68}
69
70/*
Linus Torvalds1da177e2005-04-16 15:20:36 -070071 * Convert user-nice values [ -20 ... 0 ... 19 ]
72 * to static priority [ MAX_RT_PRIO..MAX_PRIO-1 ],
73 * and back.
74 */
75#define NICE_TO_PRIO(nice) (MAX_RT_PRIO + (nice) + 20)
76#define PRIO_TO_NICE(prio) ((prio) - MAX_RT_PRIO - 20)
77#define TASK_NICE(p) PRIO_TO_NICE((p)->static_prio)
78
79/*
80 * 'User priority' is the nice value converted to something we
81 * can work with better when scaling various scheduler parameters,
82 * it's a [ 0 ... 39 ] range.
83 */
84#define USER_PRIO(p) ((p)-MAX_RT_PRIO)
85#define TASK_USER_PRIO(p) USER_PRIO((p)->static_prio)
86#define MAX_USER_PRIO (USER_PRIO(MAX_PRIO))
87
88/*
89 * Some helpers for converting nanosecond timing to jiffy resolution
90 */
91#define NS_TO_JIFFIES(TIME) ((TIME) / (1000000000 / HZ))
92#define JIFFIES_TO_NS(TIME) ((TIME) * (1000000000 / HZ))
93
94/*
95 * These are the 'tuning knobs' of the scheduler:
96 *
97 * Minimum timeslice is 5 msecs (or 1 jiffy, whichever is larger),
98 * default timeslice is 100 msecs, maximum timeslice is 800 msecs.
99 * Timeslices get refilled after they expire.
100 */
101#define MIN_TIMESLICE max(5 * HZ / 1000, 1)
102#define DEF_TIMESLICE (100 * HZ / 1000)
103#define ON_RUNQUEUE_WEIGHT 30
104#define CHILD_PENALTY 95
105#define PARENT_PENALTY 100
106#define EXIT_WEIGHT 3
107#define PRIO_BONUS_RATIO 25
108#define MAX_BONUS (MAX_USER_PRIO * PRIO_BONUS_RATIO / 100)
109#define INTERACTIVE_DELTA 2
110#define MAX_SLEEP_AVG (DEF_TIMESLICE * MAX_BONUS)
111#define STARVATION_LIMIT (MAX_SLEEP_AVG)
112#define NS_MAX_SLEEP_AVG (JIFFIES_TO_NS(MAX_SLEEP_AVG))
113
114/*
115 * If a task is 'interactive' then we reinsert it in the active
116 * array after it has expired its current timeslice. (it will not
117 * continue to run immediately, it will still roundrobin with
118 * other interactive tasks.)
119 *
120 * This part scales the interactivity limit depending on niceness.
121 *
122 * We scale it linearly, offset by the INTERACTIVE_DELTA delta.
123 * Here are a few examples of different nice levels:
124 *
125 * TASK_INTERACTIVE(-20): [1,1,1,1,1,1,1,1,1,0,0]
126 * TASK_INTERACTIVE(-10): [1,1,1,1,1,1,1,0,0,0,0]
127 * TASK_INTERACTIVE( 0): [1,1,1,1,0,0,0,0,0,0,0]
128 * TASK_INTERACTIVE( 10): [1,1,0,0,0,0,0,0,0,0,0]
129 * TASK_INTERACTIVE( 19): [0,0,0,0,0,0,0,0,0,0,0]
130 *
131 * (the X axis represents the possible -5 ... 0 ... +5 dynamic
132 * priority range a task can explore, a value of '1' means the
133 * task is rated interactive.)
134 *
135 * Ie. nice +19 tasks can never get 'interactive' enough to be
136 * reinserted into the active array. And only heavily CPU-hog nice -20
137 * tasks will be expired. Default nice 0 tasks are somewhere between,
138 * it takes some effort for them to get interactive, but it's not
139 * too hard.
140 */
141
142#define CURRENT_BONUS(p) \
143 (NS_TO_JIFFIES((p)->sleep_avg) * MAX_BONUS / \
144 MAX_SLEEP_AVG)
145
146#define GRANULARITY (10 * HZ / 1000 ? : 1)
147
148#ifdef CONFIG_SMP
149#define TIMESLICE_GRANULARITY(p) (GRANULARITY * \
150 (1 << (((MAX_BONUS - CURRENT_BONUS(p)) ? : 1) - 1)) * \
151 num_online_cpus())
152#else
153#define TIMESLICE_GRANULARITY(p) (GRANULARITY * \
154 (1 << (((MAX_BONUS - CURRENT_BONUS(p)) ? : 1) - 1)))
155#endif
156
157#define SCALE(v1,v1_max,v2_max) \
158 (v1) * (v2_max) / (v1_max)
159
160#define DELTA(p) \
Martin Andersson013d3862006-03-27 01:15:18 -0800161 (SCALE(TASK_NICE(p) + 20, 40, MAX_BONUS) - 20 * MAX_BONUS / 40 + \
162 INTERACTIVE_DELTA)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700163
164#define TASK_INTERACTIVE(p) \
165 ((p)->prio <= (p)->static_prio - DELTA(p))
166
167#define INTERACTIVE_SLEEP(p) \
168 (JIFFIES_TO_NS(MAX_SLEEP_AVG * \
169 (MAX_BONUS / 2 + DELTA((p)) + 1) / MAX_BONUS - 1))
170
171#define TASK_PREEMPTS_CURR(p, rq) \
Dmitry Adamushkobd53f962007-05-08 00:33:06 -0700172 (((p)->prio < (rq)->curr->prio) && ((p)->array == (rq)->active))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700173
Linus Torvalds1da177e2005-04-16 15:20:36 -0700174#define SCALE_PRIO(x, prio) \
Peter Williams2dd73a42006-06-27 02:54:34 -0700175 max(x * (MAX_PRIO - prio) / (MAX_USER_PRIO / 2), MIN_TIMESLICE)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700176
Peter Williams2dd73a42006-06-27 02:54:34 -0700177static unsigned int static_prio_timeslice(int static_prio)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700178{
Peter Williams2dd73a42006-06-27 02:54:34 -0700179 if (static_prio < NICE_TO_PRIO(0))
180 return SCALE_PRIO(DEF_TIMESLICE * 4, static_prio);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700181 else
Peter Williams2dd73a42006-06-27 02:54:34 -0700182 return SCALE_PRIO(DEF_TIMESLICE, static_prio);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700183}
Peter Williams2dd73a42006-06-27 02:54:34 -0700184
Eric Dumazet5517d862007-05-08 00:32:57 -0700185#ifdef CONFIG_SMP
186/*
187 * Divide a load by a sched group cpu_power : (load / sg->__cpu_power)
188 * Since cpu_power is a 'constant', we can use a reciprocal divide.
189 */
190static inline u32 sg_div_cpu_power(const struct sched_group *sg, u32 load)
191{
192 return reciprocal_divide(load, sg->reciprocal_cpu_power);
193}
194
195/*
196 * Each time a sched group cpu_power is changed,
197 * we must compute its reciprocal value
198 */
199static inline void sg_inc_cpu_power(struct sched_group *sg, u32 val)
200{
201 sg->__cpu_power += val;
202 sg->reciprocal_cpu_power = reciprocal_value(sg->__cpu_power);
203}
204#endif
205
Borislav Petkov91fcdd42006-10-19 23:28:29 -0700206/*
207 * task_timeslice() scales user-nice values [ -20 ... 0 ... 19 ]
208 * to time slice values: [800ms ... 100ms ... 5ms]
209 *
210 * The higher a thread's priority, the bigger timeslices
211 * it gets during one round of execution. But even the lowest
212 * priority thread gets MIN_TIMESLICE worth of execution time.
213 */
214
Ingo Molnar36c8b582006-07-03 00:25:41 -0700215static inline unsigned int task_timeslice(struct task_struct *p)
Peter Williams2dd73a42006-06-27 02:54:34 -0700216{
217 return static_prio_timeslice(p->static_prio);
218}
219
Linus Torvalds1da177e2005-04-16 15:20:36 -0700220/*
221 * These are the runqueue data structures:
222 */
223
Linus Torvalds1da177e2005-04-16 15:20:36 -0700224struct prio_array {
225 unsigned int nr_active;
Steven Rostedtd4448862006-06-27 02:54:29 -0700226 DECLARE_BITMAP(bitmap, MAX_PRIO+1); /* include 1 bit for delimiter */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700227 struct list_head queue[MAX_PRIO];
228};
229
230/*
231 * This is the main, per-CPU runqueue data structure.
232 *
233 * Locking rule: those places that want to lock multiple runqueues
234 * (such as the load balancing or the thread migration code), lock
235 * acquire operations must be ordered by ascending &runqueue.
236 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700237struct rq {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700238 spinlock_t lock;
239
240 /*
241 * nr_running and cpu_load should be in the same cacheline because
242 * remote CPUs use both these fields when doing load calculation.
243 */
244 unsigned long nr_running;
Peter Williams2dd73a42006-06-27 02:54:34 -0700245 unsigned long raw_weighted_load;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700246#ifdef CONFIG_SMP
Nick Piggin78979862005-06-25 14:57:13 -0700247 unsigned long cpu_load[3];
Siddha, Suresh Bbdecea32007-05-08 00:32:48 -0700248 unsigned char idle_at_tick;
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -0700249#ifdef CONFIG_NO_HZ
250 unsigned char in_nohz_recently;
251#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -0700252#endif
253 unsigned long long nr_switches;
254
255 /*
256 * This is part of a global counter where only the total sum
257 * over all CPUs matters. A task can increase this counter on
258 * one CPU and if it got migrated afterwards it may decrease
259 * it on another CPU. Always updated under the runqueue lock:
260 */
261 unsigned long nr_uninterruptible;
262
263 unsigned long expired_timestamp;
Mike Galbraithb18ec802006-12-10 02:20:31 -0800264 /* Cached timestamp set by update_cpu_clock() */
265 unsigned long long most_recent_timestamp;
Ingo Molnar36c8b582006-07-03 00:25:41 -0700266 struct task_struct *curr, *idle;
Christoph Lameterc9819f42006-12-10 02:20:25 -0800267 unsigned long next_balance;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700268 struct mm_struct *prev_mm;
Ingo Molnar70b97a72006-07-03 00:25:42 -0700269 struct prio_array *active, *expired, arrays[2];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700270 int best_expired_prio;
271 atomic_t nr_iowait;
272
273#ifdef CONFIG_SMP
274 struct sched_domain *sd;
275
276 /* For active balancing */
277 int active_balance;
278 int push_cpu;
Christoph Lameter0a2966b2006-09-25 23:30:51 -0700279 int cpu; /* cpu of this runqueue */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700280
Ingo Molnar36c8b582006-07-03 00:25:41 -0700281 struct task_struct *migration_thread;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700282 struct list_head migration_queue;
283#endif
284
285#ifdef CONFIG_SCHEDSTATS
286 /* latency stats */
287 struct sched_info rq_sched_info;
288
289 /* sys_sched_yield() stats */
290 unsigned long yld_exp_empty;
291 unsigned long yld_act_empty;
292 unsigned long yld_both_empty;
293 unsigned long yld_cnt;
294
295 /* schedule() stats */
296 unsigned long sched_switch;
297 unsigned long sched_cnt;
298 unsigned long sched_goidle;
299
300 /* try_to_wake_up() stats */
301 unsigned long ttwu_cnt;
302 unsigned long ttwu_local;
303#endif
Ingo Molnarfcb99372006-07-03 00:25:10 -0700304 struct lock_class_key rq_lock_key;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700305};
306
Siddha, Suresh Bc3396622007-05-08 00:33:09 -0700307static DEFINE_PER_CPU(struct rq, runqueues) ____cacheline_aligned_in_smp;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700308
Christoph Lameter0a2966b2006-09-25 23:30:51 -0700309static inline int cpu_of(struct rq *rq)
310{
311#ifdef CONFIG_SMP
312 return rq->cpu;
313#else
314 return 0;
315#endif
316}
317
Nick Piggin674311d2005-06-25 14:57:27 -0700318/*
319 * The domain tree (rq->sd) is protected by RCU's quiescent state transition.
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -0700320 * See detach_destroy_domains: synchronize_sched for details.
Nick Piggin674311d2005-06-25 14:57:27 -0700321 *
322 * The domain tree of any CPU may only be accessed from within
323 * preempt-disabled sections.
324 */
Ingo Molnar48f24c42006-07-03 00:25:40 -0700325#define for_each_domain(cpu, __sd) \
326 for (__sd = rcu_dereference(cpu_rq(cpu)->sd); __sd; __sd = __sd->parent)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700327
328#define cpu_rq(cpu) (&per_cpu(runqueues, (cpu)))
329#define this_rq() (&__get_cpu_var(runqueues))
330#define task_rq(p) cpu_rq(task_cpu(p))
331#define cpu_curr(cpu) (cpu_rq(cpu)->curr)
332
Linus Torvalds1da177e2005-04-16 15:20:36 -0700333#ifndef prepare_arch_switch
Nick Piggin4866cde2005-06-25 14:57:23 -0700334# define prepare_arch_switch(next) do { } while (0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700335#endif
Nick Piggin4866cde2005-06-25 14:57:23 -0700336#ifndef finish_arch_switch
337# define finish_arch_switch(prev) do { } while (0)
338#endif
339
340#ifndef __ARCH_WANT_UNLOCKED_CTXSW
Ingo Molnar70b97a72006-07-03 00:25:42 -0700341static inline int task_running(struct rq *rq, struct task_struct *p)
Nick Piggin4866cde2005-06-25 14:57:23 -0700342{
343 return rq->curr == p;
344}
345
Ingo Molnar70b97a72006-07-03 00:25:42 -0700346static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
Nick Piggin4866cde2005-06-25 14:57:23 -0700347{
348}
349
Ingo Molnar70b97a72006-07-03 00:25:42 -0700350static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
Nick Piggin4866cde2005-06-25 14:57:23 -0700351{
Ingo Molnarda04c032005-09-13 11:17:59 +0200352#ifdef CONFIG_DEBUG_SPINLOCK
353 /* this is a valid case when another task releases the spinlock */
354 rq->lock.owner = current;
355#endif
Ingo Molnar8a25d5d2006-07-03 00:24:54 -0700356 /*
357 * If we are tracking spinlock dependencies then we have to
358 * fix up the runqueue lock - which gets 'carried over' from
359 * prev into current:
360 */
361 spin_acquire(&rq->lock.dep_map, 0, 0, _THIS_IP_);
362
Nick Piggin4866cde2005-06-25 14:57:23 -0700363 spin_unlock_irq(&rq->lock);
364}
365
366#else /* __ARCH_WANT_UNLOCKED_CTXSW */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700367static inline int task_running(struct rq *rq, struct task_struct *p)
Nick Piggin4866cde2005-06-25 14:57:23 -0700368{
369#ifdef CONFIG_SMP
370 return p->oncpu;
371#else
372 return rq->curr == p;
373#endif
374}
375
Ingo Molnar70b97a72006-07-03 00:25:42 -0700376static inline void prepare_lock_switch(struct rq *rq, struct task_struct *next)
Nick Piggin4866cde2005-06-25 14:57:23 -0700377{
378#ifdef CONFIG_SMP
379 /*
380 * We can optimise this out completely for !SMP, because the
381 * SMP rebalancing from interrupt is the only thing that cares
382 * here.
383 */
384 next->oncpu = 1;
385#endif
386#ifdef __ARCH_WANT_INTERRUPTS_ON_CTXSW
387 spin_unlock_irq(&rq->lock);
388#else
389 spin_unlock(&rq->lock);
390#endif
391}
392
Ingo Molnar70b97a72006-07-03 00:25:42 -0700393static inline void finish_lock_switch(struct rq *rq, struct task_struct *prev)
Nick Piggin4866cde2005-06-25 14:57:23 -0700394{
395#ifdef CONFIG_SMP
396 /*
397 * After ->oncpu is cleared, the task can be moved to a different CPU.
398 * We must ensure this doesn't happen until the switch is completely
399 * finished.
400 */
401 smp_wmb();
402 prev->oncpu = 0;
403#endif
404#ifndef __ARCH_WANT_INTERRUPTS_ON_CTXSW
405 local_irq_enable();
406#endif
407}
408#endif /* __ARCH_WANT_UNLOCKED_CTXSW */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700409
410/*
Ingo Molnarb29739f2006-06-27 02:54:51 -0700411 * __task_rq_lock - lock the runqueue a given task resides on.
412 * Must be called interrupts disabled.
413 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700414static inline struct rq *__task_rq_lock(struct task_struct *p)
Ingo Molnarb29739f2006-06-27 02:54:51 -0700415 __acquires(rq->lock)
416{
Ingo Molnar70b97a72006-07-03 00:25:42 -0700417 struct rq *rq;
Ingo Molnarb29739f2006-06-27 02:54:51 -0700418
419repeat_lock_task:
420 rq = task_rq(p);
421 spin_lock(&rq->lock);
422 if (unlikely(rq != task_rq(p))) {
423 spin_unlock(&rq->lock);
424 goto repeat_lock_task;
425 }
426 return rq;
427}
428
429/*
Linus Torvalds1da177e2005-04-16 15:20:36 -0700430 * task_rq_lock - lock the runqueue a given task resides on and disable
431 * interrupts. Note the ordering: we can safely lookup the task_rq without
432 * explicitly disabling preemption.
433 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700434static struct rq *task_rq_lock(struct task_struct *p, unsigned long *flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700435 __acquires(rq->lock)
436{
Ingo Molnar70b97a72006-07-03 00:25:42 -0700437 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700438
439repeat_lock_task:
440 local_irq_save(*flags);
441 rq = task_rq(p);
442 spin_lock(&rq->lock);
443 if (unlikely(rq != task_rq(p))) {
444 spin_unlock_irqrestore(&rq->lock, *flags);
445 goto repeat_lock_task;
446 }
447 return rq;
448}
449
Ingo Molnar70b97a72006-07-03 00:25:42 -0700450static inline void __task_rq_unlock(struct rq *rq)
Ingo Molnarb29739f2006-06-27 02:54:51 -0700451 __releases(rq->lock)
452{
453 spin_unlock(&rq->lock);
454}
455
Ingo Molnar70b97a72006-07-03 00:25:42 -0700456static inline void task_rq_unlock(struct rq *rq, unsigned long *flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700457 __releases(rq->lock)
458{
459 spin_unlock_irqrestore(&rq->lock, *flags);
460}
461
462#ifdef CONFIG_SCHEDSTATS
463/*
464 * bump this up when changing the output format or the meaning of an existing
465 * format, so that tools can adapt (or abort)
466 */
Chen, Kenneth W06066712006-12-10 02:20:35 -0800467#define SCHEDSTAT_VERSION 14
Linus Torvalds1da177e2005-04-16 15:20:36 -0700468
469static int show_schedstat(struct seq_file *seq, void *v)
470{
471 int cpu;
472
473 seq_printf(seq, "version %d\n", SCHEDSTAT_VERSION);
474 seq_printf(seq, "timestamp %lu\n", jiffies);
475 for_each_online_cpu(cpu) {
Ingo Molnar70b97a72006-07-03 00:25:42 -0700476 struct rq *rq = cpu_rq(cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700477#ifdef CONFIG_SMP
478 struct sched_domain *sd;
479 int dcnt = 0;
480#endif
481
482 /* runqueue-specific stats */
483 seq_printf(seq,
484 "cpu%d %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu",
485 cpu, rq->yld_both_empty,
486 rq->yld_act_empty, rq->yld_exp_empty, rq->yld_cnt,
487 rq->sched_switch, rq->sched_cnt, rq->sched_goidle,
488 rq->ttwu_cnt, rq->ttwu_local,
489 rq->rq_sched_info.cpu_time,
490 rq->rq_sched_info.run_delay, rq->rq_sched_info.pcnt);
491
492 seq_printf(seq, "\n");
493
494#ifdef CONFIG_SMP
495 /* domain-specific stats */
Nick Piggin674311d2005-06-25 14:57:27 -0700496 preempt_disable();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700497 for_each_domain(cpu, sd) {
498 enum idle_type itype;
499 char mask_str[NR_CPUS];
500
501 cpumask_scnprintf(mask_str, NR_CPUS, sd->span);
502 seq_printf(seq, "domain%d %s", dcnt++, mask_str);
503 for (itype = SCHED_IDLE; itype < MAX_IDLE_TYPES;
504 itype++) {
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -0800505 seq_printf(seq, " %lu %lu %lu %lu %lu %lu %lu "
506 "%lu",
Linus Torvalds1da177e2005-04-16 15:20:36 -0700507 sd->lb_cnt[itype],
508 sd->lb_balanced[itype],
509 sd->lb_failed[itype],
510 sd->lb_imbalance[itype],
511 sd->lb_gained[itype],
512 sd->lb_hot_gained[itype],
513 sd->lb_nobusyq[itype],
Chen, Kenneth W06066712006-12-10 02:20:35 -0800514 sd->lb_nobusyg[itype]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700515 }
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -0800516 seq_printf(seq, " %lu %lu %lu %lu %lu %lu %lu %lu %lu"
517 " %lu %lu %lu\n",
Linus Torvalds1da177e2005-04-16 15:20:36 -0700518 sd->alb_cnt, sd->alb_failed, sd->alb_pushed,
Nick Piggin68767a02005-06-25 14:57:20 -0700519 sd->sbe_cnt, sd->sbe_balanced, sd->sbe_pushed,
520 sd->sbf_cnt, sd->sbf_balanced, sd->sbf_pushed,
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -0800521 sd->ttwu_wake_remote, sd->ttwu_move_affine,
522 sd->ttwu_move_balance);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700523 }
Nick Piggin674311d2005-06-25 14:57:27 -0700524 preempt_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700525#endif
526 }
527 return 0;
528}
529
530static int schedstat_open(struct inode *inode, struct file *file)
531{
532 unsigned int size = PAGE_SIZE * (1 + num_online_cpus() / 32);
533 char *buf = kmalloc(size, GFP_KERNEL);
534 struct seq_file *m;
535 int res;
536
537 if (!buf)
538 return -ENOMEM;
539 res = single_open(file, show_schedstat, NULL);
540 if (!res) {
541 m = file->private_data;
542 m->buf = buf;
543 m->size = size;
544 } else
545 kfree(buf);
546 return res;
547}
548
Helge Deller15ad7cd2006-12-06 20:40:36 -0800549const struct file_operations proc_schedstat_operations = {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700550 .open = schedstat_open,
551 .read = seq_read,
552 .llseek = seq_lseek,
553 .release = single_release,
554};
555
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700556/*
557 * Expects runqueue lock to be held for atomicity of update
558 */
559static inline void
560rq_sched_info_arrive(struct rq *rq, unsigned long delta_jiffies)
561{
562 if (rq) {
563 rq->rq_sched_info.run_delay += delta_jiffies;
564 rq->rq_sched_info.pcnt++;
565 }
566}
567
568/*
569 * Expects runqueue lock to be held for atomicity of update
570 */
571static inline void
572rq_sched_info_depart(struct rq *rq, unsigned long delta_jiffies)
573{
574 if (rq)
575 rq->rq_sched_info.cpu_time += delta_jiffies;
576}
Linus Torvalds1da177e2005-04-16 15:20:36 -0700577# define schedstat_inc(rq, field) do { (rq)->field++; } while (0)
578# define schedstat_add(rq, field, amt) do { (rq)->field += (amt); } while (0)
579#else /* !CONFIG_SCHEDSTATS */
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700580static inline void
581rq_sched_info_arrive(struct rq *rq, unsigned long delta_jiffies)
582{}
583static inline void
584rq_sched_info_depart(struct rq *rq, unsigned long delta_jiffies)
585{}
Linus Torvalds1da177e2005-04-16 15:20:36 -0700586# define schedstat_inc(rq, field) do { } while (0)
587# define schedstat_add(rq, field, amt) do { } while (0)
588#endif
589
590/*
Robert P. J. Daycc2a73b2006-12-10 02:20:00 -0800591 * this_rq_lock - lock this runqueue and disable interrupts.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700592 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700593static inline struct rq *this_rq_lock(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700594 __acquires(rq->lock)
595{
Ingo Molnar70b97a72006-07-03 00:25:42 -0700596 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700597
598 local_irq_disable();
599 rq = this_rq();
600 spin_lock(&rq->lock);
601
602 return rq;
603}
604
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700605#if defined(CONFIG_SCHEDSTATS) || defined(CONFIG_TASK_DELAY_ACCT)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700606/*
607 * Called when a process is dequeued from the active array and given
608 * the cpu. We should note that with the exception of interactive
609 * tasks, the expired queue will become the active queue after the active
610 * queue is empty, without explicitly dequeuing and requeuing tasks in the
611 * expired queue. (Interactive tasks may be requeued directly to the
612 * active queue, thus delaying tasks in the expired queue from running;
613 * see scheduler_tick()).
614 *
615 * This function is only called from sched_info_arrive(), rather than
616 * dequeue_task(). Even though a task may be queued and dequeued multiple
617 * times as it is shuffled about, we're really interested in knowing how
618 * long it was from the *first* time it was queued to the time that it
619 * finally hit a cpu.
620 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700621static inline void sched_info_dequeued(struct task_struct *t)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700622{
623 t->sched_info.last_queued = 0;
624}
625
626/*
627 * Called when a task finally hits the cpu. We can now calculate how
628 * long it was waiting to run. We also note when it began so that we
629 * can keep stats on how long its timeslice is.
630 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700631static void sched_info_arrive(struct task_struct *t)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700632{
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700633 unsigned long now = jiffies, delta_jiffies = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700634
635 if (t->sched_info.last_queued)
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700636 delta_jiffies = now - t->sched_info.last_queued;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700637 sched_info_dequeued(t);
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700638 t->sched_info.run_delay += delta_jiffies;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700639 t->sched_info.last_arrival = now;
640 t->sched_info.pcnt++;
641
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700642 rq_sched_info_arrive(task_rq(t), delta_jiffies);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700643}
644
645/*
646 * Called when a process is queued into either the active or expired
647 * array. The time is noted and later used to determine how long we
648 * had to wait for us to reach the cpu. Since the expired queue will
649 * become the active queue after active queue is empty, without dequeuing
650 * and requeuing any tasks, we are interested in queuing to either. It
651 * is unusual but not impossible for tasks to be dequeued and immediately
652 * requeued in the same or another array: this can happen in sched_yield(),
653 * set_user_nice(), and even load_balance() as it moves tasks from runqueue
654 * to runqueue.
655 *
656 * This function is only called from enqueue_task(), but also only updates
657 * the timestamp if it is already not set. It's assumed that
658 * sched_info_dequeued() will clear that stamp when appropriate.
659 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700660static inline void sched_info_queued(struct task_struct *t)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700661{
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700662 if (unlikely(sched_info_on()))
663 if (!t->sched_info.last_queued)
664 t->sched_info.last_queued = jiffies;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700665}
666
667/*
668 * Called when a process ceases being the active-running process, either
669 * voluntarily or involuntarily. Now we can calculate how long we ran.
670 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700671static inline void sched_info_depart(struct task_struct *t)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700672{
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700673 unsigned long delta_jiffies = jiffies - t->sched_info.last_arrival;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700674
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700675 t->sched_info.cpu_time += delta_jiffies;
676 rq_sched_info_depart(task_rq(t), delta_jiffies);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700677}
678
679/*
680 * Called when tasks are switched involuntarily due, typically, to expiring
681 * their time slice. (This may also be called when switching to or from
682 * the idle task.) We are only called when prev != next.
683 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700684static inline void
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700685__sched_info_switch(struct task_struct *prev, struct task_struct *next)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700686{
Ingo Molnar70b97a72006-07-03 00:25:42 -0700687 struct rq *rq = task_rq(prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700688
689 /*
690 * prev now departs the cpu. It's not interesting to record
691 * stats about how efficient we were at scheduling the idle
692 * process, however.
693 */
694 if (prev != rq->idle)
695 sched_info_depart(prev);
696
697 if (next != rq->idle)
698 sched_info_arrive(next);
699}
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700700static inline void
701sched_info_switch(struct task_struct *prev, struct task_struct *next)
702{
703 if (unlikely(sched_info_on()))
704 __sched_info_switch(prev, next);
705}
Linus Torvalds1da177e2005-04-16 15:20:36 -0700706#else
707#define sched_info_queued(t) do { } while (0)
708#define sched_info_switch(t, next) do { } while (0)
Chandra Seetharaman52f17b62006-07-14 00:24:38 -0700709#endif /* CONFIG_SCHEDSTATS || CONFIG_TASK_DELAY_ACCT */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700710
711/*
712 * Adding/removing a task to/from a priority array:
713 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700714static void dequeue_task(struct task_struct *p, struct prio_array *array)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700715{
716 array->nr_active--;
717 list_del(&p->run_list);
718 if (list_empty(array->queue + p->prio))
719 __clear_bit(p->prio, array->bitmap);
720}
721
Ingo Molnar70b97a72006-07-03 00:25:42 -0700722static void enqueue_task(struct task_struct *p, struct prio_array *array)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700723{
724 sched_info_queued(p);
725 list_add_tail(&p->run_list, array->queue + p->prio);
726 __set_bit(p->prio, array->bitmap);
727 array->nr_active++;
728 p->array = array;
729}
730
731/*
732 * Put task to the end of the run list without the overhead of dequeue
733 * followed by enqueue.
734 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700735static void requeue_task(struct task_struct *p, struct prio_array *array)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700736{
737 list_move_tail(&p->run_list, array->queue + p->prio);
738}
739
Ingo Molnar70b97a72006-07-03 00:25:42 -0700740static inline void
741enqueue_task_head(struct task_struct *p, struct prio_array *array)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700742{
743 list_add(&p->run_list, array->queue + p->prio);
744 __set_bit(p->prio, array->bitmap);
745 array->nr_active++;
746 p->array = array;
747}
748
749/*
Ingo Molnarb29739f2006-06-27 02:54:51 -0700750 * __normal_prio - return the priority that is based on the static
Linus Torvalds1da177e2005-04-16 15:20:36 -0700751 * priority but is modified by bonuses/penalties.
752 *
753 * We scale the actual sleep average [0 .... MAX_SLEEP_AVG]
754 * into the -5 ... 0 ... +5 bonus/penalty range.
755 *
756 * We use 25% of the full 0...39 priority range so that:
757 *
758 * 1) nice +19 interactive tasks do not preempt nice 0 CPU hogs.
759 * 2) nice -20 CPU hogs do not get preempted by nice 0 tasks.
760 *
761 * Both properties are important to certain workloads.
762 */
Ingo Molnarb29739f2006-06-27 02:54:51 -0700763
Ingo Molnar36c8b582006-07-03 00:25:41 -0700764static inline int __normal_prio(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700765{
766 int bonus, prio;
767
Linus Torvalds1da177e2005-04-16 15:20:36 -0700768 bonus = CURRENT_BONUS(p) - MAX_BONUS / 2;
769
770 prio = p->static_prio - bonus;
771 if (prio < MAX_RT_PRIO)
772 prio = MAX_RT_PRIO;
773 if (prio > MAX_PRIO-1)
774 prio = MAX_PRIO-1;
775 return prio;
776}
777
778/*
Peter Williams2dd73a42006-06-27 02:54:34 -0700779 * To aid in avoiding the subversion of "niceness" due to uneven distribution
780 * of tasks with abnormal "nice" values across CPUs the contribution that
781 * each task makes to its run queue's load is weighted according to its
782 * scheduling class and "nice" value. For SCHED_NORMAL tasks this is just a
783 * scaled version of the new time slice allocation that they receive on time
784 * slice expiry etc.
785 */
786
787/*
788 * Assume: static_prio_timeslice(NICE_TO_PRIO(0)) == DEF_TIMESLICE
789 * If static_prio_timeslice() is ever changed to break this assumption then
790 * this code will need modification
791 */
792#define TIME_SLICE_NICE_ZERO DEF_TIMESLICE
793#define LOAD_WEIGHT(lp) \
794 (((lp) * SCHED_LOAD_SCALE) / TIME_SLICE_NICE_ZERO)
795#define PRIO_TO_LOAD_WEIGHT(prio) \
796 LOAD_WEIGHT(static_prio_timeslice(prio))
797#define RTPRIO_TO_LOAD_WEIGHT(rp) \
798 (PRIO_TO_LOAD_WEIGHT(MAX_RT_PRIO) + LOAD_WEIGHT(rp))
799
Ingo Molnar36c8b582006-07-03 00:25:41 -0700800static void set_load_weight(struct task_struct *p)
Peter Williams2dd73a42006-06-27 02:54:34 -0700801{
Ingo Molnarb29739f2006-06-27 02:54:51 -0700802 if (has_rt_policy(p)) {
Peter Williams2dd73a42006-06-27 02:54:34 -0700803#ifdef CONFIG_SMP
804 if (p == task_rq(p)->migration_thread)
805 /*
806 * The migration thread does the actual balancing.
807 * Giving its load any weight will skew balancing
808 * adversely.
809 */
810 p->load_weight = 0;
811 else
812#endif
813 p->load_weight = RTPRIO_TO_LOAD_WEIGHT(p->rt_priority);
814 } else
815 p->load_weight = PRIO_TO_LOAD_WEIGHT(p->static_prio);
816}
817
Ingo Molnar36c8b582006-07-03 00:25:41 -0700818static inline void
Ingo Molnar70b97a72006-07-03 00:25:42 -0700819inc_raw_weighted_load(struct rq *rq, const struct task_struct *p)
Peter Williams2dd73a42006-06-27 02:54:34 -0700820{
821 rq->raw_weighted_load += p->load_weight;
822}
823
Ingo Molnar36c8b582006-07-03 00:25:41 -0700824static inline void
Ingo Molnar70b97a72006-07-03 00:25:42 -0700825dec_raw_weighted_load(struct rq *rq, const struct task_struct *p)
Peter Williams2dd73a42006-06-27 02:54:34 -0700826{
827 rq->raw_weighted_load -= p->load_weight;
828}
829
Ingo Molnar70b97a72006-07-03 00:25:42 -0700830static inline void inc_nr_running(struct task_struct *p, struct rq *rq)
Peter Williams2dd73a42006-06-27 02:54:34 -0700831{
832 rq->nr_running++;
833 inc_raw_weighted_load(rq, p);
834}
835
Ingo Molnar70b97a72006-07-03 00:25:42 -0700836static inline void dec_nr_running(struct task_struct *p, struct rq *rq)
Peter Williams2dd73a42006-06-27 02:54:34 -0700837{
838 rq->nr_running--;
839 dec_raw_weighted_load(rq, p);
840}
841
842/*
Ingo Molnarb29739f2006-06-27 02:54:51 -0700843 * Calculate the expected normal priority: i.e. priority
844 * without taking RT-inheritance into account. Might be
845 * boosted by interactivity modifiers. Changes upon fork,
846 * setprio syscalls, and whenever the interactivity
847 * estimator recalculates.
848 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700849static inline int normal_prio(struct task_struct *p)
Ingo Molnarb29739f2006-06-27 02:54:51 -0700850{
851 int prio;
852
853 if (has_rt_policy(p))
854 prio = MAX_RT_PRIO-1 - p->rt_priority;
855 else
856 prio = __normal_prio(p);
857 return prio;
858}
859
860/*
861 * Calculate the current priority, i.e. the priority
862 * taken into account by the scheduler. This value might
863 * be boosted by RT tasks, or might be boosted by
864 * interactivity modifiers. Will be RT if the task got
865 * RT-boosted. If not then it returns p->normal_prio.
866 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700867static int effective_prio(struct task_struct *p)
Ingo Molnarb29739f2006-06-27 02:54:51 -0700868{
869 p->normal_prio = normal_prio(p);
870 /*
871 * If we are RT tasks or we were boosted to RT priority,
872 * keep the priority unchanged. Otherwise, update priority
873 * to the normal priority:
874 */
875 if (!rt_prio(p->prio))
876 return p->normal_prio;
877 return p->prio;
878}
879
880/*
Linus Torvalds1da177e2005-04-16 15:20:36 -0700881 * __activate_task - move a task to the runqueue.
882 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700883static void __activate_task(struct task_struct *p, struct rq *rq)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700884{
Ingo Molnar70b97a72006-07-03 00:25:42 -0700885 struct prio_array *target = rq->active;
Con Kolivasd425b272006-03-31 02:31:29 -0800886
Linus Torvaldsf1adad72006-05-21 18:54:09 -0700887 if (batch_task(p))
Con Kolivasd425b272006-03-31 02:31:29 -0800888 target = rq->expired;
889 enqueue_task(p, target);
Peter Williams2dd73a42006-06-27 02:54:34 -0700890 inc_nr_running(p, rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700891}
892
893/*
894 * __activate_idle_task - move idle task to the _front_ of runqueue.
895 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700896static inline void __activate_idle_task(struct task_struct *p, struct rq *rq)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700897{
898 enqueue_task_head(p, rq->active);
Peter Williams2dd73a42006-06-27 02:54:34 -0700899 inc_nr_running(p, rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700900}
901
Ingo Molnarb29739f2006-06-27 02:54:51 -0700902/*
903 * Recalculate p->normal_prio and p->prio after having slept,
904 * updating the sleep-average too:
905 */
Ingo Molnar36c8b582006-07-03 00:25:41 -0700906static int recalc_task_prio(struct task_struct *p, unsigned long long now)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700907{
908 /* Caller must always ensure 'now >= p->timestamp' */
Con Kolivas72d28542006-06-27 02:54:30 -0700909 unsigned long sleep_time = now - p->timestamp;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700910
Con Kolivasd425b272006-03-31 02:31:29 -0800911 if (batch_task(p))
Ingo Molnarb0a94992006-01-14 13:20:41 -0800912 sleep_time = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700913
914 if (likely(sleep_time > 0)) {
915 /*
Con Kolivas72d28542006-06-27 02:54:30 -0700916 * This ceiling is set to the lowest priority that would allow
917 * a task to be reinserted into the active array on timeslice
918 * completion.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700919 */
Con Kolivas72d28542006-06-27 02:54:30 -0700920 unsigned long ceiling = INTERACTIVE_SLEEP(p);
Con Kolivase72ff0b2006-03-31 02:31:26 -0800921
Con Kolivas72d28542006-06-27 02:54:30 -0700922 if (p->mm && sleep_time > ceiling && p->sleep_avg < ceiling) {
923 /*
924 * Prevents user tasks from achieving best priority
925 * with one single large enough sleep.
926 */
927 p->sleep_avg = ceiling;
928 /*
929 * Using INTERACTIVE_SLEEP() as a ceiling places a
930 * nice(0) task 1ms sleep away from promotion, and
931 * gives it 700ms to round-robin with no chance of
932 * being demoted. This is more than generous, so
933 * mark this sleep as non-interactive to prevent the
934 * on-runqueue bonus logic from intervening should
935 * this task not receive cpu immediately.
936 */
937 p->sleep_type = SLEEP_NONINTERACTIVE;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700938 } else {
939 /*
Linus Torvalds1da177e2005-04-16 15:20:36 -0700940 * Tasks waking from uninterruptible sleep are
941 * limited in their sleep_avg rise as they
942 * are likely to be waiting on I/O
943 */
Con Kolivas3dee3862006-03-31 02:31:23 -0800944 if (p->sleep_type == SLEEP_NONINTERACTIVE && p->mm) {
Con Kolivas72d28542006-06-27 02:54:30 -0700945 if (p->sleep_avg >= ceiling)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700946 sleep_time = 0;
947 else if (p->sleep_avg + sleep_time >=
Con Kolivas72d28542006-06-27 02:54:30 -0700948 ceiling) {
949 p->sleep_avg = ceiling;
950 sleep_time = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700951 }
952 }
953
954 /*
955 * This code gives a bonus to interactive tasks.
956 *
957 * The boost works by updating the 'average sleep time'
958 * value here, based on ->timestamp. The more time a
959 * task spends sleeping, the higher the average gets -
960 * and the higher the priority boost gets as well.
961 */
962 p->sleep_avg += sleep_time;
963
Linus Torvalds1da177e2005-04-16 15:20:36 -0700964 }
Con Kolivas72d28542006-06-27 02:54:30 -0700965 if (p->sleep_avg > NS_MAX_SLEEP_AVG)
966 p->sleep_avg = NS_MAX_SLEEP_AVG;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700967 }
968
Chen Shanga3464a12005-06-25 14:57:31 -0700969 return effective_prio(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700970}
971
972/*
973 * activate_task - move a task to the runqueue and do priority recalculation
974 *
975 * Update all the scheduling statistics stuff. (sleep average
976 * calculation, priority modifiers, etc.)
977 */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700978static void activate_task(struct task_struct *p, struct rq *rq, int local)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700979{
980 unsigned long long now;
981
Chen, Kenneth W62ab6162006-12-10 02:20:36 -0800982 if (rt_task(p))
983 goto out;
984
Linus Torvalds1da177e2005-04-16 15:20:36 -0700985 now = sched_clock();
986#ifdef CONFIG_SMP
987 if (!local) {
988 /* Compensate for drifting sched_clock */
Ingo Molnar70b97a72006-07-03 00:25:42 -0700989 struct rq *this_rq = this_rq();
Mike Galbraithb18ec802006-12-10 02:20:31 -0800990 now = (now - this_rq->most_recent_timestamp)
991 + rq->most_recent_timestamp;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700992 }
993#endif
994
Ingo Molnarece8a682006-12-06 20:37:24 -0800995 /*
996 * Sleep time is in units of nanosecs, so shift by 20 to get a
997 * milliseconds-range estimation of the amount of time that the task
998 * spent sleeping:
999 */
1000 if (unlikely(prof_on == SLEEP_PROFILING)) {
1001 if (p->state == TASK_UNINTERRUPTIBLE)
1002 profile_hits(SLEEP_PROFILING, (void *)get_wchan(p),
1003 (now - p->timestamp) >> 20);
1004 }
1005
Chen, Kenneth W62ab6162006-12-10 02:20:36 -08001006 p->prio = recalc_task_prio(p, now);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001007
1008 /*
1009 * This checks to make sure it's not an uninterruptible task
1010 * that is now waking up.
1011 */
Con Kolivas3dee3862006-03-31 02:31:23 -08001012 if (p->sleep_type == SLEEP_NORMAL) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001013 /*
1014 * Tasks which were woken up by interrupts (ie. hw events)
1015 * are most likely of interactive nature. So we give them
1016 * the credit of extending their sleep time to the period
1017 * of time they spend on the runqueue, waiting for execution
1018 * on a CPU, first time around:
1019 */
1020 if (in_interrupt())
Con Kolivas3dee3862006-03-31 02:31:23 -08001021 p->sleep_type = SLEEP_INTERRUPTED;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001022 else {
1023 /*
1024 * Normal first-time wakeups get a credit too for
1025 * on-runqueue time, but it will be weighted down:
1026 */
Con Kolivas3dee3862006-03-31 02:31:23 -08001027 p->sleep_type = SLEEP_INTERACTIVE;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001028 }
1029 }
1030 p->timestamp = now;
Chen, Kenneth W62ab6162006-12-10 02:20:36 -08001031out:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001032 __activate_task(p, rq);
1033}
1034
1035/*
1036 * deactivate_task - remove a task from the runqueue.
1037 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07001038static void deactivate_task(struct task_struct *p, struct rq *rq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001039{
Peter Williams2dd73a42006-06-27 02:54:34 -07001040 dec_nr_running(p, rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001041 dequeue_task(p, p->array);
1042 p->array = NULL;
1043}
1044
1045/*
1046 * resched_task - mark a task 'to be rescheduled now'.
1047 *
1048 * On UP this means the setting of the need_resched flag, on SMP it
1049 * might also involve a cross-CPU call to trigger the scheduler on
1050 * the target CPU.
1051 */
1052#ifdef CONFIG_SMP
Andi Kleen495ab9c2006-06-26 13:59:11 +02001053
1054#ifndef tsk_is_polling
1055#define tsk_is_polling(t) test_tsk_thread_flag(t, TIF_POLLING_NRFLAG)
1056#endif
1057
Ingo Molnar36c8b582006-07-03 00:25:41 -07001058static void resched_task(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001059{
Nick Piggin64c7c8f2005-11-08 21:39:04 -08001060 int cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001061
1062 assert_spin_locked(&task_rq(p)->lock);
1063
Nick Piggin64c7c8f2005-11-08 21:39:04 -08001064 if (unlikely(test_tsk_thread_flag(p, TIF_NEED_RESCHED)))
1065 return;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001066
Nick Piggin64c7c8f2005-11-08 21:39:04 -08001067 set_tsk_thread_flag(p, TIF_NEED_RESCHED);
1068
1069 cpu = task_cpu(p);
1070 if (cpu == smp_processor_id())
1071 return;
1072
Andi Kleen495ab9c2006-06-26 13:59:11 +02001073 /* NEED_RESCHED must be visible before we test polling */
Nick Piggin64c7c8f2005-11-08 21:39:04 -08001074 smp_mb();
Andi Kleen495ab9c2006-06-26 13:59:11 +02001075 if (!tsk_is_polling(p))
Nick Piggin64c7c8f2005-11-08 21:39:04 -08001076 smp_send_reschedule(cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001077}
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07001078
1079static void resched_cpu(int cpu)
1080{
1081 struct rq *rq = cpu_rq(cpu);
1082 unsigned long flags;
1083
1084 if (!spin_trylock_irqsave(&rq->lock, flags))
1085 return;
1086 resched_task(cpu_curr(cpu));
1087 spin_unlock_irqrestore(&rq->lock, flags);
1088}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001089#else
Ingo Molnar36c8b582006-07-03 00:25:41 -07001090static inline void resched_task(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001091{
Nick Piggin64c7c8f2005-11-08 21:39:04 -08001092 assert_spin_locked(&task_rq(p)->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001093 set_tsk_need_resched(p);
1094}
1095#endif
1096
1097/**
1098 * task_curr - is this task currently executing on a CPU?
1099 * @p: the task in question.
1100 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001101inline int task_curr(const struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001102{
1103 return cpu_curr(task_cpu(p)) == p;
1104}
1105
Peter Williams2dd73a42006-06-27 02:54:34 -07001106/* Used instead of source_load when we know the type == 0 */
1107unsigned long weighted_cpuload(const int cpu)
1108{
1109 return cpu_rq(cpu)->raw_weighted_load;
1110}
1111
Linus Torvalds1da177e2005-04-16 15:20:36 -07001112#ifdef CONFIG_SMP
Ingo Molnar70b97a72006-07-03 00:25:42 -07001113struct migration_req {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001114 struct list_head list;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001115
Ingo Molnar36c8b582006-07-03 00:25:41 -07001116 struct task_struct *task;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001117 int dest_cpu;
1118
Linus Torvalds1da177e2005-04-16 15:20:36 -07001119 struct completion done;
Ingo Molnar70b97a72006-07-03 00:25:42 -07001120};
Linus Torvalds1da177e2005-04-16 15:20:36 -07001121
1122/*
1123 * The task's runqueue lock must be held.
1124 * Returns true if you have to wait for migration thread.
1125 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001126static int
Ingo Molnar70b97a72006-07-03 00:25:42 -07001127migrate_task(struct task_struct *p, int dest_cpu, struct migration_req *req)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001128{
Ingo Molnar70b97a72006-07-03 00:25:42 -07001129 struct rq *rq = task_rq(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001130
1131 /*
1132 * If the task is not on a runqueue (and not running), then
1133 * it is sufficient to simply update the task's cpu field.
1134 */
1135 if (!p->array && !task_running(rq, p)) {
1136 set_task_cpu(p, dest_cpu);
1137 return 0;
1138 }
1139
1140 init_completion(&req->done);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001141 req->task = p;
1142 req->dest_cpu = dest_cpu;
1143 list_add(&req->list, &rq->migration_queue);
Ingo Molnar48f24c42006-07-03 00:25:40 -07001144
Linus Torvalds1da177e2005-04-16 15:20:36 -07001145 return 1;
1146}
1147
1148/*
1149 * wait_task_inactive - wait for a thread to unschedule.
1150 *
1151 * The caller must ensure that the task *will* unschedule sometime soon,
1152 * else this function might spin for a *long* time. This function can't
1153 * be called with interrupts off, or it may introduce deadlock with
1154 * smp_call_function() if an IPI is sent by the same process we are
1155 * waiting to become inactive.
1156 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001157void wait_task_inactive(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001158{
1159 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07001160 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001161 int preempted;
1162
1163repeat:
1164 rq = task_rq_lock(p, &flags);
1165 /* Must be off runqueue entirely, not preempted. */
1166 if (unlikely(p->array || task_running(rq, p))) {
1167 /* If it's preempted, we yield. It could be a while. */
1168 preempted = !task_running(rq, p);
1169 task_rq_unlock(rq, &flags);
1170 cpu_relax();
1171 if (preempted)
1172 yield();
1173 goto repeat;
1174 }
1175 task_rq_unlock(rq, &flags);
1176}
1177
1178/***
1179 * kick_process - kick a running thread to enter/exit the kernel
1180 * @p: the to-be-kicked thread
1181 *
1182 * Cause a process which is running on another CPU to enter
1183 * kernel-mode, without any delay. (to get signals handled.)
1184 *
1185 * NOTE: this function doesnt have to take the runqueue lock,
1186 * because all it wants to ensure is that the remote task enters
1187 * the kernel. If the IPI races and the task has been migrated
1188 * to another CPU then no harm is done and the purpose has been
1189 * achieved as well.
1190 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001191void kick_process(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001192{
1193 int cpu;
1194
1195 preempt_disable();
1196 cpu = task_cpu(p);
1197 if ((cpu != smp_processor_id()) && task_curr(p))
1198 smp_send_reschedule(cpu);
1199 preempt_enable();
1200}
1201
1202/*
Peter Williams2dd73a42006-06-27 02:54:34 -07001203 * Return a low guess at the load of a migration-source cpu weighted
1204 * according to the scheduling class and "nice" value.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001205 *
1206 * We want to under-estimate the load of migration sources, to
1207 * balance conservatively.
1208 */
Con Kolivasb9104722005-11-08 21:38:55 -08001209static inline unsigned long source_load(int cpu, int type)
1210{
Ingo Molnar70b97a72006-07-03 00:25:42 -07001211 struct rq *rq = cpu_rq(cpu);
Nick Piggina2000572006-02-10 01:51:02 -08001212
Peter Williams2dd73a42006-06-27 02:54:34 -07001213 if (type == 0)
1214 return rq->raw_weighted_load;
1215
1216 return min(rq->cpu_load[type-1], rq->raw_weighted_load);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001217}
1218
1219/*
Peter Williams2dd73a42006-06-27 02:54:34 -07001220 * Return a high guess at the load of a migration-target cpu weighted
1221 * according to the scheduling class and "nice" value.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001222 */
Con Kolivasb9104722005-11-08 21:38:55 -08001223static inline unsigned long target_load(int cpu, int type)
1224{
Ingo Molnar70b97a72006-07-03 00:25:42 -07001225 struct rq *rq = cpu_rq(cpu);
Nick Piggina2000572006-02-10 01:51:02 -08001226
Peter Williams2dd73a42006-06-27 02:54:34 -07001227 if (type == 0)
1228 return rq->raw_weighted_load;
1229
1230 return max(rq->cpu_load[type-1], rq->raw_weighted_load);
1231}
1232
1233/*
1234 * Return the average load per task on the cpu's run queue
1235 */
1236static inline unsigned long cpu_avg_load_per_task(int cpu)
1237{
Ingo Molnar70b97a72006-07-03 00:25:42 -07001238 struct rq *rq = cpu_rq(cpu);
Peter Williams2dd73a42006-06-27 02:54:34 -07001239 unsigned long n = rq->nr_running;
1240
Ingo Molnar48f24c42006-07-03 00:25:40 -07001241 return n ? rq->raw_weighted_load / n : SCHED_LOAD_SCALE;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001242}
1243
Nick Piggin147cbb42005-06-25 14:57:19 -07001244/*
1245 * find_idlest_group finds and returns the least busy CPU group within the
1246 * domain.
1247 */
1248static struct sched_group *
1249find_idlest_group(struct sched_domain *sd, struct task_struct *p, int this_cpu)
1250{
1251 struct sched_group *idlest = NULL, *this = NULL, *group = sd->groups;
1252 unsigned long min_load = ULONG_MAX, this_load = 0;
1253 int load_idx = sd->forkexec_idx;
1254 int imbalance = 100 + (sd->imbalance_pct-100)/2;
1255
1256 do {
1257 unsigned long load, avg_load;
1258 int local_group;
1259 int i;
1260
M.Baris Demirayda5a5522005-09-10 00:26:09 -07001261 /* Skip over this group if it has no CPUs allowed */
1262 if (!cpus_intersects(group->cpumask, p->cpus_allowed))
1263 goto nextgroup;
1264
Nick Piggin147cbb42005-06-25 14:57:19 -07001265 local_group = cpu_isset(this_cpu, group->cpumask);
Nick Piggin147cbb42005-06-25 14:57:19 -07001266
1267 /* Tally up the load of all CPUs in the group */
1268 avg_load = 0;
1269
1270 for_each_cpu_mask(i, group->cpumask) {
1271 /* Bias balancing toward cpus of our domain */
1272 if (local_group)
1273 load = source_load(i, load_idx);
1274 else
1275 load = target_load(i, load_idx);
1276
1277 avg_load += load;
1278 }
1279
1280 /* Adjust by relative CPU power of the group */
Eric Dumazet5517d862007-05-08 00:32:57 -07001281 avg_load = sg_div_cpu_power(group,
1282 avg_load * SCHED_LOAD_SCALE);
Nick Piggin147cbb42005-06-25 14:57:19 -07001283
1284 if (local_group) {
1285 this_load = avg_load;
1286 this = group;
1287 } else if (avg_load < min_load) {
1288 min_load = avg_load;
1289 idlest = group;
1290 }
M.Baris Demirayda5a5522005-09-10 00:26:09 -07001291nextgroup:
Nick Piggin147cbb42005-06-25 14:57:19 -07001292 group = group->next;
1293 } while (group != sd->groups);
1294
1295 if (!idlest || 100*this_load < imbalance*min_load)
1296 return NULL;
1297 return idlest;
1298}
1299
1300/*
Satoru Takeuchi0feaece2006-10-03 01:14:10 -07001301 * find_idlest_cpu - find the idlest cpu among the cpus in group.
Nick Piggin147cbb42005-06-25 14:57:19 -07001302 */
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07001303static int
1304find_idlest_cpu(struct sched_group *group, struct task_struct *p, int this_cpu)
Nick Piggin147cbb42005-06-25 14:57:19 -07001305{
M.Baris Demirayda5a5522005-09-10 00:26:09 -07001306 cpumask_t tmp;
Nick Piggin147cbb42005-06-25 14:57:19 -07001307 unsigned long load, min_load = ULONG_MAX;
1308 int idlest = -1;
1309 int i;
1310
M.Baris Demirayda5a5522005-09-10 00:26:09 -07001311 /* Traverse only the allowed CPUs */
1312 cpus_and(tmp, group->cpumask, p->cpus_allowed);
1313
1314 for_each_cpu_mask(i, tmp) {
Peter Williams2dd73a42006-06-27 02:54:34 -07001315 load = weighted_cpuload(i);
Nick Piggin147cbb42005-06-25 14:57:19 -07001316
1317 if (load < min_load || (load == min_load && i == this_cpu)) {
1318 min_load = load;
1319 idlest = i;
1320 }
1321 }
1322
1323 return idlest;
1324}
1325
Nick Piggin476d1392005-06-25 14:57:29 -07001326/*
1327 * sched_balance_self: balance the current task (running on cpu) in domains
1328 * that have the 'flag' flag set. In practice, this is SD_BALANCE_FORK and
1329 * SD_BALANCE_EXEC.
1330 *
1331 * Balance, ie. select the least loaded group.
1332 *
1333 * Returns the target CPU number, or the same CPU if no balancing is needed.
1334 *
1335 * preempt must be disabled.
1336 */
1337static int sched_balance_self(int cpu, int flag)
1338{
1339 struct task_struct *t = current;
1340 struct sched_domain *tmp, *sd = NULL;
Nick Piggin147cbb42005-06-25 14:57:19 -07001341
Chen, Kenneth Wc96d1452006-06-27 02:54:28 -07001342 for_each_domain(cpu, tmp) {
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07001343 /*
1344 * If power savings logic is enabled for a domain, stop there.
1345 */
1346 if (tmp->flags & SD_POWERSAVINGS_BALANCE)
1347 break;
Nick Piggin476d1392005-06-25 14:57:29 -07001348 if (tmp->flags & flag)
1349 sd = tmp;
Chen, Kenneth Wc96d1452006-06-27 02:54:28 -07001350 }
Nick Piggin476d1392005-06-25 14:57:29 -07001351
1352 while (sd) {
1353 cpumask_t span;
1354 struct sched_group *group;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07001355 int new_cpu, weight;
1356
1357 if (!(sd->flags & flag)) {
1358 sd = sd->child;
1359 continue;
1360 }
Nick Piggin476d1392005-06-25 14:57:29 -07001361
1362 span = sd->span;
1363 group = find_idlest_group(sd, t, cpu);
Siddha, Suresh B1a848872006-10-03 01:14:08 -07001364 if (!group) {
1365 sd = sd->child;
1366 continue;
1367 }
Nick Piggin476d1392005-06-25 14:57:29 -07001368
M.Baris Demirayda5a5522005-09-10 00:26:09 -07001369 new_cpu = find_idlest_cpu(group, t, cpu);
Siddha, Suresh B1a848872006-10-03 01:14:08 -07001370 if (new_cpu == -1 || new_cpu == cpu) {
1371 /* Now try balancing at a lower domain level of cpu */
1372 sd = sd->child;
1373 continue;
1374 }
Nick Piggin476d1392005-06-25 14:57:29 -07001375
Siddha, Suresh B1a848872006-10-03 01:14:08 -07001376 /* Now try balancing at a lower domain level of new_cpu */
Nick Piggin476d1392005-06-25 14:57:29 -07001377 cpu = new_cpu;
Nick Piggin476d1392005-06-25 14:57:29 -07001378 sd = NULL;
1379 weight = cpus_weight(span);
1380 for_each_domain(cpu, tmp) {
1381 if (weight <= cpus_weight(tmp->span))
1382 break;
1383 if (tmp->flags & flag)
1384 sd = tmp;
1385 }
1386 /* while loop will break here if sd == NULL */
1387 }
1388
1389 return cpu;
1390}
1391
1392#endif /* CONFIG_SMP */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001393
1394/*
1395 * wake_idle() will wake a task on an idle cpu if task->cpu is
1396 * not idle and an idle cpu is available. The span of cpus to
1397 * search starts with cpus closest then further out as needed,
1398 * so we always favor a closer, idle cpu.
1399 *
1400 * Returns the CPU we should wake onto.
1401 */
1402#if defined(ARCH_HAS_SCHED_WAKE_IDLE)
Ingo Molnar36c8b582006-07-03 00:25:41 -07001403static int wake_idle(int cpu, struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001404{
1405 cpumask_t tmp;
1406 struct sched_domain *sd;
1407 int i;
1408
Siddha, Suresh B49531982007-05-08 00:33:01 -07001409 /*
1410 * If it is idle, then it is the best cpu to run this task.
1411 *
1412 * This cpu is also the best, if it has more than one task already.
1413 * Siblings must be also busy(in most cases) as they didn't already
1414 * pickup the extra load from this cpu and hence we need not check
1415 * sibling runqueue info. This will avoid the checks and cache miss
1416 * penalities associated with that.
1417 */
1418 if (idle_cpu(cpu) || cpu_rq(cpu)->nr_running > 1)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001419 return cpu;
1420
1421 for_each_domain(cpu, sd) {
1422 if (sd->flags & SD_WAKE_IDLE) {
Nick Piggine0f364f2005-06-25 14:57:06 -07001423 cpus_and(tmp, sd->span, p->cpus_allowed);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001424 for_each_cpu_mask(i, tmp) {
1425 if (idle_cpu(i))
1426 return i;
1427 }
1428 }
Nick Piggine0f364f2005-06-25 14:57:06 -07001429 else
1430 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001431 }
1432 return cpu;
1433}
1434#else
Ingo Molnar36c8b582006-07-03 00:25:41 -07001435static inline int wake_idle(int cpu, struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001436{
1437 return cpu;
1438}
1439#endif
1440
1441/***
1442 * try_to_wake_up - wake up a thread
1443 * @p: the to-be-woken-up thread
1444 * @state: the mask of task states that can be woken
1445 * @sync: do a synchronous wakeup?
1446 *
1447 * Put it on the run-queue if it's not already there. The "current"
1448 * thread is always on the run-queue (except when the actual
1449 * re-schedule is in progress), and as such you're allowed to do
1450 * the simpler "current->state = TASK_RUNNING" to mark yourself
1451 * runnable without the overhead of this.
1452 *
1453 * returns failure only if the task is already active.
1454 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001455static int try_to_wake_up(struct task_struct *p, unsigned int state, int sync)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001456{
1457 int cpu, this_cpu, success = 0;
1458 unsigned long flags;
1459 long old_state;
Ingo Molnar70b97a72006-07-03 00:25:42 -07001460 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001461#ifdef CONFIG_SMP
Nick Piggin78979862005-06-25 14:57:13 -07001462 struct sched_domain *sd, *this_sd = NULL;
Ingo Molnar70b97a72006-07-03 00:25:42 -07001463 unsigned long load, this_load;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001464 int new_cpu;
1465#endif
1466
1467 rq = task_rq_lock(p, &flags);
1468 old_state = p->state;
1469 if (!(old_state & state))
1470 goto out;
1471
1472 if (p->array)
1473 goto out_running;
1474
1475 cpu = task_cpu(p);
1476 this_cpu = smp_processor_id();
1477
1478#ifdef CONFIG_SMP
1479 if (unlikely(task_running(rq, p)))
1480 goto out_activate;
1481
Nick Piggin78979862005-06-25 14:57:13 -07001482 new_cpu = cpu;
1483
Linus Torvalds1da177e2005-04-16 15:20:36 -07001484 schedstat_inc(rq, ttwu_cnt);
1485 if (cpu == this_cpu) {
1486 schedstat_inc(rq, ttwu_local);
Nick Piggin78979862005-06-25 14:57:13 -07001487 goto out_set_cpu;
1488 }
1489
1490 for_each_domain(this_cpu, sd) {
1491 if (cpu_isset(cpu, sd->span)) {
1492 schedstat_inc(sd, ttwu_wake_remote);
1493 this_sd = sd;
1494 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001495 }
1496 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001497
Nick Piggin78979862005-06-25 14:57:13 -07001498 if (unlikely(!cpu_isset(this_cpu, p->cpus_allowed)))
Linus Torvalds1da177e2005-04-16 15:20:36 -07001499 goto out_set_cpu;
1500
Linus Torvalds1da177e2005-04-16 15:20:36 -07001501 /*
Nick Piggin78979862005-06-25 14:57:13 -07001502 * Check for affine wakeup and passive balancing possibilities.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001503 */
Nick Piggin78979862005-06-25 14:57:13 -07001504 if (this_sd) {
1505 int idx = this_sd->wake_idx;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001506 unsigned int imbalance;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001507
Nick Piggina3f21bc2005-06-25 14:57:15 -07001508 imbalance = 100 + (this_sd->imbalance_pct - 100) / 2;
1509
Nick Piggin78979862005-06-25 14:57:13 -07001510 load = source_load(cpu, idx);
1511 this_load = target_load(this_cpu, idx);
1512
Nick Piggin78979862005-06-25 14:57:13 -07001513 new_cpu = this_cpu; /* Wake to this CPU if we can */
1514
Nick Piggina3f21bc2005-06-25 14:57:15 -07001515 if (this_sd->flags & SD_WAKE_AFFINE) {
1516 unsigned long tl = this_load;
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08001517 unsigned long tl_per_task;
1518
1519 tl_per_task = cpu_avg_load_per_task(this_cpu);
Peter Williams2dd73a42006-06-27 02:54:34 -07001520
Linus Torvalds1da177e2005-04-16 15:20:36 -07001521 /*
Nick Piggina3f21bc2005-06-25 14:57:15 -07001522 * If sync wakeup then subtract the (maximum possible)
1523 * effect of the currently running task from the load
1524 * of the current CPU:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001525 */
Nick Piggina3f21bc2005-06-25 14:57:15 -07001526 if (sync)
Peter Williams2dd73a42006-06-27 02:54:34 -07001527 tl -= current->load_weight;
Nick Piggina3f21bc2005-06-25 14:57:15 -07001528
1529 if ((tl <= load &&
Peter Williams2dd73a42006-06-27 02:54:34 -07001530 tl + target_load(cpu, idx) <= tl_per_task) ||
1531 100*(tl + p->load_weight) <= imbalance*load) {
Nick Piggina3f21bc2005-06-25 14:57:15 -07001532 /*
1533 * This domain has SD_WAKE_AFFINE and
1534 * p is cache cold in this domain, and
1535 * there is no bad imbalance.
1536 */
1537 schedstat_inc(this_sd, ttwu_move_affine);
1538 goto out_set_cpu;
1539 }
1540 }
1541
1542 /*
1543 * Start passive balancing when half the imbalance_pct
1544 * limit is reached.
1545 */
1546 if (this_sd->flags & SD_WAKE_BALANCE) {
1547 if (imbalance*this_load <= 100*load) {
1548 schedstat_inc(this_sd, ttwu_move_balance);
1549 goto out_set_cpu;
1550 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001551 }
1552 }
1553
1554 new_cpu = cpu; /* Could not wake to this_cpu. Wake to cpu instead */
1555out_set_cpu:
1556 new_cpu = wake_idle(new_cpu, p);
1557 if (new_cpu != cpu) {
1558 set_task_cpu(p, new_cpu);
1559 task_rq_unlock(rq, &flags);
1560 /* might preempt at this point */
1561 rq = task_rq_lock(p, &flags);
1562 old_state = p->state;
1563 if (!(old_state & state))
1564 goto out;
1565 if (p->array)
1566 goto out_running;
1567
1568 this_cpu = smp_processor_id();
1569 cpu = task_cpu(p);
1570 }
1571
1572out_activate:
1573#endif /* CONFIG_SMP */
1574 if (old_state == TASK_UNINTERRUPTIBLE) {
1575 rq->nr_uninterruptible--;
1576 /*
1577 * Tasks on involuntary sleep don't earn
1578 * sleep_avg beyond just interactive state.
1579 */
Con Kolivas3dee3862006-03-31 02:31:23 -08001580 p->sleep_type = SLEEP_NONINTERACTIVE;
Con Kolivase7c38cb2006-03-31 02:31:25 -08001581 } else
Linus Torvalds1da177e2005-04-16 15:20:36 -07001582
1583 /*
Ingo Molnard79fc0f2005-09-10 00:26:12 -07001584 * Tasks that have marked their sleep as noninteractive get
Con Kolivase7c38cb2006-03-31 02:31:25 -08001585 * woken up with their sleep average not weighted in an
1586 * interactive way.
Ingo Molnard79fc0f2005-09-10 00:26:12 -07001587 */
Con Kolivase7c38cb2006-03-31 02:31:25 -08001588 if (old_state & TASK_NONINTERACTIVE)
1589 p->sleep_type = SLEEP_NONINTERACTIVE;
1590
1591
1592 activate_task(p, rq, cpu == this_cpu);
Ingo Molnard79fc0f2005-09-10 00:26:12 -07001593 /*
Linus Torvalds1da177e2005-04-16 15:20:36 -07001594 * Sync wakeups (i.e. those types of wakeups where the waker
1595 * has indicated that it will leave the CPU in short order)
1596 * don't trigger a preemption, if the woken up task will run on
1597 * this cpu. (in this case the 'I will reschedule' promise of
1598 * the waker guarantees that the freshly woken up task is going
1599 * to be considered on this CPU.)
1600 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07001601 if (!sync || cpu != this_cpu) {
1602 if (TASK_PREEMPTS_CURR(p, rq))
1603 resched_task(rq->curr);
1604 }
1605 success = 1;
1606
1607out_running:
1608 p->state = TASK_RUNNING;
1609out:
1610 task_rq_unlock(rq, &flags);
1611
1612 return success;
1613}
1614
Ingo Molnar36c8b582006-07-03 00:25:41 -07001615int fastcall wake_up_process(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001616{
1617 return try_to_wake_up(p, TASK_STOPPED | TASK_TRACED |
1618 TASK_INTERRUPTIBLE | TASK_UNINTERRUPTIBLE, 0);
1619}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001620EXPORT_SYMBOL(wake_up_process);
1621
Ingo Molnar36c8b582006-07-03 00:25:41 -07001622int fastcall wake_up_state(struct task_struct *p, unsigned int state)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001623{
1624 return try_to_wake_up(p, state, 0);
1625}
1626
Peter Williamsbc947632006-12-19 12:48:50 +10001627static void task_running_tick(struct rq *rq, struct task_struct *p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001628/*
1629 * Perform scheduler related setup for a newly forked process p.
1630 * p is forked by current.
1631 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001632void fastcall sched_fork(struct task_struct *p, int clone_flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001633{
Nick Piggin476d1392005-06-25 14:57:29 -07001634 int cpu = get_cpu();
1635
1636#ifdef CONFIG_SMP
1637 cpu = sched_balance_self(cpu, SD_BALANCE_FORK);
1638#endif
1639 set_task_cpu(p, cpu);
1640
Linus Torvalds1da177e2005-04-16 15:20:36 -07001641 /*
1642 * We mark the process as running here, but have not actually
1643 * inserted it onto the runqueue yet. This guarantees that
1644 * nobody will actually run it, and a signal or other external
1645 * event cannot wake it up and insert it on the runqueue either.
1646 */
1647 p->state = TASK_RUNNING;
Ingo Molnarb29739f2006-06-27 02:54:51 -07001648
1649 /*
1650 * Make sure we do not leak PI boosting priority to the child:
1651 */
1652 p->prio = current->normal_prio;
1653
Linus Torvalds1da177e2005-04-16 15:20:36 -07001654 INIT_LIST_HEAD(&p->run_list);
1655 p->array = NULL;
Chandra Seetharaman52f17b62006-07-14 00:24:38 -07001656#if defined(CONFIG_SCHEDSTATS) || defined(CONFIG_TASK_DELAY_ACCT)
1657 if (unlikely(sched_info_on()))
1658 memset(&p->sched_info, 0, sizeof(p->sched_info));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001659#endif
Chen, Kenneth Wd6077cb2006-02-14 13:53:10 -08001660#if defined(CONFIG_SMP) && defined(__ARCH_WANT_UNLOCKED_CTXSW)
Nick Piggin4866cde2005-06-25 14:57:23 -07001661 p->oncpu = 0;
1662#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07001663#ifdef CONFIG_PREEMPT
Nick Piggin4866cde2005-06-25 14:57:23 -07001664 /* Want to start with kernel preemption disabled. */
Al Viroa1261f52005-11-13 16:06:55 -08001665 task_thread_info(p)->preempt_count = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001666#endif
1667 /*
1668 * Share the timeslice between parent and child, thus the
1669 * total amount of pending timeslices in the system doesn't change,
1670 * resulting in more scheduling fairness.
1671 */
1672 local_irq_disable();
1673 p->time_slice = (current->time_slice + 1) >> 1;
1674 /*
1675 * The remainder of the first timeslice might be recovered by
1676 * the parent if the child exits early enough.
1677 */
1678 p->first_time_slice = 1;
1679 current->time_slice >>= 1;
1680 p->timestamp = sched_clock();
1681 if (unlikely(!current->time_slice)) {
1682 /*
1683 * This case is rare, it happens when the parent has only
1684 * a single jiffy left from its timeslice. Taking the
1685 * runqueue lock is not a problem.
1686 */
1687 current->time_slice = 1;
Peter Williamsbc947632006-12-19 12:48:50 +10001688 task_running_tick(cpu_rq(cpu), current);
Nick Piggin476d1392005-06-25 14:57:29 -07001689 }
1690 local_irq_enable();
1691 put_cpu();
Linus Torvalds1da177e2005-04-16 15:20:36 -07001692}
1693
1694/*
1695 * wake_up_new_task - wake up a newly created task for the first time.
1696 *
1697 * This function will do some initial scheduler statistics housekeeping
1698 * that must be done for every newly created context, then puts the task
1699 * on the runqueue and wakes it.
1700 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001701void fastcall wake_up_new_task(struct task_struct *p, unsigned long clone_flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001702{
Ingo Molnar70b97a72006-07-03 00:25:42 -07001703 struct rq *rq, *this_rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001704 unsigned long flags;
1705 int this_cpu, cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001706
1707 rq = task_rq_lock(p, &flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001708 BUG_ON(p->state != TASK_RUNNING);
Nick Piggin147cbb42005-06-25 14:57:19 -07001709 this_cpu = smp_processor_id();
1710 cpu = task_cpu(p);
1711
Linus Torvalds1da177e2005-04-16 15:20:36 -07001712 /*
1713 * We decrease the sleep average of forking parents
1714 * and children as well, to keep max-interactive tasks
1715 * from forking tasks that are max-interactive. The parent
1716 * (current) is done further down, under its lock.
1717 */
1718 p->sleep_avg = JIFFIES_TO_NS(CURRENT_BONUS(p) *
1719 CHILD_PENALTY / 100 * MAX_SLEEP_AVG / MAX_BONUS);
1720
1721 p->prio = effective_prio(p);
1722
1723 if (likely(cpu == this_cpu)) {
1724 if (!(clone_flags & CLONE_VM)) {
1725 /*
1726 * The VM isn't cloned, so we're in a good position to
1727 * do child-runs-first in anticipation of an exec. This
1728 * usually avoids a lot of COW overhead.
1729 */
1730 if (unlikely(!current->array))
1731 __activate_task(p, rq);
1732 else {
1733 p->prio = current->prio;
Ingo Molnarb29739f2006-06-27 02:54:51 -07001734 p->normal_prio = current->normal_prio;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001735 list_add_tail(&p->run_list, &current->run_list);
1736 p->array = current->array;
1737 p->array->nr_active++;
Peter Williams2dd73a42006-06-27 02:54:34 -07001738 inc_nr_running(p, rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001739 }
1740 set_need_resched();
1741 } else
1742 /* Run child last */
1743 __activate_task(p, rq);
1744 /*
1745 * We skip the following code due to cpu == this_cpu
1746 *
1747 * task_rq_unlock(rq, &flags);
1748 * this_rq = task_rq_lock(current, &flags);
1749 */
1750 this_rq = rq;
1751 } else {
1752 this_rq = cpu_rq(this_cpu);
1753
1754 /*
1755 * Not the local CPU - must adjust timestamp. This should
1756 * get optimised away in the !CONFIG_SMP case.
1757 */
Mike Galbraithb18ec802006-12-10 02:20:31 -08001758 p->timestamp = (p->timestamp - this_rq->most_recent_timestamp)
1759 + rq->most_recent_timestamp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001760 __activate_task(p, rq);
1761 if (TASK_PREEMPTS_CURR(p, rq))
1762 resched_task(rq->curr);
1763
1764 /*
1765 * Parent and child are on different CPUs, now get the
1766 * parent runqueue to update the parent's ->sleep_avg:
1767 */
1768 task_rq_unlock(rq, &flags);
1769 this_rq = task_rq_lock(current, &flags);
1770 }
1771 current->sleep_avg = JIFFIES_TO_NS(CURRENT_BONUS(current) *
1772 PARENT_PENALTY / 100 * MAX_SLEEP_AVG / MAX_BONUS);
1773 task_rq_unlock(this_rq, &flags);
1774}
1775
1776/*
1777 * Potentially available exiting-child timeslices are
1778 * retrieved here - this way the parent does not get
1779 * penalized for creating too many threads.
1780 *
1781 * (this cannot be used to 'generate' timeslices
1782 * artificially, because any timeslice recovered here
1783 * was given away by the parent in the first place.)
1784 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001785void fastcall sched_exit(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001786{
1787 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07001788 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001789
1790 /*
1791 * If the child was a (relative-) CPU hog then decrease
1792 * the sleep_avg of the parent as well.
1793 */
1794 rq = task_rq_lock(p->parent, &flags);
Oleg Nesterov889dfaf2005-11-04 18:54:30 +03001795 if (p->first_time_slice && task_cpu(p) == task_cpu(p->parent)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001796 p->parent->time_slice += p->time_slice;
1797 if (unlikely(p->parent->time_slice > task_timeslice(p)))
1798 p->parent->time_slice = task_timeslice(p);
1799 }
1800 if (p->sleep_avg < p->parent->sleep_avg)
1801 p->parent->sleep_avg = p->parent->sleep_avg /
1802 (EXIT_WEIGHT + 1) * EXIT_WEIGHT + p->sleep_avg /
1803 (EXIT_WEIGHT + 1);
1804 task_rq_unlock(rq, &flags);
1805}
1806
1807/**
Nick Piggin4866cde2005-06-25 14:57:23 -07001808 * prepare_task_switch - prepare to switch tasks
1809 * @rq: the runqueue preparing to switch
1810 * @next: the task we are going to switch to.
1811 *
1812 * This is called with the rq lock held and interrupts off. It must
1813 * be paired with a subsequent finish_task_switch after the context
1814 * switch.
1815 *
1816 * prepare_task_switch sets up locking and calls architecture specific
1817 * hooks.
1818 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07001819static inline void prepare_task_switch(struct rq *rq, struct task_struct *next)
Nick Piggin4866cde2005-06-25 14:57:23 -07001820{
1821 prepare_lock_switch(rq, next);
1822 prepare_arch_switch(next);
1823}
1824
1825/**
Linus Torvalds1da177e2005-04-16 15:20:36 -07001826 * finish_task_switch - clean up after a task-switch
Jeff Garzik344baba2005-09-07 01:15:17 -04001827 * @rq: runqueue associated with task-switch
Linus Torvalds1da177e2005-04-16 15:20:36 -07001828 * @prev: the thread we just switched away from.
1829 *
Nick Piggin4866cde2005-06-25 14:57:23 -07001830 * finish_task_switch must be called after the context switch, paired
1831 * with a prepare_task_switch call before the context switch.
1832 * finish_task_switch will reconcile locking set up by prepare_task_switch,
1833 * and do any other architecture-specific cleanup actions.
Linus Torvalds1da177e2005-04-16 15:20:36 -07001834 *
1835 * Note that we may have delayed dropping an mm in context_switch(). If
1836 * so, we finish that here outside of the runqueue lock. (Doing it
1837 * with the lock held can cause deadlocks; see schedule() for
1838 * details.)
1839 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07001840static inline void finish_task_switch(struct rq *rq, struct task_struct *prev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001841 __releases(rq->lock)
1842{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001843 struct mm_struct *mm = rq->prev_mm;
Oleg Nesterov55a101f2006-09-29 02:01:10 -07001844 long prev_state;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001845
1846 rq->prev_mm = NULL;
1847
1848 /*
1849 * A task struct has one reference for the use as "current".
Oleg Nesterovc394cc92006-09-29 02:01:11 -07001850 * If a task dies, then it sets TASK_DEAD in tsk->state and calls
Oleg Nesterov55a101f2006-09-29 02:01:10 -07001851 * schedule one last time. The schedule call will never return, and
1852 * the scheduled task must drop that reference.
Oleg Nesterovc394cc92006-09-29 02:01:11 -07001853 * The test for TASK_DEAD must occur while the runqueue locks are
Linus Torvalds1da177e2005-04-16 15:20:36 -07001854 * still held, otherwise prev could be scheduled on another cpu, die
1855 * there before we look at prev->state, and then the reference would
1856 * be dropped twice.
1857 * Manfred Spraul <manfred@colorfullife.com>
1858 */
Oleg Nesterov55a101f2006-09-29 02:01:10 -07001859 prev_state = prev->state;
Nick Piggin4866cde2005-06-25 14:57:23 -07001860 finish_arch_switch(prev);
1861 finish_lock_switch(rq, prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001862 if (mm)
1863 mmdrop(mm);
Oleg Nesterovc394cc92006-09-29 02:01:11 -07001864 if (unlikely(prev_state == TASK_DEAD)) {
bibo maoc6fd91f2006-03-26 01:38:20 -08001865 /*
1866 * Remove function-return probe instances associated with this
1867 * task and put them back on the free list.
1868 */
1869 kprobe_flush_task(prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001870 put_task_struct(prev);
bibo maoc6fd91f2006-03-26 01:38:20 -08001871 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001872}
1873
1874/**
1875 * schedule_tail - first thing a freshly forked thread must call.
1876 * @prev: the thread we just switched away from.
1877 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001878asmlinkage void schedule_tail(struct task_struct *prev)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001879 __releases(rq->lock)
1880{
Ingo Molnar70b97a72006-07-03 00:25:42 -07001881 struct rq *rq = this_rq();
1882
Nick Piggin4866cde2005-06-25 14:57:23 -07001883 finish_task_switch(rq, prev);
1884#ifdef __ARCH_WANT_UNLOCKED_CTXSW
1885 /* In this case, finish_task_switch does not reenable preemption */
1886 preempt_enable();
1887#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07001888 if (current->set_child_tid)
1889 put_user(current->pid, current->set_child_tid);
1890}
1891
1892/*
1893 * context_switch - switch to the new MM and the new
1894 * thread's register state.
1895 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07001896static inline struct task_struct *
Ingo Molnar70b97a72006-07-03 00:25:42 -07001897context_switch(struct rq *rq, struct task_struct *prev,
Ingo Molnar36c8b582006-07-03 00:25:41 -07001898 struct task_struct *next)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001899{
1900 struct mm_struct *mm = next->mm;
1901 struct mm_struct *oldmm = prev->active_mm;
1902
Zachary Amsden9226d122007-02-13 13:26:21 +01001903 /*
1904 * For paravirt, this is coupled with an exit in switch_to to
1905 * combine the page table reload and the switch backend into
1906 * one hypercall.
1907 */
1908 arch_enter_lazy_cpu_mode();
1909
Nick Pigginbeed33a2006-10-11 01:21:52 -07001910 if (!mm) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001911 next->active_mm = oldmm;
1912 atomic_inc(&oldmm->mm_count);
1913 enter_lazy_tlb(oldmm, next);
1914 } else
1915 switch_mm(oldmm, mm, next);
1916
Nick Pigginbeed33a2006-10-11 01:21:52 -07001917 if (!prev->mm) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001918 prev->active_mm = NULL;
1919 WARN_ON(rq->prev_mm);
1920 rq->prev_mm = oldmm;
1921 }
Ingo Molnar3a5f5e42006-07-14 00:24:27 -07001922 /*
1923 * Since the runqueue lock will be released by the next
1924 * task (which is an invalid locking op but in the case
1925 * of the scheduler it's an obvious special-case), so we
1926 * do an early lockdep release here:
1927 */
1928#ifndef __ARCH_WANT_UNLOCKED_CTXSW
Ingo Molnar8a25d5d2006-07-03 00:24:54 -07001929 spin_release(&rq->lock.dep_map, 1, _THIS_IP_);
Ingo Molnar3a5f5e42006-07-14 00:24:27 -07001930#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07001931
1932 /* Here we just switch the register state and the stack. */
1933 switch_to(prev, next, prev);
1934
1935 return prev;
1936}
1937
1938/*
1939 * nr_running, nr_uninterruptible and nr_context_switches:
1940 *
1941 * externally visible scheduler statistics: current number of runnable
1942 * threads, current number of uninterruptible-sleeping threads, total
1943 * number of context switches performed since bootup.
1944 */
1945unsigned long nr_running(void)
1946{
1947 unsigned long i, sum = 0;
1948
1949 for_each_online_cpu(i)
1950 sum += cpu_rq(i)->nr_running;
1951
1952 return sum;
1953}
1954
1955unsigned long nr_uninterruptible(void)
1956{
1957 unsigned long i, sum = 0;
1958
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08001959 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001960 sum += cpu_rq(i)->nr_uninterruptible;
1961
1962 /*
1963 * Since we read the counters lockless, it might be slightly
1964 * inaccurate. Do not allow it to go below zero though:
1965 */
1966 if (unlikely((long)sum < 0))
1967 sum = 0;
1968
1969 return sum;
1970}
1971
1972unsigned long long nr_context_switches(void)
1973{
Steven Rostedtcc94abf2006-06-27 02:54:31 -07001974 int i;
1975 unsigned long long sum = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001976
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08001977 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001978 sum += cpu_rq(i)->nr_switches;
1979
1980 return sum;
1981}
1982
1983unsigned long nr_iowait(void)
1984{
1985 unsigned long i, sum = 0;
1986
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08001987 for_each_possible_cpu(i)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001988 sum += atomic_read(&cpu_rq(i)->nr_iowait);
1989
1990 return sum;
1991}
1992
Jack Steinerdb1b1fe2006-03-31 02:31:21 -08001993unsigned long nr_active(void)
1994{
1995 unsigned long i, running = 0, uninterruptible = 0;
1996
1997 for_each_online_cpu(i) {
1998 running += cpu_rq(i)->nr_running;
1999 uninterruptible += cpu_rq(i)->nr_uninterruptible;
2000 }
2001
2002 if (unlikely((long)uninterruptible < 0))
2003 uninterruptible = 0;
2004
2005 return running + uninterruptible;
2006}
2007
Linus Torvalds1da177e2005-04-16 15:20:36 -07002008#ifdef CONFIG_SMP
2009
2010/*
Ingo Molnar48f24c42006-07-03 00:25:40 -07002011 * Is this task likely cache-hot:
2012 */
2013static inline int
2014task_hot(struct task_struct *p, unsigned long long now, struct sched_domain *sd)
2015{
2016 return (long long)(now - p->last_ran) < (long long)sd->cache_hot_time;
2017}
2018
2019/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07002020 * double_rq_lock - safely lock two runqueues
2021 *
2022 * Note this does not disable interrupts like task_rq_lock,
2023 * you need to do so manually before calling.
2024 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002025static void double_rq_lock(struct rq *rq1, struct rq *rq2)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002026 __acquires(rq1->lock)
2027 __acquires(rq2->lock)
2028{
Kirill Korotaev054b9102006-12-10 02:20:11 -08002029 BUG_ON(!irqs_disabled());
Linus Torvalds1da177e2005-04-16 15:20:36 -07002030 if (rq1 == rq2) {
2031 spin_lock(&rq1->lock);
2032 __acquire(rq2->lock); /* Fake it out ;) */
2033 } else {
Chen, Kenneth Wc96d1452006-06-27 02:54:28 -07002034 if (rq1 < rq2) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002035 spin_lock(&rq1->lock);
2036 spin_lock(&rq2->lock);
2037 } else {
2038 spin_lock(&rq2->lock);
2039 spin_lock(&rq1->lock);
2040 }
2041 }
2042}
2043
2044/*
2045 * double_rq_unlock - safely unlock two runqueues
2046 *
2047 * Note this does not restore interrupts like task_rq_unlock,
2048 * you need to do so manually after calling.
2049 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002050static void double_rq_unlock(struct rq *rq1, struct rq *rq2)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002051 __releases(rq1->lock)
2052 __releases(rq2->lock)
2053{
2054 spin_unlock(&rq1->lock);
2055 if (rq1 != rq2)
2056 spin_unlock(&rq2->lock);
2057 else
2058 __release(rq2->lock);
2059}
2060
2061/*
2062 * double_lock_balance - lock the busiest runqueue, this_rq is locked already.
2063 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002064static void double_lock_balance(struct rq *this_rq, struct rq *busiest)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002065 __releases(this_rq->lock)
2066 __acquires(busiest->lock)
2067 __acquires(this_rq->lock)
2068{
Kirill Korotaev054b9102006-12-10 02:20:11 -08002069 if (unlikely(!irqs_disabled())) {
2070 /* printk() doesn't work good under rq->lock */
2071 spin_unlock(&this_rq->lock);
2072 BUG_ON(1);
2073 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002074 if (unlikely(!spin_trylock(&busiest->lock))) {
Chen, Kenneth Wc96d1452006-06-27 02:54:28 -07002075 if (busiest < this_rq) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002076 spin_unlock(&this_rq->lock);
2077 spin_lock(&busiest->lock);
2078 spin_lock(&this_rq->lock);
2079 } else
2080 spin_lock(&busiest->lock);
2081 }
2082}
2083
2084/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07002085 * If dest_cpu is allowed for this process, migrate the task to it.
2086 * This is accomplished by forcing the cpu_allowed mask to only
2087 * allow dest_cpu, which will force the cpu onto dest_cpu. Then
2088 * the cpu_allowed mask is restored.
2089 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07002090static void sched_migrate_task(struct task_struct *p, int dest_cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002091{
Ingo Molnar70b97a72006-07-03 00:25:42 -07002092 struct migration_req req;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002093 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07002094 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002095
2096 rq = task_rq_lock(p, &flags);
2097 if (!cpu_isset(dest_cpu, p->cpus_allowed)
2098 || unlikely(cpu_is_offline(dest_cpu)))
2099 goto out;
2100
2101 /* force the process onto the specified CPU */
2102 if (migrate_task(p, dest_cpu, &req)) {
2103 /* Need to wait for migration thread (might exit: take ref). */
2104 struct task_struct *mt = rq->migration_thread;
Ingo Molnar36c8b582006-07-03 00:25:41 -07002105
Linus Torvalds1da177e2005-04-16 15:20:36 -07002106 get_task_struct(mt);
2107 task_rq_unlock(rq, &flags);
2108 wake_up_process(mt);
2109 put_task_struct(mt);
2110 wait_for_completion(&req.done);
Ingo Molnar36c8b582006-07-03 00:25:41 -07002111
Linus Torvalds1da177e2005-04-16 15:20:36 -07002112 return;
2113 }
2114out:
2115 task_rq_unlock(rq, &flags);
2116}
2117
2118/*
Nick Piggin476d1392005-06-25 14:57:29 -07002119 * sched_exec - execve() is a valuable balancing opportunity, because at
2120 * this point the task has the smallest effective memory and cache footprint.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002121 */
2122void sched_exec(void)
2123{
Linus Torvalds1da177e2005-04-16 15:20:36 -07002124 int new_cpu, this_cpu = get_cpu();
Nick Piggin476d1392005-06-25 14:57:29 -07002125 new_cpu = sched_balance_self(this_cpu, SD_BALANCE_EXEC);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002126 put_cpu();
Nick Piggin476d1392005-06-25 14:57:29 -07002127 if (new_cpu != this_cpu)
2128 sched_migrate_task(current, new_cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002129}
2130
2131/*
2132 * pull_task - move a task from a remote runqueue to the local runqueue.
2133 * Both runqueues must be locked.
2134 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002135static void pull_task(struct rq *src_rq, struct prio_array *src_array,
2136 struct task_struct *p, struct rq *this_rq,
2137 struct prio_array *this_array, int this_cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002138{
2139 dequeue_task(p, src_array);
Peter Williams2dd73a42006-06-27 02:54:34 -07002140 dec_nr_running(p, src_rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002141 set_task_cpu(p, this_cpu);
Peter Williams2dd73a42006-06-27 02:54:34 -07002142 inc_nr_running(p, this_rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002143 enqueue_task(p, this_array);
Mike Galbraithb18ec802006-12-10 02:20:31 -08002144 p->timestamp = (p->timestamp - src_rq->most_recent_timestamp)
2145 + this_rq->most_recent_timestamp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002146 /*
2147 * Note that idle threads have a prio of MAX_PRIO, for this test
2148 * to be always true for them.
2149 */
2150 if (TASK_PREEMPTS_CURR(p, this_rq))
2151 resched_task(this_rq->curr);
2152}
2153
2154/*
2155 * can_migrate_task - may task p from runqueue rq be migrated to this_cpu?
2156 */
Arjan van de Ven858119e2006-01-14 13:20:43 -08002157static
Ingo Molnar70b97a72006-07-03 00:25:42 -07002158int can_migrate_task(struct task_struct *p, struct rq *rq, int this_cpu,
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07002159 struct sched_domain *sd, enum idle_type idle,
2160 int *all_pinned)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002161{
2162 /*
2163 * We do not migrate tasks that are:
2164 * 1) running (obviously), or
2165 * 2) cannot be migrated to this CPU due to cpus_allowed, or
2166 * 3) are cache-hot on their current CPU.
2167 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07002168 if (!cpu_isset(this_cpu, p->cpus_allowed))
2169 return 0;
Nick Piggin81026792005-06-25 14:57:07 -07002170 *all_pinned = 0;
2171
2172 if (task_running(rq, p))
2173 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002174
2175 /*
2176 * Aggressive migration if:
Nick Piggincafb20c2005-06-25 14:57:17 -07002177 * 1) task is cache cold, or
Linus Torvalds1da177e2005-04-16 15:20:36 -07002178 * 2) too many balance attempts have failed.
2179 */
2180
Mike Galbraithb18ec802006-12-10 02:20:31 -08002181 if (sd->nr_balance_failed > sd->cache_nice_tries) {
2182#ifdef CONFIG_SCHEDSTATS
2183 if (task_hot(p, rq->most_recent_timestamp, sd))
2184 schedstat_inc(sd, lb_hot_gained[idle]);
2185#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002186 return 1;
Mike Galbraithb18ec802006-12-10 02:20:31 -08002187 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002188
Mike Galbraithb18ec802006-12-10 02:20:31 -08002189 if (task_hot(p, rq->most_recent_timestamp, sd))
Nick Piggin81026792005-06-25 14:57:07 -07002190 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002191 return 1;
2192}
2193
Peter Williams615052d2006-06-27 02:54:37 -07002194#define rq_best_prio(rq) min((rq)->curr->prio, (rq)->best_expired_prio)
Ingo Molnar48f24c42006-07-03 00:25:40 -07002195
Linus Torvalds1da177e2005-04-16 15:20:36 -07002196/*
Peter Williams2dd73a42006-06-27 02:54:34 -07002197 * move_tasks tries to move up to max_nr_move tasks and max_load_move weighted
2198 * load from busiest to this_rq, as part of a balancing operation within
2199 * "domain". Returns the number of tasks moved.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002200 *
2201 * Called with both runqueues locked.
2202 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002203static int move_tasks(struct rq *this_rq, int this_cpu, struct rq *busiest,
Peter Williams2dd73a42006-06-27 02:54:34 -07002204 unsigned long max_nr_move, unsigned long max_load_move,
2205 struct sched_domain *sd, enum idle_type idle,
2206 int *all_pinned)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002207{
Ingo Molnar48f24c42006-07-03 00:25:40 -07002208 int idx, pulled = 0, pinned = 0, this_best_prio, best_prio,
2209 best_prio_seen, skip_for_load;
Ingo Molnar70b97a72006-07-03 00:25:42 -07002210 struct prio_array *array, *dst_array;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002211 struct list_head *head, *curr;
Ingo Molnar36c8b582006-07-03 00:25:41 -07002212 struct task_struct *tmp;
Peter Williams2dd73a42006-06-27 02:54:34 -07002213 long rem_load_move;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002214
Peter Williams2dd73a42006-06-27 02:54:34 -07002215 if (max_nr_move == 0 || max_load_move == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002216 goto out;
2217
Peter Williams2dd73a42006-06-27 02:54:34 -07002218 rem_load_move = max_load_move;
Nick Piggin81026792005-06-25 14:57:07 -07002219 pinned = 1;
Peter Williams615052d2006-06-27 02:54:37 -07002220 this_best_prio = rq_best_prio(this_rq);
Ingo Molnar48f24c42006-07-03 00:25:40 -07002221 best_prio = rq_best_prio(busiest);
Peter Williams615052d2006-06-27 02:54:37 -07002222 /*
2223 * Enable handling of the case where there is more than one task
2224 * with the best priority. If the current running task is one
Ingo Molnar48f24c42006-07-03 00:25:40 -07002225 * of those with prio==best_prio we know it won't be moved
Peter Williams615052d2006-06-27 02:54:37 -07002226 * and therefore it's safe to override the skip (based on load) of
2227 * any task we find with that prio.
2228 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07002229 best_prio_seen = best_prio == busiest->curr->prio;
Nick Piggin81026792005-06-25 14:57:07 -07002230
Linus Torvalds1da177e2005-04-16 15:20:36 -07002231 /*
2232 * We first consider expired tasks. Those will likely not be
2233 * executed in the near future, and they are most likely to
2234 * be cache-cold, thus switching CPUs has the least effect
2235 * on them.
2236 */
2237 if (busiest->expired->nr_active) {
2238 array = busiest->expired;
2239 dst_array = this_rq->expired;
2240 } else {
2241 array = busiest->active;
2242 dst_array = this_rq->active;
2243 }
2244
2245new_array:
2246 /* Start searching at priority 0: */
2247 idx = 0;
2248skip_bitmap:
2249 if (!idx)
2250 idx = sched_find_first_bit(array->bitmap);
2251 else
2252 idx = find_next_bit(array->bitmap, MAX_PRIO, idx);
2253 if (idx >= MAX_PRIO) {
2254 if (array == busiest->expired && busiest->active->nr_active) {
2255 array = busiest->active;
2256 dst_array = this_rq->active;
2257 goto new_array;
2258 }
2259 goto out;
2260 }
2261
2262 head = array->queue + idx;
2263 curr = head->prev;
2264skip_queue:
Ingo Molnar36c8b582006-07-03 00:25:41 -07002265 tmp = list_entry(curr, struct task_struct, run_list);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002266
2267 curr = curr->prev;
2268
Peter Williams50ddd962006-06-27 02:54:36 -07002269 /*
2270 * To help distribute high priority tasks accross CPUs we don't
2271 * skip a task if it will be the highest priority task (i.e. smallest
2272 * prio value) on its new queue regardless of its load weight
2273 */
Peter Williams615052d2006-06-27 02:54:37 -07002274 skip_for_load = tmp->load_weight > rem_load_move;
2275 if (skip_for_load && idx < this_best_prio)
Ingo Molnar48f24c42006-07-03 00:25:40 -07002276 skip_for_load = !best_prio_seen && idx == best_prio;
Peter Williams615052d2006-06-27 02:54:37 -07002277 if (skip_for_load ||
Peter Williams2dd73a42006-06-27 02:54:34 -07002278 !can_migrate_task(tmp, busiest, this_cpu, sd, idle, &pinned)) {
Ingo Molnar48f24c42006-07-03 00:25:40 -07002279
2280 best_prio_seen |= idx == best_prio;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002281 if (curr != head)
2282 goto skip_queue;
2283 idx++;
2284 goto skip_bitmap;
2285 }
2286
Linus Torvalds1da177e2005-04-16 15:20:36 -07002287 pull_task(busiest, array, tmp, this_rq, dst_array, this_cpu);
2288 pulled++;
Peter Williams2dd73a42006-06-27 02:54:34 -07002289 rem_load_move -= tmp->load_weight;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002290
Peter Williams2dd73a42006-06-27 02:54:34 -07002291 /*
2292 * We only want to steal up to the prescribed number of tasks
2293 * and the prescribed amount of weighted load.
2294 */
2295 if (pulled < max_nr_move && rem_load_move > 0) {
Peter Williams615052d2006-06-27 02:54:37 -07002296 if (idx < this_best_prio)
2297 this_best_prio = idx;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002298 if (curr != head)
2299 goto skip_queue;
2300 idx++;
2301 goto skip_bitmap;
2302 }
2303out:
2304 /*
2305 * Right now, this is the only place pull_task() is called,
2306 * so we can safely collect pull_task() stats here rather than
2307 * inside pull_task().
2308 */
2309 schedstat_add(sd, lb_gained[idle], pulled);
Nick Piggin81026792005-06-25 14:57:07 -07002310
2311 if (all_pinned)
2312 *all_pinned = pinned;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002313 return pulled;
2314}
2315
2316/*
2317 * find_busiest_group finds and returns the busiest CPU group within the
Ingo Molnar48f24c42006-07-03 00:25:40 -07002318 * domain. It calculates and returns the amount of weighted load which
2319 * should be moved to restore balance via the imbalance parameter.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002320 */
2321static struct sched_group *
2322find_busiest_group(struct sched_domain *sd, int this_cpu,
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002323 unsigned long *imbalance, enum idle_type idle, int *sd_idle,
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002324 cpumask_t *cpus, int *balance)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002325{
2326 struct sched_group *busiest = NULL, *this = NULL, *group = sd->groups;
2327 unsigned long max_load, avg_load, total_load, this_load, total_pwr;
Siddha, Suresh B0c117f12005-09-10 00:26:21 -07002328 unsigned long max_pull;
Peter Williams2dd73a42006-06-27 02:54:34 -07002329 unsigned long busiest_load_per_task, busiest_nr_running;
2330 unsigned long this_load_per_task, this_nr_running;
Nick Piggin78979862005-06-25 14:57:13 -07002331 int load_idx;
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002332#if defined(CONFIG_SCHED_MC) || defined(CONFIG_SCHED_SMT)
2333 int power_savings_balance = 1;
2334 unsigned long leader_nr_running = 0, min_load_per_task = 0;
2335 unsigned long min_nr_running = ULONG_MAX;
2336 struct sched_group *group_min = NULL, *group_leader = NULL;
2337#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002338
2339 max_load = this_load = total_load = total_pwr = 0;
Peter Williams2dd73a42006-06-27 02:54:34 -07002340 busiest_load_per_task = busiest_nr_running = 0;
2341 this_load_per_task = this_nr_running = 0;
Nick Piggin78979862005-06-25 14:57:13 -07002342 if (idle == NOT_IDLE)
2343 load_idx = sd->busy_idx;
2344 else if (idle == NEWLY_IDLE)
2345 load_idx = sd->newidle_idx;
2346 else
2347 load_idx = sd->idle_idx;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002348
2349 do {
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002350 unsigned long load, group_capacity;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002351 int local_group;
2352 int i;
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002353 unsigned int balance_cpu = -1, first_idle_cpu = 0;
Peter Williams2dd73a42006-06-27 02:54:34 -07002354 unsigned long sum_nr_running, sum_weighted_load;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002355
2356 local_group = cpu_isset(this_cpu, group->cpumask);
2357
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002358 if (local_group)
2359 balance_cpu = first_cpu(group->cpumask);
2360
Linus Torvalds1da177e2005-04-16 15:20:36 -07002361 /* Tally up the load of all CPUs in the group */
Peter Williams2dd73a42006-06-27 02:54:34 -07002362 sum_weighted_load = sum_nr_running = avg_load = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002363
2364 for_each_cpu_mask(i, group->cpumask) {
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002365 struct rq *rq;
2366
2367 if (!cpu_isset(i, *cpus))
2368 continue;
2369
2370 rq = cpu_rq(i);
Peter Williams2dd73a42006-06-27 02:54:34 -07002371
Nick Piggin5969fe02005-09-10 00:26:19 -07002372 if (*sd_idle && !idle_cpu(i))
2373 *sd_idle = 0;
2374
Linus Torvalds1da177e2005-04-16 15:20:36 -07002375 /* Bias balancing toward cpus of our domain */
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002376 if (local_group) {
2377 if (idle_cpu(i) && !first_idle_cpu) {
2378 first_idle_cpu = 1;
2379 balance_cpu = i;
2380 }
2381
Nick Piggina2000572006-02-10 01:51:02 -08002382 load = target_load(i, load_idx);
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002383 } else
Nick Piggina2000572006-02-10 01:51:02 -08002384 load = source_load(i, load_idx);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002385
2386 avg_load += load;
Peter Williams2dd73a42006-06-27 02:54:34 -07002387 sum_nr_running += rq->nr_running;
2388 sum_weighted_load += rq->raw_weighted_load;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002389 }
2390
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002391 /*
2392 * First idle cpu or the first cpu(busiest) in this sched group
2393 * is eligible for doing load balancing at this and above
2394 * domains.
2395 */
2396 if (local_group && balance_cpu != this_cpu && balance) {
2397 *balance = 0;
2398 goto ret;
2399 }
2400
Linus Torvalds1da177e2005-04-16 15:20:36 -07002401 total_load += avg_load;
Eric Dumazet5517d862007-05-08 00:32:57 -07002402 total_pwr += group->__cpu_power;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002403
2404 /* Adjust by relative CPU power of the group */
Eric Dumazet5517d862007-05-08 00:32:57 -07002405 avg_load = sg_div_cpu_power(group,
2406 avg_load * SCHED_LOAD_SCALE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002407
Eric Dumazet5517d862007-05-08 00:32:57 -07002408 group_capacity = group->__cpu_power / SCHED_LOAD_SCALE;
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002409
Linus Torvalds1da177e2005-04-16 15:20:36 -07002410 if (local_group) {
2411 this_load = avg_load;
2412 this = group;
Peter Williams2dd73a42006-06-27 02:54:34 -07002413 this_nr_running = sum_nr_running;
2414 this_load_per_task = sum_weighted_load;
2415 } else if (avg_load > max_load &&
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002416 sum_nr_running > group_capacity) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002417 max_load = avg_load;
2418 busiest = group;
Peter Williams2dd73a42006-06-27 02:54:34 -07002419 busiest_nr_running = sum_nr_running;
2420 busiest_load_per_task = sum_weighted_load;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002421 }
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002422
2423#if defined(CONFIG_SCHED_MC) || defined(CONFIG_SCHED_SMT)
2424 /*
2425 * Busy processors will not participate in power savings
2426 * balance.
2427 */
2428 if (idle == NOT_IDLE || !(sd->flags & SD_POWERSAVINGS_BALANCE))
2429 goto group_next;
2430
2431 /*
2432 * If the local group is idle or completely loaded
2433 * no need to do power savings balance at this domain
2434 */
2435 if (local_group && (this_nr_running >= group_capacity ||
2436 !this_nr_running))
2437 power_savings_balance = 0;
2438
2439 /*
2440 * If a group is already running at full capacity or idle,
2441 * don't include that group in power savings calculations
2442 */
2443 if (!power_savings_balance || sum_nr_running >= group_capacity
2444 || !sum_nr_running)
2445 goto group_next;
2446
2447 /*
2448 * Calculate the group which has the least non-idle load.
2449 * This is the group from where we need to pick up the load
2450 * for saving power
2451 */
2452 if ((sum_nr_running < min_nr_running) ||
2453 (sum_nr_running == min_nr_running &&
2454 first_cpu(group->cpumask) <
2455 first_cpu(group_min->cpumask))) {
2456 group_min = group;
2457 min_nr_running = sum_nr_running;
2458 min_load_per_task = sum_weighted_load /
2459 sum_nr_running;
2460 }
2461
2462 /*
2463 * Calculate the group which is almost near its
2464 * capacity but still has some space to pick up some load
2465 * from other group and save more power
2466 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07002467 if (sum_nr_running <= group_capacity - 1) {
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002468 if (sum_nr_running > leader_nr_running ||
2469 (sum_nr_running == leader_nr_running &&
2470 first_cpu(group->cpumask) >
2471 first_cpu(group_leader->cpumask))) {
2472 group_leader = group;
2473 leader_nr_running = sum_nr_running;
2474 }
Ingo Molnar48f24c42006-07-03 00:25:40 -07002475 }
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002476group_next:
2477#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07002478 group = group->next;
2479 } while (group != sd->groups);
2480
Peter Williams2dd73a42006-06-27 02:54:34 -07002481 if (!busiest || this_load >= max_load || busiest_nr_running == 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002482 goto out_balanced;
2483
2484 avg_load = (SCHED_LOAD_SCALE * total_load) / total_pwr;
2485
2486 if (this_load >= avg_load ||
2487 100*max_load <= sd->imbalance_pct*this_load)
2488 goto out_balanced;
2489
Peter Williams2dd73a42006-06-27 02:54:34 -07002490 busiest_load_per_task /= busiest_nr_running;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002491 /*
2492 * We're trying to get all the cpus to the average_load, so we don't
2493 * want to push ourselves above the average load, nor do we wish to
2494 * reduce the max loaded cpu below the average load, as either of these
2495 * actions would just result in more rebalancing later, and ping-pong
2496 * tasks around. Thus we look for the minimum possible imbalance.
2497 * Negative imbalances (*we* are more loaded than anyone else) will
2498 * be counted as no imbalance for these purposes -- we can't fix that
2499 * by pulling tasks to us. Be careful of negative numbers as they'll
2500 * appear as very large values with unsigned longs.
2501 */
Peter Williams2dd73a42006-06-27 02:54:34 -07002502 if (max_load <= busiest_load_per_task)
2503 goto out_balanced;
2504
2505 /*
2506 * In the presence of smp nice balancing, certain scenarios can have
2507 * max load less than avg load(as we skip the groups at or below
2508 * its cpu_power, while calculating max_load..)
2509 */
2510 if (max_load < avg_load) {
2511 *imbalance = 0;
2512 goto small_imbalance;
2513 }
Siddha, Suresh B0c117f12005-09-10 00:26:21 -07002514
2515 /* Don't want to pull so many tasks that a group would go idle */
Peter Williams2dd73a42006-06-27 02:54:34 -07002516 max_pull = min(max_load - avg_load, max_load - busiest_load_per_task);
Siddha, Suresh B0c117f12005-09-10 00:26:21 -07002517
Linus Torvalds1da177e2005-04-16 15:20:36 -07002518 /* How much load to actually move to equalise the imbalance */
Eric Dumazet5517d862007-05-08 00:32:57 -07002519 *imbalance = min(max_pull * busiest->__cpu_power,
2520 (avg_load - this_load) * this->__cpu_power)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002521 / SCHED_LOAD_SCALE;
2522
Peter Williams2dd73a42006-06-27 02:54:34 -07002523 /*
2524 * if *imbalance is less than the average load per runnable task
2525 * there is no gaurantee that any tasks will be moved so we'll have
2526 * a think about bumping its value to force at least one task to be
2527 * moved
2528 */
2529 if (*imbalance < busiest_load_per_task) {
Ingo Molnar48f24c42006-07-03 00:25:40 -07002530 unsigned long tmp, pwr_now, pwr_move;
Peter Williams2dd73a42006-06-27 02:54:34 -07002531 unsigned int imbn;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002532
Peter Williams2dd73a42006-06-27 02:54:34 -07002533small_imbalance:
2534 pwr_move = pwr_now = 0;
2535 imbn = 2;
2536 if (this_nr_running) {
2537 this_load_per_task /= this_nr_running;
2538 if (busiest_load_per_task > this_load_per_task)
2539 imbn = 1;
2540 } else
2541 this_load_per_task = SCHED_LOAD_SCALE;
2542
2543 if (max_load - this_load >= busiest_load_per_task * imbn) {
2544 *imbalance = busiest_load_per_task;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002545 return busiest;
2546 }
2547
2548 /*
2549 * OK, we don't have enough imbalance to justify moving tasks,
2550 * however we may be able to increase total CPU power used by
2551 * moving them.
2552 */
2553
Eric Dumazet5517d862007-05-08 00:32:57 -07002554 pwr_now += busiest->__cpu_power *
2555 min(busiest_load_per_task, max_load);
2556 pwr_now += this->__cpu_power *
2557 min(this_load_per_task, this_load);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002558 pwr_now /= SCHED_LOAD_SCALE;
2559
2560 /* Amount of load we'd subtract */
Eric Dumazet5517d862007-05-08 00:32:57 -07002561 tmp = sg_div_cpu_power(busiest,
2562 busiest_load_per_task * SCHED_LOAD_SCALE);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002563 if (max_load > tmp)
Eric Dumazet5517d862007-05-08 00:32:57 -07002564 pwr_move += busiest->__cpu_power *
Peter Williams2dd73a42006-06-27 02:54:34 -07002565 min(busiest_load_per_task, max_load - tmp);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002566
2567 /* Amount of load we'd add */
Eric Dumazet5517d862007-05-08 00:32:57 -07002568 if (max_load * busiest->__cpu_power <
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08002569 busiest_load_per_task * SCHED_LOAD_SCALE)
Eric Dumazet5517d862007-05-08 00:32:57 -07002570 tmp = sg_div_cpu_power(this,
2571 max_load * busiest->__cpu_power);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002572 else
Eric Dumazet5517d862007-05-08 00:32:57 -07002573 tmp = sg_div_cpu_power(this,
2574 busiest_load_per_task * SCHED_LOAD_SCALE);
2575 pwr_move += this->__cpu_power *
2576 min(this_load_per_task, this_load + tmp);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002577 pwr_move /= SCHED_LOAD_SCALE;
2578
2579 /* Move if we gain throughput */
2580 if (pwr_move <= pwr_now)
2581 goto out_balanced;
2582
Peter Williams2dd73a42006-06-27 02:54:34 -07002583 *imbalance = busiest_load_per_task;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002584 }
2585
Linus Torvalds1da177e2005-04-16 15:20:36 -07002586 return busiest;
2587
2588out_balanced:
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002589#if defined(CONFIG_SCHED_MC) || defined(CONFIG_SCHED_SMT)
2590 if (idle == NOT_IDLE || !(sd->flags & SD_POWERSAVINGS_BALANCE))
2591 goto ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002592
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002593 if (this == group_leader && group_leader != group_min) {
2594 *imbalance = min_load_per_task;
2595 return group_min;
2596 }
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002597#endif
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002598ret:
Linus Torvalds1da177e2005-04-16 15:20:36 -07002599 *imbalance = 0;
2600 return NULL;
2601}
2602
2603/*
2604 * find_busiest_queue - find the busiest runqueue among the cpus in group.
2605 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002606static struct rq *
Ingo Molnar48f24c42006-07-03 00:25:40 -07002607find_busiest_queue(struct sched_group *group, enum idle_type idle,
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002608 unsigned long imbalance, cpumask_t *cpus)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002609{
Ingo Molnar70b97a72006-07-03 00:25:42 -07002610 struct rq *busiest = NULL, *rq;
Peter Williams2dd73a42006-06-27 02:54:34 -07002611 unsigned long max_load = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002612 int i;
2613
2614 for_each_cpu_mask(i, group->cpumask) {
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002615
2616 if (!cpu_isset(i, *cpus))
2617 continue;
2618
Ingo Molnar48f24c42006-07-03 00:25:40 -07002619 rq = cpu_rq(i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002620
Ingo Molnar48f24c42006-07-03 00:25:40 -07002621 if (rq->nr_running == 1 && rq->raw_weighted_load > imbalance)
Peter Williams2dd73a42006-06-27 02:54:34 -07002622 continue;
2623
Ingo Molnar48f24c42006-07-03 00:25:40 -07002624 if (rq->raw_weighted_load > max_load) {
2625 max_load = rq->raw_weighted_load;
2626 busiest = rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002627 }
2628 }
2629
2630 return busiest;
2631}
2632
2633/*
Nick Piggin77391d72005-06-25 14:57:30 -07002634 * Max backoff if we encounter pinned tasks. Pretty arbitrary value, but
2635 * so long as it is large enough.
2636 */
2637#define MAX_PINNED_INTERVAL 512
2638
Ingo Molnar48f24c42006-07-03 00:25:40 -07002639static inline unsigned long minus_1_or_zero(unsigned long n)
2640{
2641 return n > 0 ? n - 1 : 0;
2642}
2643
Nick Piggin77391d72005-06-25 14:57:30 -07002644/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07002645 * Check this_cpu to ensure it is balanced within domain. Attempt to move
2646 * tasks if there is an imbalance.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002647 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002648static int load_balance(int this_cpu, struct rq *this_rq,
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002649 struct sched_domain *sd, enum idle_type idle,
2650 int *balance)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002651{
Ingo Molnar48f24c42006-07-03 00:25:40 -07002652 int nr_moved, all_pinned = 0, active_balance = 0, sd_idle = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002653 struct sched_group *group;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002654 unsigned long imbalance;
Ingo Molnar70b97a72006-07-03 00:25:42 -07002655 struct rq *busiest;
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002656 cpumask_t cpus = CPU_MASK_ALL;
Christoph Lameterfe2eea32006-12-10 02:20:21 -08002657 unsigned long flags;
Nick Piggin5969fe02005-09-10 00:26:19 -07002658
Siddha, Suresh B89c47102006-10-03 01:14:09 -07002659 /*
2660 * When power savings policy is enabled for the parent domain, idle
2661 * sibling can pick up load irrespective of busy siblings. In this case,
2662 * let the state of idle sibling percolate up as IDLE, instead of
2663 * portraying it as NOT_IDLE.
2664 */
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002665 if (idle != NOT_IDLE && sd->flags & SD_SHARE_CPUPOWER &&
Siddha, Suresh B89c47102006-10-03 01:14:09 -07002666 !test_sd_parent(sd, SD_POWERSAVINGS_BALANCE))
Nick Piggin5969fe02005-09-10 00:26:19 -07002667 sd_idle = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002668
Linus Torvalds1da177e2005-04-16 15:20:36 -07002669 schedstat_inc(sd, lb_cnt[idle]);
2670
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002671redo:
2672 group = find_busiest_group(sd, this_cpu, &imbalance, idle, &sd_idle,
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002673 &cpus, balance);
2674
Chen, Kenneth W06066712006-12-10 02:20:35 -08002675 if (*balance == 0)
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002676 goto out_balanced;
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002677
Linus Torvalds1da177e2005-04-16 15:20:36 -07002678 if (!group) {
2679 schedstat_inc(sd, lb_nobusyg[idle]);
2680 goto out_balanced;
2681 }
2682
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002683 busiest = find_busiest_queue(group, idle, imbalance, &cpus);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002684 if (!busiest) {
2685 schedstat_inc(sd, lb_nobusyq[idle]);
2686 goto out_balanced;
2687 }
2688
Nick Piggindb935db2005-06-25 14:57:11 -07002689 BUG_ON(busiest == this_rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002690
2691 schedstat_add(sd, lb_imbalance[idle], imbalance);
2692
2693 nr_moved = 0;
2694 if (busiest->nr_running > 1) {
2695 /*
2696 * Attempt to move tasks. If find_busiest_group has found
2697 * an imbalance but busiest->nr_running <= 1, the group is
2698 * still unbalanced. nr_moved simply stays zero, so it is
2699 * correctly treated as an imbalance.
2700 */
Christoph Lameterfe2eea32006-12-10 02:20:21 -08002701 local_irq_save(flags);
Nick Piggine17224b2005-09-10 00:26:18 -07002702 double_rq_lock(this_rq, busiest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002703 nr_moved = move_tasks(this_rq, this_cpu, busiest,
Ingo Molnar48f24c42006-07-03 00:25:40 -07002704 minus_1_or_zero(busiest->nr_running),
2705 imbalance, sd, idle, &all_pinned);
Nick Piggine17224b2005-09-10 00:26:18 -07002706 double_rq_unlock(this_rq, busiest);
Christoph Lameterfe2eea32006-12-10 02:20:21 -08002707 local_irq_restore(flags);
Nick Piggin81026792005-06-25 14:57:07 -07002708
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07002709 /*
2710 * some other cpu did the load balance for us.
2711 */
2712 if (nr_moved && this_cpu != smp_processor_id())
2713 resched_cpu(this_cpu);
2714
Nick Piggin81026792005-06-25 14:57:07 -07002715 /* All tasks on this runqueue were pinned by CPU affinity */
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002716 if (unlikely(all_pinned)) {
2717 cpu_clear(cpu_of(busiest), cpus);
2718 if (!cpus_empty(cpus))
2719 goto redo;
Nick Piggin81026792005-06-25 14:57:07 -07002720 goto out_balanced;
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002721 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002722 }
Nick Piggin81026792005-06-25 14:57:07 -07002723
Linus Torvalds1da177e2005-04-16 15:20:36 -07002724 if (!nr_moved) {
2725 schedstat_inc(sd, lb_failed[idle]);
2726 sd->nr_balance_failed++;
2727
2728 if (unlikely(sd->nr_balance_failed > sd->cache_nice_tries+2)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002729
Christoph Lameterfe2eea32006-12-10 02:20:21 -08002730 spin_lock_irqsave(&busiest->lock, flags);
Siddha, Suresh Bfa3b6dd2005-09-10 00:26:21 -07002731
2732 /* don't kick the migration_thread, if the curr
2733 * task on busiest cpu can't be moved to this_cpu
2734 */
2735 if (!cpu_isset(this_cpu, busiest->curr->cpus_allowed)) {
Christoph Lameterfe2eea32006-12-10 02:20:21 -08002736 spin_unlock_irqrestore(&busiest->lock, flags);
Siddha, Suresh Bfa3b6dd2005-09-10 00:26:21 -07002737 all_pinned = 1;
2738 goto out_one_pinned;
2739 }
2740
Linus Torvalds1da177e2005-04-16 15:20:36 -07002741 if (!busiest->active_balance) {
2742 busiest->active_balance = 1;
2743 busiest->push_cpu = this_cpu;
Nick Piggin81026792005-06-25 14:57:07 -07002744 active_balance = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002745 }
Christoph Lameterfe2eea32006-12-10 02:20:21 -08002746 spin_unlock_irqrestore(&busiest->lock, flags);
Nick Piggin81026792005-06-25 14:57:07 -07002747 if (active_balance)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002748 wake_up_process(busiest->migration_thread);
2749
2750 /*
2751 * We've kicked active balancing, reset the failure
2752 * counter.
2753 */
Nick Piggin39507452005-06-25 14:57:09 -07002754 sd->nr_balance_failed = sd->cache_nice_tries+1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002755 }
Nick Piggin81026792005-06-25 14:57:07 -07002756 } else
Linus Torvalds1da177e2005-04-16 15:20:36 -07002757 sd->nr_balance_failed = 0;
2758
Nick Piggin81026792005-06-25 14:57:07 -07002759 if (likely(!active_balance)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002760 /* We were unbalanced, so reset the balancing interval */
2761 sd->balance_interval = sd->min_interval;
Nick Piggin81026792005-06-25 14:57:07 -07002762 } else {
2763 /*
2764 * If we've begun active balancing, start to back off. This
2765 * case may not be covered by the all_pinned logic if there
2766 * is only 1 task on the busy runqueue (because we don't call
2767 * move_tasks).
2768 */
2769 if (sd->balance_interval < sd->max_interval)
2770 sd->balance_interval *= 2;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002771 }
2772
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07002773 if (!nr_moved && !sd_idle && sd->flags & SD_SHARE_CPUPOWER &&
Siddha, Suresh B89c47102006-10-03 01:14:09 -07002774 !test_sd_parent(sd, SD_POWERSAVINGS_BALANCE))
Nick Piggin5969fe02005-09-10 00:26:19 -07002775 return -1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002776 return nr_moved;
2777
2778out_balanced:
Linus Torvalds1da177e2005-04-16 15:20:36 -07002779 schedstat_inc(sd, lb_balanced[idle]);
2780
Nick Piggin16cfb1c2005-06-25 14:57:08 -07002781 sd->nr_balance_failed = 0;
Siddha, Suresh Bfa3b6dd2005-09-10 00:26:21 -07002782
2783out_one_pinned:
Linus Torvalds1da177e2005-04-16 15:20:36 -07002784 /* tune up the balancing interval */
Nick Piggin77391d72005-06-25 14:57:30 -07002785 if ((all_pinned && sd->balance_interval < MAX_PINNED_INTERVAL) ||
2786 (sd->balance_interval < sd->max_interval))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002787 sd->balance_interval *= 2;
2788
Ingo Molnar48f24c42006-07-03 00:25:40 -07002789 if (!sd_idle && sd->flags & SD_SHARE_CPUPOWER &&
Siddha, Suresh B89c47102006-10-03 01:14:09 -07002790 !test_sd_parent(sd, SD_POWERSAVINGS_BALANCE))
Nick Piggin5969fe02005-09-10 00:26:19 -07002791 return -1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002792 return 0;
2793}
2794
2795/*
2796 * Check this_cpu to ensure it is balanced within domain. Attempt to move
2797 * tasks if there is an imbalance.
2798 *
2799 * Called from schedule when this_rq is about to become idle (NEWLY_IDLE).
2800 * this_rq is locked.
2801 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07002802static int
Ingo Molnar70b97a72006-07-03 00:25:42 -07002803load_balance_newidle(int this_cpu, struct rq *this_rq, struct sched_domain *sd)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002804{
2805 struct sched_group *group;
Ingo Molnar70b97a72006-07-03 00:25:42 -07002806 struct rq *busiest = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002807 unsigned long imbalance;
2808 int nr_moved = 0;
Nick Piggin5969fe02005-09-10 00:26:19 -07002809 int sd_idle = 0;
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002810 cpumask_t cpus = CPU_MASK_ALL;
Nick Piggin5969fe02005-09-10 00:26:19 -07002811
Siddha, Suresh B89c47102006-10-03 01:14:09 -07002812 /*
2813 * When power savings policy is enabled for the parent domain, idle
2814 * sibling can pick up load irrespective of busy siblings. In this case,
2815 * let the state of idle sibling percolate up as IDLE, instead of
2816 * portraying it as NOT_IDLE.
2817 */
2818 if (sd->flags & SD_SHARE_CPUPOWER &&
2819 !test_sd_parent(sd, SD_POWERSAVINGS_BALANCE))
Nick Piggin5969fe02005-09-10 00:26:19 -07002820 sd_idle = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002821
2822 schedstat_inc(sd, lb_cnt[NEWLY_IDLE]);
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002823redo:
2824 group = find_busiest_group(sd, this_cpu, &imbalance, NEWLY_IDLE,
Siddha, Suresh B783609c2006-12-10 02:20:33 -08002825 &sd_idle, &cpus, NULL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002826 if (!group) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002827 schedstat_inc(sd, lb_nobusyg[NEWLY_IDLE]);
Nick Piggin16cfb1c2005-06-25 14:57:08 -07002828 goto out_balanced;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002829 }
2830
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002831 busiest = find_busiest_queue(group, NEWLY_IDLE, imbalance,
2832 &cpus);
Nick Piggindb935db2005-06-25 14:57:11 -07002833 if (!busiest) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002834 schedstat_inc(sd, lb_nobusyq[NEWLY_IDLE]);
Nick Piggin16cfb1c2005-06-25 14:57:08 -07002835 goto out_balanced;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002836 }
2837
Nick Piggindb935db2005-06-25 14:57:11 -07002838 BUG_ON(busiest == this_rq);
2839
Linus Torvalds1da177e2005-04-16 15:20:36 -07002840 schedstat_add(sd, lb_imbalance[NEWLY_IDLE], imbalance);
Nick Piggind6d5cfa2005-09-10 00:26:16 -07002841
2842 nr_moved = 0;
2843 if (busiest->nr_running > 1) {
2844 /* Attempt to move tasks */
2845 double_lock_balance(this_rq, busiest);
2846 nr_moved = move_tasks(this_rq, this_cpu, busiest,
Peter Williams2dd73a42006-06-27 02:54:34 -07002847 minus_1_or_zero(busiest->nr_running),
Nick Piggin81026792005-06-25 14:57:07 -07002848 imbalance, sd, NEWLY_IDLE, NULL);
Nick Piggind6d5cfa2005-09-10 00:26:16 -07002849 spin_unlock(&busiest->lock);
Christoph Lameter0a2966b2006-09-25 23:30:51 -07002850
2851 if (!nr_moved) {
2852 cpu_clear(cpu_of(busiest), cpus);
2853 if (!cpus_empty(cpus))
2854 goto redo;
2855 }
Nick Piggind6d5cfa2005-09-10 00:26:16 -07002856 }
2857
Nick Piggin5969fe02005-09-10 00:26:19 -07002858 if (!nr_moved) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002859 schedstat_inc(sd, lb_failed[NEWLY_IDLE]);
Siddha, Suresh B89c47102006-10-03 01:14:09 -07002860 if (!sd_idle && sd->flags & SD_SHARE_CPUPOWER &&
2861 !test_sd_parent(sd, SD_POWERSAVINGS_BALANCE))
Nick Piggin5969fe02005-09-10 00:26:19 -07002862 return -1;
2863 } else
Nick Piggin16cfb1c2005-06-25 14:57:08 -07002864 sd->nr_balance_failed = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002865
Linus Torvalds1da177e2005-04-16 15:20:36 -07002866 return nr_moved;
Nick Piggin16cfb1c2005-06-25 14:57:08 -07002867
2868out_balanced:
2869 schedstat_inc(sd, lb_balanced[NEWLY_IDLE]);
Ingo Molnar48f24c42006-07-03 00:25:40 -07002870 if (!sd_idle && sd->flags & SD_SHARE_CPUPOWER &&
Siddha, Suresh B89c47102006-10-03 01:14:09 -07002871 !test_sd_parent(sd, SD_POWERSAVINGS_BALANCE))
Nick Piggin5969fe02005-09-10 00:26:19 -07002872 return -1;
Nick Piggin16cfb1c2005-06-25 14:57:08 -07002873 sd->nr_balance_failed = 0;
Ingo Molnar48f24c42006-07-03 00:25:40 -07002874
Nick Piggin16cfb1c2005-06-25 14:57:08 -07002875 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002876}
2877
2878/*
2879 * idle_balance is called by schedule() if this_cpu is about to become
2880 * idle. Attempts to pull tasks from other CPUs.
2881 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002882static void idle_balance(int this_cpu, struct rq *this_rq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002883{
2884 struct sched_domain *sd;
Christoph Lameter1bd77f22006-12-10 02:20:27 -08002885 int pulled_task = 0;
2886 unsigned long next_balance = jiffies + 60 * HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002887
2888 for_each_domain(this_cpu, sd) {
2889 if (sd->flags & SD_BALANCE_NEWIDLE) {
Ingo Molnar48f24c42006-07-03 00:25:40 -07002890 /* If we've pulled tasks over stop searching: */
Christoph Lameter1bd77f22006-12-10 02:20:27 -08002891 pulled_task = load_balance_newidle(this_cpu,
2892 this_rq, sd);
2893 if (time_after(next_balance,
2894 sd->last_balance + sd->balance_interval))
2895 next_balance = sd->last_balance
2896 + sd->balance_interval;
2897 if (pulled_task)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002898 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002899 }
2900 }
Christoph Lameter1bd77f22006-12-10 02:20:27 -08002901 if (!pulled_task)
2902 /*
2903 * We are going idle. next_balance may be set based on
2904 * a busy processor. So reset next_balance.
2905 */
2906 this_rq->next_balance = next_balance;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002907}
2908
2909/*
2910 * active_load_balance is run by migration threads. It pushes running tasks
2911 * off the busiest CPU onto idle CPUs. It requires at least 1 task to be
2912 * running on each physical CPU where possible, and avoids physical /
2913 * logical imbalances.
2914 *
2915 * Called with busiest_rq locked.
2916 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07002917static void active_load_balance(struct rq *busiest_rq, int busiest_cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002918{
Nick Piggin39507452005-06-25 14:57:09 -07002919 int target_cpu = busiest_rq->push_cpu;
Ingo Molnar70b97a72006-07-03 00:25:42 -07002920 struct sched_domain *sd;
2921 struct rq *target_rq;
Nick Piggin39507452005-06-25 14:57:09 -07002922
Ingo Molnar48f24c42006-07-03 00:25:40 -07002923 /* Is there any task to move? */
Nick Piggin39507452005-06-25 14:57:09 -07002924 if (busiest_rq->nr_running <= 1)
Nick Piggin39507452005-06-25 14:57:09 -07002925 return;
2926
2927 target_rq = cpu_rq(target_cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002928
2929 /*
Nick Piggin39507452005-06-25 14:57:09 -07002930 * This condition is "impossible", if it occurs
2931 * we need to fix it. Originally reported by
2932 * Bjorn Helgaas on a 128-cpu setup.
Linus Torvalds1da177e2005-04-16 15:20:36 -07002933 */
Nick Piggin39507452005-06-25 14:57:09 -07002934 BUG_ON(busiest_rq == target_rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002935
Nick Piggin39507452005-06-25 14:57:09 -07002936 /* move a task from busiest_rq to target_rq */
2937 double_lock_balance(busiest_rq, target_rq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002938
Nick Piggin39507452005-06-25 14:57:09 -07002939 /* Search for an sd spanning us and the target CPU. */
Chen, Kenneth Wc96d1452006-06-27 02:54:28 -07002940 for_each_domain(target_cpu, sd) {
Nick Piggin39507452005-06-25 14:57:09 -07002941 if ((sd->flags & SD_LOAD_BALANCE) &&
Ingo Molnar48f24c42006-07-03 00:25:40 -07002942 cpu_isset(busiest_cpu, sd->span))
Nick Piggin39507452005-06-25 14:57:09 -07002943 break;
Chen, Kenneth Wc96d1452006-06-27 02:54:28 -07002944 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002945
Ingo Molnar48f24c42006-07-03 00:25:40 -07002946 if (likely(sd)) {
2947 schedstat_inc(sd, alb_cnt);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002948
Ingo Molnar48f24c42006-07-03 00:25:40 -07002949 if (move_tasks(target_rq, target_cpu, busiest_rq, 1,
2950 RTPRIO_TO_LOAD_WEIGHT(100), sd, SCHED_IDLE,
2951 NULL))
2952 schedstat_inc(sd, alb_pushed);
2953 else
2954 schedstat_inc(sd, alb_failed);
2955 }
Nick Piggin39507452005-06-25 14:57:09 -07002956 spin_unlock(&target_rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002957}
2958
Christoph Lameter7835b982006-12-10 02:20:22 -08002959static void update_load(struct rq *this_rq)
Ingo Molnar48f24c42006-07-03 00:25:40 -07002960{
Christoph Lameter7835b982006-12-10 02:20:22 -08002961 unsigned long this_load;
Nick Pigginff916912007-02-12 00:53:51 -08002962 unsigned int i, scale;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002963
Peter Williams2dd73a42006-06-27 02:54:34 -07002964 this_load = this_rq->raw_weighted_load;
Ingo Molnar48f24c42006-07-03 00:25:40 -07002965
2966 /* Update our load: */
Nick Pigginff916912007-02-12 00:53:51 -08002967 for (i = 0, scale = 1; i < 3; i++, scale += scale) {
Ingo Molnar48f24c42006-07-03 00:25:40 -07002968 unsigned long old_load, new_load;
2969
Nick Pigginff916912007-02-12 00:53:51 -08002970 /* scale is effectively 1 << i now, and >> i divides by scale */
2971
Nick Piggin78979862005-06-25 14:57:13 -07002972 old_load = this_rq->cpu_load[i];
Ingo Molnar48f24c42006-07-03 00:25:40 -07002973 new_load = this_load;
Nick Piggin78979862005-06-25 14:57:13 -07002974 /*
2975 * Round up the averaging division if load is increasing. This
2976 * prevents us from getting stuck on 9 if the load is 10, for
2977 * example.
2978 */
2979 if (new_load > old_load)
2980 new_load += scale-1;
Nick Pigginff916912007-02-12 00:53:51 -08002981 this_rq->cpu_load[i] = (old_load*(scale-1) + new_load) >> i;
Nick Piggin78979862005-06-25 14:57:13 -07002982 }
Christoph Lameter7835b982006-12-10 02:20:22 -08002983}
2984
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07002985#ifdef CONFIG_NO_HZ
2986static struct {
2987 atomic_t load_balancer;
2988 cpumask_t cpu_mask;
2989} nohz ____cacheline_aligned = {
2990 .load_balancer = ATOMIC_INIT(-1),
2991 .cpu_mask = CPU_MASK_NONE,
2992};
2993
Christoph Lameter7835b982006-12-10 02:20:22 -08002994/*
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07002995 * This routine will try to nominate the ilb (idle load balancing)
2996 * owner among the cpus whose ticks are stopped. ilb owner will do the idle
2997 * load balancing on behalf of all those cpus. If all the cpus in the system
2998 * go into this tickless mode, then there will be no ilb owner (as there is
2999 * no need for one) and all the cpus will sleep till the next wakeup event
3000 * arrives...
Christoph Lameter7835b982006-12-10 02:20:22 -08003001 *
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003002 * For the ilb owner, tick is not stopped. And this tick will be used
3003 * for idle load balancing. ilb owner will still be part of
3004 * nohz.cpu_mask..
3005 *
3006 * While stopping the tick, this cpu will become the ilb owner if there
3007 * is no other owner. And will be the owner till that cpu becomes busy
3008 * or if all cpus in the system stop their ticks at which point
3009 * there is no need for ilb owner.
3010 *
3011 * When the ilb owner becomes busy, it nominates another owner, during the
3012 * next busy scheduler_tick()
3013 */
3014int select_nohz_load_balancer(int stop_tick)
3015{
3016 int cpu = smp_processor_id();
3017
3018 if (stop_tick) {
3019 cpu_set(cpu, nohz.cpu_mask);
3020 cpu_rq(cpu)->in_nohz_recently = 1;
3021
3022 /*
3023 * If we are going offline and still the leader, give up!
3024 */
3025 if (cpu_is_offline(cpu) &&
3026 atomic_read(&nohz.load_balancer) == cpu) {
3027 if (atomic_cmpxchg(&nohz.load_balancer, cpu, -1) != cpu)
3028 BUG();
3029 return 0;
3030 }
3031
3032 /* time for ilb owner also to sleep */
3033 if (cpus_weight(nohz.cpu_mask) == num_online_cpus()) {
3034 if (atomic_read(&nohz.load_balancer) == cpu)
3035 atomic_set(&nohz.load_balancer, -1);
3036 return 0;
3037 }
3038
3039 if (atomic_read(&nohz.load_balancer) == -1) {
3040 /* make me the ilb owner */
3041 if (atomic_cmpxchg(&nohz.load_balancer, -1, cpu) == -1)
3042 return 1;
3043 } else if (atomic_read(&nohz.load_balancer) == cpu)
3044 return 1;
3045 } else {
3046 if (!cpu_isset(cpu, nohz.cpu_mask))
3047 return 0;
3048
3049 cpu_clear(cpu, nohz.cpu_mask);
3050
3051 if (atomic_read(&nohz.load_balancer) == cpu)
3052 if (atomic_cmpxchg(&nohz.load_balancer, cpu, -1) != cpu)
3053 BUG();
3054 }
3055 return 0;
3056}
3057#endif
3058
3059static DEFINE_SPINLOCK(balancing);
3060
3061/*
Christoph Lameter7835b982006-12-10 02:20:22 -08003062 * It checks each scheduling domain to see if it is due to be balanced,
3063 * and initiates a balancing operation if so.
3064 *
3065 * Balancing parameters are set up in arch_init_sched_domains.
3066 */
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003067static inline void rebalance_domains(int cpu, enum idle_type idle)
Christoph Lameter7835b982006-12-10 02:20:22 -08003068{
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003069 int balance = 1;
3070 struct rq *rq = cpu_rq(cpu);
Christoph Lameter7835b982006-12-10 02:20:22 -08003071 unsigned long interval;
3072 struct sched_domain *sd;
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003073 /* Earliest time when we have to do rebalance again */
Christoph Lameterc9819f42006-12-10 02:20:25 -08003074 unsigned long next_balance = jiffies + 60*HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003075
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003076 for_each_domain(cpu, sd) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003077 if (!(sd->flags & SD_LOAD_BALANCE))
3078 continue;
3079
3080 interval = sd->balance_interval;
3081 if (idle != SCHED_IDLE)
3082 interval *= sd->busy_factor;
3083
3084 /* scale ms to jiffies */
3085 interval = msecs_to_jiffies(interval);
3086 if (unlikely(!interval))
3087 interval = 1;
3088
Christoph Lameter08c183f2006-12-10 02:20:29 -08003089 if (sd->flags & SD_SERIALIZE) {
3090 if (!spin_trylock(&balancing))
3091 goto out;
3092 }
3093
Christoph Lameterc9819f42006-12-10 02:20:25 -08003094 if (time_after_eq(jiffies, sd->last_balance + interval)) {
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003095 if (load_balance(cpu, rq, sd, idle, &balance)) {
Siddha, Suresh Bfa3b6dd2005-09-10 00:26:21 -07003096 /*
3097 * We've pulled tasks over so either we're no
Nick Piggin5969fe02005-09-10 00:26:19 -07003098 * longer idle, or one of our SMT siblings is
3099 * not idle.
3100 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003101 idle = NOT_IDLE;
3102 }
Christoph Lameter1bd77f22006-12-10 02:20:27 -08003103 sd->last_balance = jiffies;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003104 }
Christoph Lameter08c183f2006-12-10 02:20:29 -08003105 if (sd->flags & SD_SERIALIZE)
3106 spin_unlock(&balancing);
3107out:
Christoph Lameterc9819f42006-12-10 02:20:25 -08003108 if (time_after(next_balance, sd->last_balance + interval))
3109 next_balance = sd->last_balance + interval;
Siddha, Suresh B783609c2006-12-10 02:20:33 -08003110
3111 /*
3112 * Stop the load balance at this level. There is another
3113 * CPU in our sched group which is doing load balancing more
3114 * actively.
3115 */
3116 if (!balance)
3117 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003118 }
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003119 rq->next_balance = next_balance;
3120}
3121
3122/*
3123 * run_rebalance_domains is triggered when needed from the scheduler tick.
3124 * In CONFIG_NO_HZ case, the idle load balance owner will do the
3125 * rebalancing for all the cpus for whom scheduler ticks are stopped.
3126 */
3127static void run_rebalance_domains(struct softirq_action *h)
3128{
3129 int local_cpu = smp_processor_id();
3130 struct rq *local_rq = cpu_rq(local_cpu);
3131 enum idle_type idle = local_rq->idle_at_tick ? SCHED_IDLE : NOT_IDLE;
3132
3133 rebalance_domains(local_cpu, idle);
3134
3135#ifdef CONFIG_NO_HZ
3136 /*
3137 * If this cpu is the owner for idle load balancing, then do the
3138 * balancing on behalf of the other idle cpus whose ticks are
3139 * stopped.
3140 */
3141 if (local_rq->idle_at_tick &&
3142 atomic_read(&nohz.load_balancer) == local_cpu) {
3143 cpumask_t cpus = nohz.cpu_mask;
3144 struct rq *rq;
3145 int balance_cpu;
3146
3147 cpu_clear(local_cpu, cpus);
3148 for_each_cpu_mask(balance_cpu, cpus) {
3149 /*
3150 * If this cpu gets work to do, stop the load balancing
3151 * work being done for other cpus. Next load
3152 * balancing owner will pick it up.
3153 */
3154 if (need_resched())
3155 break;
3156
3157 rebalance_domains(balance_cpu, SCHED_IDLE);
3158
3159 rq = cpu_rq(balance_cpu);
3160 if (time_after(local_rq->next_balance, rq->next_balance))
3161 local_rq->next_balance = rq->next_balance;
3162 }
3163 }
3164#endif
3165}
3166
3167/*
3168 * Trigger the SCHED_SOFTIRQ if it is time to do periodic load balancing.
3169 *
3170 * In case of CONFIG_NO_HZ, this is the place where we nominate a new
3171 * idle load balancing owner or decide to stop the periodic load balancing,
3172 * if the whole system is idle.
3173 */
3174static inline void trigger_load_balance(int cpu)
3175{
3176 struct rq *rq = cpu_rq(cpu);
3177#ifdef CONFIG_NO_HZ
3178 /*
3179 * If we were in the nohz mode recently and busy at the current
3180 * scheduler tick, then check if we need to nominate new idle
3181 * load balancer.
3182 */
3183 if (rq->in_nohz_recently && !rq->idle_at_tick) {
3184 rq->in_nohz_recently = 0;
3185
3186 if (atomic_read(&nohz.load_balancer) == cpu) {
3187 cpu_clear(cpu, nohz.cpu_mask);
3188 atomic_set(&nohz.load_balancer, -1);
3189 }
3190
3191 if (atomic_read(&nohz.load_balancer) == -1) {
3192 /*
3193 * simple selection for now: Nominate the
3194 * first cpu in the nohz list to be the next
3195 * ilb owner.
3196 *
3197 * TBD: Traverse the sched domains and nominate
3198 * the nearest cpu in the nohz.cpu_mask.
3199 */
3200 int ilb = first_cpu(nohz.cpu_mask);
3201
3202 if (ilb != NR_CPUS)
3203 resched_cpu(ilb);
3204 }
3205 }
3206
3207 /*
3208 * If this cpu is idle and doing idle load balancing for all the
3209 * cpus with ticks stopped, is it time for that to stop?
3210 */
3211 if (rq->idle_at_tick && atomic_read(&nohz.load_balancer) == cpu &&
3212 cpus_weight(nohz.cpu_mask) == num_online_cpus()) {
3213 resched_cpu(cpu);
3214 return;
3215 }
3216
3217 /*
3218 * If this cpu is idle and the idle load balancing is done by
3219 * someone else, then no need raise the SCHED_SOFTIRQ
3220 */
3221 if (rq->idle_at_tick && atomic_read(&nohz.load_balancer) != cpu &&
3222 cpu_isset(cpu, nohz.cpu_mask))
3223 return;
3224#endif
3225 if (time_after_eq(jiffies, rq->next_balance))
3226 raise_softirq(SCHED_SOFTIRQ);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003227}
3228#else
3229/*
3230 * on UP we do not need to balance between CPUs:
3231 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07003232static inline void idle_balance(int cpu, struct rq *rq)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003233{
3234}
3235#endif
3236
Linus Torvalds1da177e2005-04-16 15:20:36 -07003237DEFINE_PER_CPU(struct kernel_stat, kstat);
3238
3239EXPORT_PER_CPU_SYMBOL(kstat);
3240
3241/*
3242 * This is called on clock ticks and on context switches.
3243 * Bank in p->sched_time the ns elapsed since the last tick or switch.
3244 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07003245static inline void
Ingo Molnar70b97a72006-07-03 00:25:42 -07003246update_cpu_clock(struct task_struct *p, struct rq *rq, unsigned long long now)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003247{
Mike Galbraithb18ec802006-12-10 02:20:31 -08003248 p->sched_time += now - p->last_ran;
3249 p->last_ran = rq->most_recent_timestamp = now;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003250}
3251
3252/*
3253 * Return current->sched_time plus any more ns on the sched_clock
3254 * that have not yet been banked.
3255 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07003256unsigned long long current_sched_time(const struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003257{
3258 unsigned long long ns;
3259 unsigned long flags;
Ingo Molnar48f24c42006-07-03 00:25:40 -07003260
Linus Torvalds1da177e2005-04-16 15:20:36 -07003261 local_irq_save(flags);
Mike Galbraithb18ec802006-12-10 02:20:31 -08003262 ns = p->sched_time + sched_clock() - p->last_ran;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003263 local_irq_restore(flags);
Ingo Molnar48f24c42006-07-03 00:25:40 -07003264
Linus Torvalds1da177e2005-04-16 15:20:36 -07003265 return ns;
3266}
3267
3268/*
Linus Torvaldsf1adad72006-05-21 18:54:09 -07003269 * We place interactive tasks back into the active array, if possible.
3270 *
3271 * To guarantee that this does not starve expired tasks we ignore the
3272 * interactivity of a task if the first expired task had to wait more
3273 * than a 'reasonable' amount of time. This deadline timeout is
3274 * load-dependent, as the frequency of array switched decreases with
3275 * increasing number of running tasks. We also ignore the interactivity
3276 * if a better static_prio task has expired:
3277 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07003278static inline int expired_starving(struct rq *rq)
Ingo Molnar48f24c42006-07-03 00:25:40 -07003279{
3280 if (rq->curr->static_prio > rq->best_expired_prio)
3281 return 1;
3282 if (!STARVATION_LIMIT || !rq->expired_timestamp)
3283 return 0;
3284 if (jiffies - rq->expired_timestamp > STARVATION_LIMIT * rq->nr_running)
3285 return 1;
3286 return 0;
3287}
Linus Torvaldsf1adad72006-05-21 18:54:09 -07003288
3289/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07003290 * Account user cpu time to a process.
3291 * @p: the process that the cpu time gets accounted to
3292 * @hardirq_offset: the offset to subtract from hardirq_count()
3293 * @cputime: the cpu time spent in user space since the last update
3294 */
3295void account_user_time(struct task_struct *p, cputime_t cputime)
3296{
3297 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
3298 cputime64_t tmp;
3299
3300 p->utime = cputime_add(p->utime, cputime);
3301
3302 /* Add user time to cpustat. */
3303 tmp = cputime_to_cputime64(cputime);
3304 if (TASK_NICE(p) > 0)
3305 cpustat->nice = cputime64_add(cpustat->nice, tmp);
3306 else
3307 cpustat->user = cputime64_add(cpustat->user, tmp);
3308}
3309
3310/*
3311 * Account system cpu time to a process.
3312 * @p: the process that the cpu time gets accounted to
3313 * @hardirq_offset: the offset to subtract from hardirq_count()
3314 * @cputime: the cpu time spent in kernel space since the last update
3315 */
3316void account_system_time(struct task_struct *p, int hardirq_offset,
3317 cputime_t cputime)
3318{
3319 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
Ingo Molnar70b97a72006-07-03 00:25:42 -07003320 struct rq *rq = this_rq();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003321 cputime64_t tmp;
3322
3323 p->stime = cputime_add(p->stime, cputime);
3324
3325 /* Add system time to cpustat. */
3326 tmp = cputime_to_cputime64(cputime);
3327 if (hardirq_count() - hardirq_offset)
3328 cpustat->irq = cputime64_add(cpustat->irq, tmp);
3329 else if (softirq_count())
3330 cpustat->softirq = cputime64_add(cpustat->softirq, tmp);
3331 else if (p != rq->idle)
3332 cpustat->system = cputime64_add(cpustat->system, tmp);
3333 else if (atomic_read(&rq->nr_iowait) > 0)
3334 cpustat->iowait = cputime64_add(cpustat->iowait, tmp);
3335 else
3336 cpustat->idle = cputime64_add(cpustat->idle, tmp);
3337 /* Account for system time used */
3338 acct_update_integrals(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003339}
3340
3341/*
3342 * Account for involuntary wait time.
3343 * @p: the process from which the cpu time has been stolen
3344 * @steal: the cpu time spent in involuntary wait
3345 */
3346void account_steal_time(struct task_struct *p, cputime_t steal)
3347{
3348 struct cpu_usage_stat *cpustat = &kstat_this_cpu.cpustat;
3349 cputime64_t tmp = cputime_to_cputime64(steal);
Ingo Molnar70b97a72006-07-03 00:25:42 -07003350 struct rq *rq = this_rq();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003351
3352 if (p == rq->idle) {
3353 p->stime = cputime_add(p->stime, steal);
3354 if (atomic_read(&rq->nr_iowait) > 0)
3355 cpustat->iowait = cputime64_add(cpustat->iowait, tmp);
3356 else
3357 cpustat->idle = cputime64_add(cpustat->idle, tmp);
3358 } else
3359 cpustat->steal = cputime64_add(cpustat->steal, tmp);
3360}
3361
Christoph Lameter7835b982006-12-10 02:20:22 -08003362static void task_running_tick(struct rq *rq, struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003363{
Linus Torvalds1da177e2005-04-16 15:20:36 -07003364 if (p->array != rq->active) {
Christoph Lameter7835b982006-12-10 02:20:22 -08003365 /* Task has expired but was not scheduled yet */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003366 set_tsk_need_resched(p);
Christoph Lameter7835b982006-12-10 02:20:22 -08003367 return;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003368 }
3369 spin_lock(&rq->lock);
3370 /*
3371 * The task was running during this tick - update the
3372 * time slice counter. Note: we do not update a thread's
3373 * priority until it either goes to sleep or uses up its
3374 * timeslice. This makes it possible for interactive tasks
3375 * to use up their timeslices at their highest priority levels.
3376 */
3377 if (rt_task(p)) {
3378 /*
3379 * RR tasks need a special form of timeslice management.
3380 * FIFO tasks have no timeslices.
3381 */
3382 if ((p->policy == SCHED_RR) && !--p->time_slice) {
3383 p->time_slice = task_timeslice(p);
3384 p->first_time_slice = 0;
3385 set_tsk_need_resched(p);
3386
3387 /* put it at the end of the queue: */
3388 requeue_task(p, rq->active);
3389 }
3390 goto out_unlock;
3391 }
3392 if (!--p->time_slice) {
3393 dequeue_task(p, rq->active);
3394 set_tsk_need_resched(p);
3395 p->prio = effective_prio(p);
3396 p->time_slice = task_timeslice(p);
3397 p->first_time_slice = 0;
3398
3399 if (!rq->expired_timestamp)
3400 rq->expired_timestamp = jiffies;
Ingo Molnar48f24c42006-07-03 00:25:40 -07003401 if (!TASK_INTERACTIVE(p) || expired_starving(rq)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003402 enqueue_task(p, rq->expired);
3403 if (p->static_prio < rq->best_expired_prio)
3404 rq->best_expired_prio = p->static_prio;
3405 } else
3406 enqueue_task(p, rq->active);
3407 } else {
3408 /*
3409 * Prevent a too long timeslice allowing a task to monopolize
3410 * the CPU. We do this by splitting up the timeslice into
3411 * smaller pieces.
3412 *
3413 * Note: this does not mean the task's timeslices expire or
3414 * get lost in any way, they just might be preempted by
3415 * another task of equal priority. (one with higher
3416 * priority would have preempted this task already.) We
3417 * requeue this task to the end of the list on this priority
3418 * level, which is in essence a round-robin of tasks with
3419 * equal priority.
3420 *
3421 * This only applies to tasks in the interactive
3422 * delta range with at least TIMESLICE_GRANULARITY to requeue.
3423 */
3424 if (TASK_INTERACTIVE(p) && !((task_timeslice(p) -
3425 p->time_slice) % TIMESLICE_GRANULARITY(p)) &&
3426 (p->time_slice >= TIMESLICE_GRANULARITY(p)) &&
3427 (p->array == rq->active)) {
3428
3429 requeue_task(p, rq->active);
3430 set_tsk_need_resched(p);
3431 }
3432 }
3433out_unlock:
3434 spin_unlock(&rq->lock);
Christoph Lameter7835b982006-12-10 02:20:22 -08003435}
3436
3437/*
3438 * This function gets called by the timer code, with HZ frequency.
3439 * We call it with interrupts disabled.
3440 *
3441 * It also gets called by the fork code, when changing the parent's
3442 * timeslices.
3443 */
3444void scheduler_tick(void)
3445{
3446 unsigned long long now = sched_clock();
3447 struct task_struct *p = current;
3448 int cpu = smp_processor_id();
Siddha, Suresh Bbdecea32007-05-08 00:32:48 -07003449 int idle_at_tick = idle_cpu(cpu);
Christoph Lameter7835b982006-12-10 02:20:22 -08003450 struct rq *rq = cpu_rq(cpu);
Christoph Lameter7835b982006-12-10 02:20:22 -08003451
3452 update_cpu_clock(p, rq, now);
3453
Siddha, Suresh Bbdecea32007-05-08 00:32:48 -07003454 if (!idle_at_tick)
Christoph Lameter7835b982006-12-10 02:20:22 -08003455 task_running_tick(rq, p);
Christoph Lametere418e1c2006-12-10 02:20:23 -08003456#ifdef CONFIG_SMP
Christoph Lameter7835b982006-12-10 02:20:22 -08003457 update_load(rq);
Siddha, Suresh Bbdecea32007-05-08 00:32:48 -07003458 rq->idle_at_tick = idle_at_tick;
Siddha, Suresh B46cb4b72007-05-08 00:32:51 -07003459 trigger_load_balance(cpu);
Christoph Lametere418e1c2006-12-10 02:20:23 -08003460#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07003461}
3462
Linus Torvalds1da177e2005-04-16 15:20:36 -07003463#if defined(CONFIG_PREEMPT) && defined(CONFIG_DEBUG_PREEMPT)
3464
3465void fastcall add_preempt_count(int val)
3466{
3467 /*
3468 * Underflow?
3469 */
Ingo Molnar9a11b49a2006-07-03 00:24:33 -07003470 if (DEBUG_LOCKS_WARN_ON((preempt_count() < 0)))
3471 return;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003472 preempt_count() += val;
3473 /*
3474 * Spinlock count overflowing soon?
3475 */
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08003476 DEBUG_LOCKS_WARN_ON((preempt_count() & PREEMPT_MASK) >=
3477 PREEMPT_MASK - 10);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003478}
3479EXPORT_SYMBOL(add_preempt_count);
3480
3481void fastcall sub_preempt_count(int val)
3482{
3483 /*
3484 * Underflow?
3485 */
Ingo Molnar9a11b49a2006-07-03 00:24:33 -07003486 if (DEBUG_LOCKS_WARN_ON(val > preempt_count()))
3487 return;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003488 /*
3489 * Is the spinlock portion underflowing?
3490 */
Ingo Molnar9a11b49a2006-07-03 00:24:33 -07003491 if (DEBUG_LOCKS_WARN_ON((val < PREEMPT_MASK) &&
3492 !(preempt_count() & PREEMPT_MASK)))
3493 return;
3494
Linus Torvalds1da177e2005-04-16 15:20:36 -07003495 preempt_count() -= val;
3496}
3497EXPORT_SYMBOL(sub_preempt_count);
3498
3499#endif
3500
Con Kolivas3dee3862006-03-31 02:31:23 -08003501static inline int interactive_sleep(enum sleep_type sleep_type)
3502{
3503 return (sleep_type == SLEEP_INTERACTIVE ||
3504 sleep_type == SLEEP_INTERRUPTED);
3505}
3506
Linus Torvalds1da177e2005-04-16 15:20:36 -07003507/*
3508 * schedule() is the main scheduler function.
3509 */
3510asmlinkage void __sched schedule(void)
3511{
Ingo Molnar36c8b582006-07-03 00:25:41 -07003512 struct task_struct *prev, *next;
Ingo Molnar70b97a72006-07-03 00:25:42 -07003513 struct prio_array *array;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003514 struct list_head *queue;
3515 unsigned long long now;
3516 unsigned long run_time;
Chen Shanga3464a12005-06-25 14:57:31 -07003517 int cpu, idx, new_prio;
Ingo Molnar48f24c42006-07-03 00:25:40 -07003518 long *switch_count;
Ingo Molnar70b97a72006-07-03 00:25:42 -07003519 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003520
3521 /*
3522 * Test if we are atomic. Since do_exit() needs to call into
3523 * schedule() atomically, we ignore that path for now.
3524 * Otherwise, whine if we are scheduling when we should not be.
3525 */
Andreas Mohr77e4bfb2006-03-27 01:15:20 -08003526 if (unlikely(in_atomic() && !current->exit_state)) {
3527 printk(KERN_ERR "BUG: scheduling while atomic: "
3528 "%s/0x%08x/%d\n",
3529 current->comm, preempt_count(), current->pid);
Peter Zijlstraa4c410f2006-12-06 20:37:21 -08003530 debug_show_held_locks(current);
Ingo Molnar3117df02006-12-13 00:34:43 -08003531 if (irqs_disabled())
3532 print_irqtrace_events(current);
Andreas Mohr77e4bfb2006-03-27 01:15:20 -08003533 dump_stack();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003534 }
3535 profile_hit(SCHED_PROFILING, __builtin_return_address(0));
3536
3537need_resched:
3538 preempt_disable();
3539 prev = current;
3540 release_kernel_lock(prev);
3541need_resched_nonpreemptible:
3542 rq = this_rq();
3543
3544 /*
3545 * The idle thread is not allowed to schedule!
3546 * Remove this check after it has been exercised a bit.
3547 */
3548 if (unlikely(prev == rq->idle) && prev->state != TASK_RUNNING) {
3549 printk(KERN_ERR "bad: scheduling from the idle thread!\n");
3550 dump_stack();
3551 }
3552
3553 schedstat_inc(rq, sched_cnt);
3554 now = sched_clock();
Ingo Molnar238628e2005-04-18 10:58:36 -07003555 if (likely((long long)(now - prev->timestamp) < NS_MAX_SLEEP_AVG)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003556 run_time = now - prev->timestamp;
Ingo Molnar238628e2005-04-18 10:58:36 -07003557 if (unlikely((long long)(now - prev->timestamp) < 0))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003558 run_time = 0;
3559 } else
3560 run_time = NS_MAX_SLEEP_AVG;
3561
3562 /*
3563 * Tasks charged proportionately less run_time at high sleep_avg to
3564 * delay them losing their interactive status
3565 */
3566 run_time /= (CURRENT_BONUS(prev) ? : 1);
3567
3568 spin_lock_irq(&rq->lock);
3569
Linus Torvalds1da177e2005-04-16 15:20:36 -07003570 switch_count = &prev->nivcsw;
3571 if (prev->state && !(preempt_count() & PREEMPT_ACTIVE)) {
3572 switch_count = &prev->nvcsw;
3573 if (unlikely((prev->state & TASK_INTERRUPTIBLE) &&
3574 unlikely(signal_pending(prev))))
3575 prev->state = TASK_RUNNING;
3576 else {
3577 if (prev->state == TASK_UNINTERRUPTIBLE)
3578 rq->nr_uninterruptible++;
3579 deactivate_task(prev, rq);
3580 }
3581 }
3582
3583 cpu = smp_processor_id();
3584 if (unlikely(!rq->nr_running)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003585 idle_balance(cpu, rq);
3586 if (!rq->nr_running) {
3587 next = rq->idle;
3588 rq->expired_timestamp = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003589 goto switch_tasks;
3590 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003591 }
3592
3593 array = rq->active;
3594 if (unlikely(!array->nr_active)) {
3595 /*
3596 * Switch the active and expired arrays.
3597 */
3598 schedstat_inc(rq, sched_switch);
3599 rq->active = rq->expired;
3600 rq->expired = array;
3601 array = rq->active;
3602 rq->expired_timestamp = 0;
3603 rq->best_expired_prio = MAX_PRIO;
3604 }
3605
3606 idx = sched_find_first_bit(array->bitmap);
3607 queue = array->queue + idx;
Ingo Molnar36c8b582006-07-03 00:25:41 -07003608 next = list_entry(queue->next, struct task_struct, run_list);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003609
Con Kolivas3dee3862006-03-31 02:31:23 -08003610 if (!rt_task(next) && interactive_sleep(next->sleep_type)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07003611 unsigned long long delta = now - next->timestamp;
Ingo Molnar238628e2005-04-18 10:58:36 -07003612 if (unlikely((long long)(now - next->timestamp) < 0))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003613 delta = 0;
3614
Con Kolivas3dee3862006-03-31 02:31:23 -08003615 if (next->sleep_type == SLEEP_INTERACTIVE)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003616 delta = delta * (ON_RUNQUEUE_WEIGHT * 128 / 100) / 128;
3617
3618 array = next->array;
Chen Shanga3464a12005-06-25 14:57:31 -07003619 new_prio = recalc_task_prio(next, next->timestamp + delta);
3620
3621 if (unlikely(next->prio != new_prio)) {
3622 dequeue_task(next, array);
3623 next->prio = new_prio;
3624 enqueue_task(next, array);
Con Kolivas7c4bb1f2006-03-31 02:31:29 -08003625 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07003626 }
Con Kolivas3dee3862006-03-31 02:31:23 -08003627 next->sleep_type = SLEEP_NORMAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003628switch_tasks:
3629 if (next == rq->idle)
3630 schedstat_inc(rq, sched_goidle);
3631 prefetch(next);
Chen, Kenneth W383f2832005-09-09 13:02:02 -07003632 prefetch_stack(next);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003633 clear_tsk_need_resched(prev);
3634 rcu_qsctr_inc(task_cpu(prev));
3635
3636 update_cpu_clock(prev, rq, now);
3637
3638 prev->sleep_avg -= run_time;
3639 if ((long)prev->sleep_avg <= 0)
3640 prev->sleep_avg = 0;
3641 prev->timestamp = prev->last_ran = now;
3642
3643 sched_info_switch(prev, next);
3644 if (likely(prev != next)) {
Thomas Gleixnerc1e16aa2007-02-28 20:12:19 -08003645 next->timestamp = next->last_ran = now;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003646 rq->nr_switches++;
3647 rq->curr = next;
3648 ++*switch_count;
3649
Nick Piggin4866cde2005-06-25 14:57:23 -07003650 prepare_task_switch(rq, next);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003651 prev = context_switch(rq, prev, next);
3652 barrier();
Nick Piggin4866cde2005-06-25 14:57:23 -07003653 /*
3654 * this_rq must be evaluated again because prev may have moved
3655 * CPUs since it called schedule(), thus the 'rq' on its stack
3656 * frame will be invalid.
3657 */
3658 finish_task_switch(this_rq(), prev);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003659 } else
3660 spin_unlock_irq(&rq->lock);
3661
3662 prev = current;
3663 if (unlikely(reacquire_kernel_lock(prev) < 0))
3664 goto need_resched_nonpreemptible;
3665 preempt_enable_no_resched();
3666 if (unlikely(test_thread_flag(TIF_NEED_RESCHED)))
3667 goto need_resched;
3668}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003669EXPORT_SYMBOL(schedule);
3670
3671#ifdef CONFIG_PREEMPT
3672/*
Andreas Mohr2ed6e342006-07-10 04:43:52 -07003673 * this is the entry point to schedule() from in-kernel preemption
Linus Torvalds1da177e2005-04-16 15:20:36 -07003674 * off of preempt_enable. Kernel preemptions off return from interrupt
3675 * occur there and call schedule directly.
3676 */
3677asmlinkage void __sched preempt_schedule(void)
3678{
3679 struct thread_info *ti = current_thread_info();
3680#ifdef CONFIG_PREEMPT_BKL
3681 struct task_struct *task = current;
3682 int saved_lock_depth;
3683#endif
3684 /*
3685 * If there is a non-zero preempt_count or interrupts are disabled,
3686 * we do not want to preempt the current task. Just return..
3687 */
Nick Pigginbeed33a2006-10-11 01:21:52 -07003688 if (likely(ti->preempt_count || irqs_disabled()))
Linus Torvalds1da177e2005-04-16 15:20:36 -07003689 return;
3690
3691need_resched:
3692 add_preempt_count(PREEMPT_ACTIVE);
3693 /*
3694 * We keep the big kernel semaphore locked, but we
3695 * clear ->lock_depth so that schedule() doesnt
3696 * auto-release the semaphore:
3697 */
3698#ifdef CONFIG_PREEMPT_BKL
3699 saved_lock_depth = task->lock_depth;
3700 task->lock_depth = -1;
3701#endif
3702 schedule();
3703#ifdef CONFIG_PREEMPT_BKL
3704 task->lock_depth = saved_lock_depth;
3705#endif
3706 sub_preempt_count(PREEMPT_ACTIVE);
3707
3708 /* we could miss a preemption opportunity between schedule and now */
3709 barrier();
3710 if (unlikely(test_thread_flag(TIF_NEED_RESCHED)))
3711 goto need_resched;
3712}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003713EXPORT_SYMBOL(preempt_schedule);
3714
3715/*
Andreas Mohr2ed6e342006-07-10 04:43:52 -07003716 * this is the entry point to schedule() from kernel preemption
Linus Torvalds1da177e2005-04-16 15:20:36 -07003717 * off of irq context.
3718 * Note, that this is called and return with irqs disabled. This will
3719 * protect us against recursive calling from irq.
3720 */
3721asmlinkage void __sched preempt_schedule_irq(void)
3722{
3723 struct thread_info *ti = current_thread_info();
3724#ifdef CONFIG_PREEMPT_BKL
3725 struct task_struct *task = current;
3726 int saved_lock_depth;
3727#endif
Andreas Mohr2ed6e342006-07-10 04:43:52 -07003728 /* Catch callers which need to be fixed */
Linus Torvalds1da177e2005-04-16 15:20:36 -07003729 BUG_ON(ti->preempt_count || !irqs_disabled());
3730
3731need_resched:
3732 add_preempt_count(PREEMPT_ACTIVE);
3733 /*
3734 * We keep the big kernel semaphore locked, but we
3735 * clear ->lock_depth so that schedule() doesnt
3736 * auto-release the semaphore:
3737 */
3738#ifdef CONFIG_PREEMPT_BKL
3739 saved_lock_depth = task->lock_depth;
3740 task->lock_depth = -1;
3741#endif
3742 local_irq_enable();
3743 schedule();
3744 local_irq_disable();
3745#ifdef CONFIG_PREEMPT_BKL
3746 task->lock_depth = saved_lock_depth;
3747#endif
3748 sub_preempt_count(PREEMPT_ACTIVE);
3749
3750 /* we could miss a preemption opportunity between schedule and now */
3751 barrier();
3752 if (unlikely(test_thread_flag(TIF_NEED_RESCHED)))
3753 goto need_resched;
3754}
3755
3756#endif /* CONFIG_PREEMPT */
3757
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07003758int default_wake_function(wait_queue_t *curr, unsigned mode, int sync,
3759 void *key)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003760{
Ingo Molnar48f24c42006-07-03 00:25:40 -07003761 return try_to_wake_up(curr->private, mode, sync);
Linus Torvalds1da177e2005-04-16 15:20:36 -07003762}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003763EXPORT_SYMBOL(default_wake_function);
3764
3765/*
3766 * The core wakeup function. Non-exclusive wakeups (nr_exclusive == 0) just
3767 * wake everything up. If it's an exclusive wakeup (nr_exclusive == small +ve
3768 * number) then we wake all the non-exclusive tasks and one exclusive task.
3769 *
3770 * There are circumstances in which we can try to wake a task which has already
3771 * started to run but is not in state TASK_RUNNING. try_to_wake_up() returns
3772 * zero in this (rare) case, and we handle it by continuing to scan the queue.
3773 */
3774static void __wake_up_common(wait_queue_head_t *q, unsigned int mode,
3775 int nr_exclusive, int sync, void *key)
3776{
3777 struct list_head *tmp, *next;
3778
3779 list_for_each_safe(tmp, next, &q->task_list) {
Ingo Molnar48f24c42006-07-03 00:25:40 -07003780 wait_queue_t *curr = list_entry(tmp, wait_queue_t, task_list);
3781 unsigned flags = curr->flags;
3782
Linus Torvalds1da177e2005-04-16 15:20:36 -07003783 if (curr->func(curr, mode, sync, key) &&
Ingo Molnar48f24c42006-07-03 00:25:40 -07003784 (flags & WQ_FLAG_EXCLUSIVE) && !--nr_exclusive)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003785 break;
3786 }
3787}
3788
3789/**
3790 * __wake_up - wake up threads blocked on a waitqueue.
3791 * @q: the waitqueue
3792 * @mode: which threads
3793 * @nr_exclusive: how many wake-one or wake-many threads to wake up
Martin Waitz67be2dd2005-05-01 08:59:26 -07003794 * @key: is directly passed to the wakeup function
Linus Torvalds1da177e2005-04-16 15:20:36 -07003795 */
3796void fastcall __wake_up(wait_queue_head_t *q, unsigned int mode,
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07003797 int nr_exclusive, void *key)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003798{
3799 unsigned long flags;
3800
3801 spin_lock_irqsave(&q->lock, flags);
3802 __wake_up_common(q, mode, nr_exclusive, 0, key);
3803 spin_unlock_irqrestore(&q->lock, flags);
3804}
Linus Torvalds1da177e2005-04-16 15:20:36 -07003805EXPORT_SYMBOL(__wake_up);
3806
3807/*
3808 * Same as __wake_up but called with the spinlock in wait_queue_head_t held.
3809 */
3810void fastcall __wake_up_locked(wait_queue_head_t *q, unsigned int mode)
3811{
3812 __wake_up_common(q, mode, 1, 0, NULL);
3813}
3814
3815/**
Martin Waitz67be2dd2005-05-01 08:59:26 -07003816 * __wake_up_sync - wake up threads blocked on a waitqueue.
Linus Torvalds1da177e2005-04-16 15:20:36 -07003817 * @q: the waitqueue
3818 * @mode: which threads
3819 * @nr_exclusive: how many wake-one or wake-many threads to wake up
3820 *
3821 * The sync wakeup differs that the waker knows that it will schedule
3822 * away soon, so while the target thread will be woken up, it will not
3823 * be migrated to another CPU - ie. the two threads are 'synchronized'
3824 * with each other. This can prevent needless bouncing between CPUs.
3825 *
3826 * On UP it can prevent extra preemption.
3827 */
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07003828void fastcall
3829__wake_up_sync(wait_queue_head_t *q, unsigned int mode, int nr_exclusive)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003830{
3831 unsigned long flags;
3832 int sync = 1;
3833
3834 if (unlikely(!q))
3835 return;
3836
3837 if (unlikely(!nr_exclusive))
3838 sync = 0;
3839
3840 spin_lock_irqsave(&q->lock, flags);
3841 __wake_up_common(q, mode, nr_exclusive, sync, NULL);
3842 spin_unlock_irqrestore(&q->lock, flags);
3843}
3844EXPORT_SYMBOL_GPL(__wake_up_sync); /* For internal use only */
3845
3846void fastcall complete(struct completion *x)
3847{
3848 unsigned long flags;
3849
3850 spin_lock_irqsave(&x->wait.lock, flags);
3851 x->done++;
3852 __wake_up_common(&x->wait, TASK_UNINTERRUPTIBLE | TASK_INTERRUPTIBLE,
3853 1, 0, NULL);
3854 spin_unlock_irqrestore(&x->wait.lock, flags);
3855}
3856EXPORT_SYMBOL(complete);
3857
3858void fastcall complete_all(struct completion *x)
3859{
3860 unsigned long flags;
3861
3862 spin_lock_irqsave(&x->wait.lock, flags);
3863 x->done += UINT_MAX/2;
3864 __wake_up_common(&x->wait, TASK_UNINTERRUPTIBLE | TASK_INTERRUPTIBLE,
3865 0, 0, NULL);
3866 spin_unlock_irqrestore(&x->wait.lock, flags);
3867}
3868EXPORT_SYMBOL(complete_all);
3869
3870void fastcall __sched wait_for_completion(struct completion *x)
3871{
3872 might_sleep();
Ingo Molnar48f24c42006-07-03 00:25:40 -07003873
Linus Torvalds1da177e2005-04-16 15:20:36 -07003874 spin_lock_irq(&x->wait.lock);
3875 if (!x->done) {
3876 DECLARE_WAITQUEUE(wait, current);
3877
3878 wait.flags |= WQ_FLAG_EXCLUSIVE;
3879 __add_wait_queue_tail(&x->wait, &wait);
3880 do {
3881 __set_current_state(TASK_UNINTERRUPTIBLE);
3882 spin_unlock_irq(&x->wait.lock);
3883 schedule();
3884 spin_lock_irq(&x->wait.lock);
3885 } while (!x->done);
3886 __remove_wait_queue(&x->wait, &wait);
3887 }
3888 x->done--;
3889 spin_unlock_irq(&x->wait.lock);
3890}
3891EXPORT_SYMBOL(wait_for_completion);
3892
3893unsigned long fastcall __sched
3894wait_for_completion_timeout(struct completion *x, unsigned long timeout)
3895{
3896 might_sleep();
3897
3898 spin_lock_irq(&x->wait.lock);
3899 if (!x->done) {
3900 DECLARE_WAITQUEUE(wait, current);
3901
3902 wait.flags |= WQ_FLAG_EXCLUSIVE;
3903 __add_wait_queue_tail(&x->wait, &wait);
3904 do {
3905 __set_current_state(TASK_UNINTERRUPTIBLE);
3906 spin_unlock_irq(&x->wait.lock);
3907 timeout = schedule_timeout(timeout);
3908 spin_lock_irq(&x->wait.lock);
3909 if (!timeout) {
3910 __remove_wait_queue(&x->wait, &wait);
3911 goto out;
3912 }
3913 } while (!x->done);
3914 __remove_wait_queue(&x->wait, &wait);
3915 }
3916 x->done--;
3917out:
3918 spin_unlock_irq(&x->wait.lock);
3919 return timeout;
3920}
3921EXPORT_SYMBOL(wait_for_completion_timeout);
3922
3923int fastcall __sched wait_for_completion_interruptible(struct completion *x)
3924{
3925 int ret = 0;
3926
3927 might_sleep();
3928
3929 spin_lock_irq(&x->wait.lock);
3930 if (!x->done) {
3931 DECLARE_WAITQUEUE(wait, current);
3932
3933 wait.flags |= WQ_FLAG_EXCLUSIVE;
3934 __add_wait_queue_tail(&x->wait, &wait);
3935 do {
3936 if (signal_pending(current)) {
3937 ret = -ERESTARTSYS;
3938 __remove_wait_queue(&x->wait, &wait);
3939 goto out;
3940 }
3941 __set_current_state(TASK_INTERRUPTIBLE);
3942 spin_unlock_irq(&x->wait.lock);
3943 schedule();
3944 spin_lock_irq(&x->wait.lock);
3945 } while (!x->done);
3946 __remove_wait_queue(&x->wait, &wait);
3947 }
3948 x->done--;
3949out:
3950 spin_unlock_irq(&x->wait.lock);
3951
3952 return ret;
3953}
3954EXPORT_SYMBOL(wait_for_completion_interruptible);
3955
3956unsigned long fastcall __sched
3957wait_for_completion_interruptible_timeout(struct completion *x,
3958 unsigned long timeout)
3959{
3960 might_sleep();
3961
3962 spin_lock_irq(&x->wait.lock);
3963 if (!x->done) {
3964 DECLARE_WAITQUEUE(wait, current);
3965
3966 wait.flags |= WQ_FLAG_EXCLUSIVE;
3967 __add_wait_queue_tail(&x->wait, &wait);
3968 do {
3969 if (signal_pending(current)) {
3970 timeout = -ERESTARTSYS;
3971 __remove_wait_queue(&x->wait, &wait);
3972 goto out;
3973 }
3974 __set_current_state(TASK_INTERRUPTIBLE);
3975 spin_unlock_irq(&x->wait.lock);
3976 timeout = schedule_timeout(timeout);
3977 spin_lock_irq(&x->wait.lock);
3978 if (!timeout) {
3979 __remove_wait_queue(&x->wait, &wait);
3980 goto out;
3981 }
3982 } while (!x->done);
3983 __remove_wait_queue(&x->wait, &wait);
3984 }
3985 x->done--;
3986out:
3987 spin_unlock_irq(&x->wait.lock);
3988 return timeout;
3989}
3990EXPORT_SYMBOL(wait_for_completion_interruptible_timeout);
3991
3992
3993#define SLEEP_ON_VAR \
3994 unsigned long flags; \
3995 wait_queue_t wait; \
3996 init_waitqueue_entry(&wait, current);
3997
3998#define SLEEP_ON_HEAD \
3999 spin_lock_irqsave(&q->lock,flags); \
4000 __add_wait_queue(q, &wait); \
4001 spin_unlock(&q->lock);
4002
4003#define SLEEP_ON_TAIL \
4004 spin_lock_irq(&q->lock); \
4005 __remove_wait_queue(q, &wait); \
4006 spin_unlock_irqrestore(&q->lock, flags);
4007
4008void fastcall __sched interruptible_sleep_on(wait_queue_head_t *q)
4009{
4010 SLEEP_ON_VAR
4011
4012 current->state = TASK_INTERRUPTIBLE;
4013
4014 SLEEP_ON_HEAD
4015 schedule();
4016 SLEEP_ON_TAIL
4017}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004018EXPORT_SYMBOL(interruptible_sleep_on);
4019
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004020long fastcall __sched
4021interruptible_sleep_on_timeout(wait_queue_head_t *q, long timeout)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004022{
4023 SLEEP_ON_VAR
4024
4025 current->state = TASK_INTERRUPTIBLE;
4026
4027 SLEEP_ON_HEAD
4028 timeout = schedule_timeout(timeout);
4029 SLEEP_ON_TAIL
4030
4031 return timeout;
4032}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004033EXPORT_SYMBOL(interruptible_sleep_on_timeout);
4034
4035void fastcall __sched sleep_on(wait_queue_head_t *q)
4036{
4037 SLEEP_ON_VAR
4038
4039 current->state = TASK_UNINTERRUPTIBLE;
4040
4041 SLEEP_ON_HEAD
4042 schedule();
4043 SLEEP_ON_TAIL
4044}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004045EXPORT_SYMBOL(sleep_on);
4046
4047long fastcall __sched sleep_on_timeout(wait_queue_head_t *q, long timeout)
4048{
4049 SLEEP_ON_VAR
4050
4051 current->state = TASK_UNINTERRUPTIBLE;
4052
4053 SLEEP_ON_HEAD
4054 timeout = schedule_timeout(timeout);
4055 SLEEP_ON_TAIL
4056
4057 return timeout;
4058}
4059
4060EXPORT_SYMBOL(sleep_on_timeout);
4061
Ingo Molnarb29739f2006-06-27 02:54:51 -07004062#ifdef CONFIG_RT_MUTEXES
4063
4064/*
4065 * rt_mutex_setprio - set the current priority of a task
4066 * @p: task
4067 * @prio: prio value (kernel-internal form)
4068 *
4069 * This function changes the 'effective' priority of a task. It does
4070 * not touch ->normal_prio like __setscheduler().
4071 *
4072 * Used by the rt_mutex code to implement priority inheritance logic.
4073 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004074void rt_mutex_setprio(struct task_struct *p, int prio)
Ingo Molnarb29739f2006-06-27 02:54:51 -07004075{
Ingo Molnar70b97a72006-07-03 00:25:42 -07004076 struct prio_array *array;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004077 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07004078 struct rq *rq;
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004079 int delta;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004080
4081 BUG_ON(prio < 0 || prio > MAX_PRIO);
4082
4083 rq = task_rq_lock(p, &flags);
4084
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004085 delta = prio - p->prio;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004086 array = p->array;
4087 if (array)
4088 dequeue_task(p, array);
4089 p->prio = prio;
4090
4091 if (array) {
4092 /*
4093 * If changing to an RT priority then queue it
4094 * in the active array!
4095 */
4096 if (rt_task(p))
4097 array = rq->active;
4098 enqueue_task(p, array);
4099 /*
4100 * Reschedule if we are currently running on this runqueue and
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004101 * our priority decreased, or if our priority became higher
4102 * than the current's.
Ingo Molnarb29739f2006-06-27 02:54:51 -07004103 */
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004104 if (TASK_PREEMPTS_CURR(p, rq) ||
4105 (delta > 0 && task_running(rq, p)))
Ingo Molnarb29739f2006-06-27 02:54:51 -07004106 resched_task(rq->curr);
4107 }
4108 task_rq_unlock(rq, &flags);
4109}
4110
4111#endif
4112
Ingo Molnar36c8b582006-07-03 00:25:41 -07004113void set_user_nice(struct task_struct *p, long nice)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004114{
Ingo Molnar70b97a72006-07-03 00:25:42 -07004115 struct prio_array *array;
Ingo Molnar48f24c42006-07-03 00:25:40 -07004116 int old_prio, delta;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004117 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07004118 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004119
4120 if (TASK_NICE(p) == nice || nice < -20 || nice > 19)
4121 return;
4122 /*
4123 * We have to be careful, if called from sys_setpriority(),
4124 * the task might be in the middle of scheduling on another CPU.
4125 */
4126 rq = task_rq_lock(p, &flags);
4127 /*
4128 * The RT priorities are set via sched_setscheduler(), but we still
4129 * allow the 'normal' nice value to be set - but as expected
4130 * it wont have any effect on scheduling until the task is
Ingo Molnarb0a94992006-01-14 13:20:41 -08004131 * not SCHED_NORMAL/SCHED_BATCH:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004132 */
Ingo Molnarb29739f2006-06-27 02:54:51 -07004133 if (has_rt_policy(p)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004134 p->static_prio = NICE_TO_PRIO(nice);
4135 goto out_unlock;
4136 }
4137 array = p->array;
Peter Williams2dd73a42006-06-27 02:54:34 -07004138 if (array) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004139 dequeue_task(p, array);
Peter Williams2dd73a42006-06-27 02:54:34 -07004140 dec_raw_weighted_load(rq, p);
4141 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004142
Linus Torvalds1da177e2005-04-16 15:20:36 -07004143 p->static_prio = NICE_TO_PRIO(nice);
Peter Williams2dd73a42006-06-27 02:54:34 -07004144 set_load_weight(p);
Ingo Molnarb29739f2006-06-27 02:54:51 -07004145 old_prio = p->prio;
4146 p->prio = effective_prio(p);
4147 delta = p->prio - old_prio;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004148
4149 if (array) {
4150 enqueue_task(p, array);
Peter Williams2dd73a42006-06-27 02:54:34 -07004151 inc_raw_weighted_load(rq, p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004152 /*
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004153 * Reschedule if we are currently running on this runqueue and
4154 * our priority decreased, or if our priority became higher
4155 * than the current's.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004156 */
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004157 if (TASK_PREEMPTS_CURR(p, rq) ||
4158 (delta > 0 && task_running(rq, p)))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004159 resched_task(rq->curr);
4160 }
4161out_unlock:
4162 task_rq_unlock(rq, &flags);
4163}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004164EXPORT_SYMBOL(set_user_nice);
4165
Matt Mackalle43379f2005-05-01 08:59:00 -07004166/*
4167 * can_nice - check if a task can reduce its nice value
4168 * @p: task
4169 * @nice: nice value
4170 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004171int can_nice(const struct task_struct *p, const int nice)
Matt Mackalle43379f2005-05-01 08:59:00 -07004172{
Matt Mackall024f4742005-08-18 11:24:19 -07004173 /* convert nice value [19,-20] to rlimit style value [1,40] */
4174 int nice_rlim = 20 - nice;
Ingo Molnar48f24c42006-07-03 00:25:40 -07004175
Matt Mackalle43379f2005-05-01 08:59:00 -07004176 return (nice_rlim <= p->signal->rlim[RLIMIT_NICE].rlim_cur ||
4177 capable(CAP_SYS_NICE));
4178}
4179
Linus Torvalds1da177e2005-04-16 15:20:36 -07004180#ifdef __ARCH_WANT_SYS_NICE
4181
4182/*
4183 * sys_nice - change the priority of the current process.
4184 * @increment: priority increment
4185 *
4186 * sys_setpriority is a more generic, but much slower function that
4187 * does similar things.
4188 */
4189asmlinkage long sys_nice(int increment)
4190{
Ingo Molnar48f24c42006-07-03 00:25:40 -07004191 long nice, retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004192
4193 /*
4194 * Setpriority might change our priority at the same moment.
4195 * We don't have to worry. Conceptually one call occurs first
4196 * and we have a single winner.
4197 */
Matt Mackalle43379f2005-05-01 08:59:00 -07004198 if (increment < -40)
4199 increment = -40;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004200 if (increment > 40)
4201 increment = 40;
4202
4203 nice = PRIO_TO_NICE(current->static_prio) + increment;
4204 if (nice < -20)
4205 nice = -20;
4206 if (nice > 19)
4207 nice = 19;
4208
Matt Mackalle43379f2005-05-01 08:59:00 -07004209 if (increment < 0 && !can_nice(current, nice))
4210 return -EPERM;
4211
Linus Torvalds1da177e2005-04-16 15:20:36 -07004212 retval = security_task_setnice(current, nice);
4213 if (retval)
4214 return retval;
4215
4216 set_user_nice(current, nice);
4217 return 0;
4218}
4219
4220#endif
4221
4222/**
4223 * task_prio - return the priority value of a given task.
4224 * @p: the task in question.
4225 *
4226 * This is the priority value as seen by users in /proc.
4227 * RT tasks are offset by -200. Normal tasks are centered
4228 * around 0, value goes from -16 to +15.
4229 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004230int task_prio(const struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004231{
4232 return p->prio - MAX_RT_PRIO;
4233}
4234
4235/**
4236 * task_nice - return the nice value of a given task.
4237 * @p: the task in question.
4238 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004239int task_nice(const struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004240{
4241 return TASK_NICE(p);
4242}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004243EXPORT_SYMBOL_GPL(task_nice);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004244
4245/**
4246 * idle_cpu - is a given cpu idle currently?
4247 * @cpu: the processor in question.
4248 */
4249int idle_cpu(int cpu)
4250{
4251 return cpu_curr(cpu) == cpu_rq(cpu)->idle;
4252}
4253
Linus Torvalds1da177e2005-04-16 15:20:36 -07004254/**
4255 * idle_task - return the idle task for a given cpu.
4256 * @cpu: the processor in question.
4257 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004258struct task_struct *idle_task(int cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004259{
4260 return cpu_rq(cpu)->idle;
4261}
4262
4263/**
4264 * find_process_by_pid - find a process with a matching PID value.
4265 * @pid: the pid in question.
4266 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07004267static inline struct task_struct *find_process_by_pid(pid_t pid)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004268{
4269 return pid ? find_task_by_pid(pid) : current;
4270}
4271
4272/* Actually do priority change: must hold rq lock. */
4273static void __setscheduler(struct task_struct *p, int policy, int prio)
4274{
4275 BUG_ON(p->array);
Ingo Molnar48f24c42006-07-03 00:25:40 -07004276
Linus Torvalds1da177e2005-04-16 15:20:36 -07004277 p->policy = policy;
4278 p->rt_priority = prio;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004279 p->normal_prio = normal_prio(p);
4280 /* we are holding p->pi_lock already */
4281 p->prio = rt_mutex_getprio(p);
4282 /*
4283 * SCHED_BATCH tasks are treated as perpetual CPU hogs:
4284 */
4285 if (policy == SCHED_BATCH)
4286 p->sleep_avg = 0;
Peter Williams2dd73a42006-06-27 02:54:34 -07004287 set_load_weight(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004288}
4289
4290/**
Robert P. J. Day72fd4a32007-02-10 01:45:59 -08004291 * sched_setscheduler - change the scheduling policy and/or RT priority of a thread.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004292 * @p: the task in question.
4293 * @policy: new policy.
4294 * @param: structure containing the new RT priority.
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004295 *
Robert P. J. Day72fd4a32007-02-10 01:45:59 -08004296 * NOTE that the task may be already dead.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004297 */
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004298int sched_setscheduler(struct task_struct *p, int policy,
4299 struct sched_param *param)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004300{
Ingo Molnar48f24c42006-07-03 00:25:40 -07004301 int retval, oldprio, oldpolicy = -1;
Ingo Molnar70b97a72006-07-03 00:25:42 -07004302 struct prio_array *array;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004303 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07004304 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004305
Steven Rostedt66e53932006-06-27 02:54:44 -07004306 /* may grab non-irq protected spin_locks */
4307 BUG_ON(in_interrupt());
Linus Torvalds1da177e2005-04-16 15:20:36 -07004308recheck:
4309 /* double check policy once rq lock held */
4310 if (policy < 0)
4311 policy = oldpolicy = p->policy;
4312 else if (policy != SCHED_FIFO && policy != SCHED_RR &&
Ingo Molnarb0a94992006-01-14 13:20:41 -08004313 policy != SCHED_NORMAL && policy != SCHED_BATCH)
4314 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004315 /*
4316 * Valid priorities for SCHED_FIFO and SCHED_RR are
Ingo Molnarb0a94992006-01-14 13:20:41 -08004317 * 1..MAX_USER_RT_PRIO-1, valid priority for SCHED_NORMAL and
4318 * SCHED_BATCH is 0.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004319 */
4320 if (param->sched_priority < 0 ||
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004321 (p->mm && param->sched_priority > MAX_USER_RT_PRIO-1) ||
Steven Rostedtd46523e2005-07-25 16:28:39 -04004322 (!p->mm && param->sched_priority > MAX_RT_PRIO-1))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004323 return -EINVAL;
Oleg Nesterov57a6f512006-09-29 02:00:49 -07004324 if (is_rt_policy(policy) != (param->sched_priority != 0))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004325 return -EINVAL;
4326
Olivier Croquette37e4ab32005-06-25 14:57:32 -07004327 /*
4328 * Allow unprivileged RT tasks to decrease priority:
4329 */
4330 if (!capable(CAP_SYS_NICE)) {
Oleg Nesterov8dc3e902006-09-29 02:00:50 -07004331 if (is_rt_policy(policy)) {
4332 unsigned long rlim_rtprio;
4333 unsigned long flags;
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004334
Oleg Nesterov8dc3e902006-09-29 02:00:50 -07004335 if (!lock_task_sighand(p, &flags))
4336 return -ESRCH;
4337 rlim_rtprio = p->signal->rlim[RLIMIT_RTPRIO].rlim_cur;
4338 unlock_task_sighand(p, &flags);
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004339
Oleg Nesterov8dc3e902006-09-29 02:00:50 -07004340 /* can't set/change the rt policy */
4341 if (policy != p->policy && !rlim_rtprio)
4342 return -EPERM;
4343
4344 /* can't increase priority */
4345 if (param->sched_priority > p->rt_priority &&
4346 param->sched_priority > rlim_rtprio)
4347 return -EPERM;
4348 }
4349
Olivier Croquette37e4ab32005-06-25 14:57:32 -07004350 /* can't change other user's priorities */
4351 if ((current->euid != p->euid) &&
4352 (current->euid != p->uid))
4353 return -EPERM;
4354 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07004355
4356 retval = security_task_setscheduler(p, policy, param);
4357 if (retval)
4358 return retval;
4359 /*
Ingo Molnarb29739f2006-06-27 02:54:51 -07004360 * make sure no PI-waiters arrive (or leave) while we are
4361 * changing the priority of the task:
4362 */
4363 spin_lock_irqsave(&p->pi_lock, flags);
4364 /*
Linus Torvalds1da177e2005-04-16 15:20:36 -07004365 * To be able to change p->policy safely, the apropriate
4366 * runqueue lock must be held.
4367 */
Ingo Molnarb29739f2006-06-27 02:54:51 -07004368 rq = __task_rq_lock(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004369 /* recheck policy now with rq lock held */
4370 if (unlikely(oldpolicy != -1 && oldpolicy != p->policy)) {
4371 policy = oldpolicy = -1;
Ingo Molnarb29739f2006-06-27 02:54:51 -07004372 __task_rq_unlock(rq);
4373 spin_unlock_irqrestore(&p->pi_lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004374 goto recheck;
4375 }
4376 array = p->array;
4377 if (array)
4378 deactivate_task(p, rq);
4379 oldprio = p->prio;
4380 __setscheduler(p, policy, param->sched_priority);
4381 if (array) {
4382 __activate_task(p, rq);
4383 /*
4384 * Reschedule if we are currently running on this runqueue and
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004385 * our priority decreased, or our priority became higher
4386 * than the current's.
Linus Torvalds1da177e2005-04-16 15:20:36 -07004387 */
Dmitry Adamushkobd53f962007-05-08 00:33:06 -07004388 if (TASK_PREEMPTS_CURR(p, rq) ||
4389 (task_running(rq, p) && p->prio > oldprio))
Linus Torvalds1da177e2005-04-16 15:20:36 -07004390 resched_task(rq->curr);
4391 }
Ingo Molnarb29739f2006-06-27 02:54:51 -07004392 __task_rq_unlock(rq);
4393 spin_unlock_irqrestore(&p->pi_lock, flags);
4394
Thomas Gleixner95e02ca2006-06-27 02:55:02 -07004395 rt_mutex_adjust_pi(p);
4396
Linus Torvalds1da177e2005-04-16 15:20:36 -07004397 return 0;
4398}
4399EXPORT_SYMBOL_GPL(sched_setscheduler);
4400
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004401static int
4402do_sched_setscheduler(pid_t pid, int policy, struct sched_param __user *param)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004403{
Linus Torvalds1da177e2005-04-16 15:20:36 -07004404 struct sched_param lparam;
4405 struct task_struct *p;
Ingo Molnar36c8b582006-07-03 00:25:41 -07004406 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004407
4408 if (!param || pid < 0)
4409 return -EINVAL;
4410 if (copy_from_user(&lparam, param, sizeof(struct sched_param)))
4411 return -EFAULT;
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004412
4413 rcu_read_lock();
4414 retval = -ESRCH;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004415 p = find_process_by_pid(pid);
Oleg Nesterov5fe1d752006-09-29 02:00:48 -07004416 if (p != NULL)
4417 retval = sched_setscheduler(p, policy, &lparam);
4418 rcu_read_unlock();
Ingo Molnar36c8b582006-07-03 00:25:41 -07004419
Linus Torvalds1da177e2005-04-16 15:20:36 -07004420 return retval;
4421}
4422
4423/**
4424 * sys_sched_setscheduler - set/change the scheduler policy and RT priority
4425 * @pid: the pid in question.
4426 * @policy: new policy.
4427 * @param: structure containing the new RT priority.
4428 */
4429asmlinkage long sys_sched_setscheduler(pid_t pid, int policy,
4430 struct sched_param __user *param)
4431{
Jason Baronc21761f2006-01-18 17:43:03 -08004432 /* negative values for policy are not valid */
4433 if (policy < 0)
4434 return -EINVAL;
4435
Linus Torvalds1da177e2005-04-16 15:20:36 -07004436 return do_sched_setscheduler(pid, policy, param);
4437}
4438
4439/**
4440 * sys_sched_setparam - set/change the RT priority of a thread
4441 * @pid: the pid in question.
4442 * @param: structure containing the new RT priority.
4443 */
4444asmlinkage long sys_sched_setparam(pid_t pid, struct sched_param __user *param)
4445{
4446 return do_sched_setscheduler(pid, -1, param);
4447}
4448
4449/**
4450 * sys_sched_getscheduler - get the policy (scheduling class) of a thread
4451 * @pid: the pid in question.
4452 */
4453asmlinkage long sys_sched_getscheduler(pid_t pid)
4454{
Ingo Molnar36c8b582006-07-03 00:25:41 -07004455 struct task_struct *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004456 int retval = -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004457
4458 if (pid < 0)
4459 goto out_nounlock;
4460
4461 retval = -ESRCH;
4462 read_lock(&tasklist_lock);
4463 p = find_process_by_pid(pid);
4464 if (p) {
4465 retval = security_task_getscheduler(p);
4466 if (!retval)
4467 retval = p->policy;
4468 }
4469 read_unlock(&tasklist_lock);
4470
4471out_nounlock:
4472 return retval;
4473}
4474
4475/**
4476 * sys_sched_getscheduler - get the RT priority of a thread
4477 * @pid: the pid in question.
4478 * @param: structure containing the RT priority.
4479 */
4480asmlinkage long sys_sched_getparam(pid_t pid, struct sched_param __user *param)
4481{
4482 struct sched_param lp;
Ingo Molnar36c8b582006-07-03 00:25:41 -07004483 struct task_struct *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004484 int retval = -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004485
4486 if (!param || pid < 0)
4487 goto out_nounlock;
4488
4489 read_lock(&tasklist_lock);
4490 p = find_process_by_pid(pid);
4491 retval = -ESRCH;
4492 if (!p)
4493 goto out_unlock;
4494
4495 retval = security_task_getscheduler(p);
4496 if (retval)
4497 goto out_unlock;
4498
4499 lp.sched_priority = p->rt_priority;
4500 read_unlock(&tasklist_lock);
4501
4502 /*
4503 * This one might sleep, we cannot do it with a spinlock held ...
4504 */
4505 retval = copy_to_user(param, &lp, sizeof(*param)) ? -EFAULT : 0;
4506
4507out_nounlock:
4508 return retval;
4509
4510out_unlock:
4511 read_unlock(&tasklist_lock);
4512 return retval;
4513}
4514
4515long sched_setaffinity(pid_t pid, cpumask_t new_mask)
4516{
Linus Torvalds1da177e2005-04-16 15:20:36 -07004517 cpumask_t cpus_allowed;
Ingo Molnar36c8b582006-07-03 00:25:41 -07004518 struct task_struct *p;
4519 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004520
4521 lock_cpu_hotplug();
4522 read_lock(&tasklist_lock);
4523
4524 p = find_process_by_pid(pid);
4525 if (!p) {
4526 read_unlock(&tasklist_lock);
4527 unlock_cpu_hotplug();
4528 return -ESRCH;
4529 }
4530
4531 /*
4532 * It is not safe to call set_cpus_allowed with the
4533 * tasklist_lock held. We will bump the task_struct's
4534 * usage count and then drop tasklist_lock.
4535 */
4536 get_task_struct(p);
4537 read_unlock(&tasklist_lock);
4538
4539 retval = -EPERM;
4540 if ((current->euid != p->euid) && (current->euid != p->uid) &&
4541 !capable(CAP_SYS_NICE))
4542 goto out_unlock;
4543
David Quigleye7834f82006-06-23 02:03:59 -07004544 retval = security_task_setscheduler(p, 0, NULL);
4545 if (retval)
4546 goto out_unlock;
4547
Linus Torvalds1da177e2005-04-16 15:20:36 -07004548 cpus_allowed = cpuset_cpus_allowed(p);
4549 cpus_and(new_mask, new_mask, cpus_allowed);
4550 retval = set_cpus_allowed(p, new_mask);
4551
4552out_unlock:
4553 put_task_struct(p);
4554 unlock_cpu_hotplug();
4555 return retval;
4556}
4557
4558static int get_user_cpu_mask(unsigned long __user *user_mask_ptr, unsigned len,
4559 cpumask_t *new_mask)
4560{
4561 if (len < sizeof(cpumask_t)) {
4562 memset(new_mask, 0, sizeof(cpumask_t));
4563 } else if (len > sizeof(cpumask_t)) {
4564 len = sizeof(cpumask_t);
4565 }
4566 return copy_from_user(new_mask, user_mask_ptr, len) ? -EFAULT : 0;
4567}
4568
4569/**
4570 * sys_sched_setaffinity - set the cpu affinity of a process
4571 * @pid: pid of the process
4572 * @len: length in bytes of the bitmask pointed to by user_mask_ptr
4573 * @user_mask_ptr: user-space pointer to the new cpu mask
4574 */
4575asmlinkage long sys_sched_setaffinity(pid_t pid, unsigned int len,
4576 unsigned long __user *user_mask_ptr)
4577{
4578 cpumask_t new_mask;
4579 int retval;
4580
4581 retval = get_user_cpu_mask(user_mask_ptr, len, &new_mask);
4582 if (retval)
4583 return retval;
4584
4585 return sched_setaffinity(pid, new_mask);
4586}
4587
4588/*
4589 * Represents all cpu's present in the system
4590 * In systems capable of hotplug, this map could dynamically grow
4591 * as new cpu's are detected in the system via any platform specific
4592 * method, such as ACPI for e.g.
4593 */
4594
Andi Kleen4cef0c62006-01-11 22:44:57 +01004595cpumask_t cpu_present_map __read_mostly;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004596EXPORT_SYMBOL(cpu_present_map);
4597
4598#ifndef CONFIG_SMP
Andi Kleen4cef0c62006-01-11 22:44:57 +01004599cpumask_t cpu_online_map __read_mostly = CPU_MASK_ALL;
Greg Bankse16b38f2006-10-02 02:17:40 -07004600EXPORT_SYMBOL(cpu_online_map);
4601
Andi Kleen4cef0c62006-01-11 22:44:57 +01004602cpumask_t cpu_possible_map __read_mostly = CPU_MASK_ALL;
Greg Bankse16b38f2006-10-02 02:17:40 -07004603EXPORT_SYMBOL(cpu_possible_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004604#endif
4605
4606long sched_getaffinity(pid_t pid, cpumask_t *mask)
4607{
Ingo Molnar36c8b582006-07-03 00:25:41 -07004608 struct task_struct *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004609 int retval;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004610
4611 lock_cpu_hotplug();
4612 read_lock(&tasklist_lock);
4613
4614 retval = -ESRCH;
4615 p = find_process_by_pid(pid);
4616 if (!p)
4617 goto out_unlock;
4618
David Quigleye7834f82006-06-23 02:03:59 -07004619 retval = security_task_getscheduler(p);
4620 if (retval)
4621 goto out_unlock;
4622
Jack Steiner2f7016d2006-02-01 03:05:18 -08004623 cpus_and(*mask, p->cpus_allowed, cpu_online_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004624
4625out_unlock:
4626 read_unlock(&tasklist_lock);
4627 unlock_cpu_hotplug();
4628 if (retval)
4629 return retval;
4630
4631 return 0;
4632}
4633
4634/**
4635 * sys_sched_getaffinity - get the cpu affinity of a process
4636 * @pid: pid of the process
4637 * @len: length in bytes of the bitmask pointed to by user_mask_ptr
4638 * @user_mask_ptr: user-space pointer to hold the current cpu mask
4639 */
4640asmlinkage long sys_sched_getaffinity(pid_t pid, unsigned int len,
4641 unsigned long __user *user_mask_ptr)
4642{
4643 int ret;
4644 cpumask_t mask;
4645
4646 if (len < sizeof(cpumask_t))
4647 return -EINVAL;
4648
4649 ret = sched_getaffinity(pid, &mask);
4650 if (ret < 0)
4651 return ret;
4652
4653 if (copy_to_user(user_mask_ptr, &mask, sizeof(cpumask_t)))
4654 return -EFAULT;
4655
4656 return sizeof(cpumask_t);
4657}
4658
4659/**
4660 * sys_sched_yield - yield the current processor to other threads.
4661 *
Robert P. J. Day72fd4a32007-02-10 01:45:59 -08004662 * This function yields the current CPU by moving the calling thread
Linus Torvalds1da177e2005-04-16 15:20:36 -07004663 * to the expired array. If there are no other threads running on this
4664 * CPU then this function will return.
4665 */
4666asmlinkage long sys_sched_yield(void)
4667{
Ingo Molnar70b97a72006-07-03 00:25:42 -07004668 struct rq *rq = this_rq_lock();
4669 struct prio_array *array = current->array, *target = rq->expired;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004670
4671 schedstat_inc(rq, yld_cnt);
4672 /*
4673 * We implement yielding by moving the task into the expired
4674 * queue.
4675 *
4676 * (special rule: RT tasks will just roundrobin in the active
4677 * array.)
4678 */
4679 if (rt_task(current))
4680 target = rq->active;
4681
Renaud Lienhart5927ad72005-09-10 00:26:20 -07004682 if (array->nr_active == 1) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004683 schedstat_inc(rq, yld_act_empty);
4684 if (!rq->expired->nr_active)
4685 schedstat_inc(rq, yld_both_empty);
4686 } else if (!rq->expired->nr_active)
4687 schedstat_inc(rq, yld_exp_empty);
4688
4689 if (array != target) {
4690 dequeue_task(current, array);
4691 enqueue_task(current, target);
4692 } else
4693 /*
4694 * requeue_task is cheaper so perform that if possible.
4695 */
4696 requeue_task(current, array);
4697
4698 /*
4699 * Since we are going to call schedule() anyway, there's
4700 * no need to preempt or enable interrupts:
4701 */
4702 __release(rq->lock);
Ingo Molnar8a25d5d2006-07-03 00:24:54 -07004703 spin_release(&rq->lock.dep_map, 1, _THIS_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004704 _raw_spin_unlock(&rq->lock);
4705 preempt_enable_no_resched();
4706
4707 schedule();
4708
4709 return 0;
4710}
4711
Andrew Mortone7b38402006-06-30 01:56:00 -07004712static void __cond_resched(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004713{
Ingo Molnar8e0a43d2006-06-23 02:05:23 -07004714#ifdef CONFIG_DEBUG_SPINLOCK_SLEEP
4715 __might_sleep(__FILE__, __LINE__);
4716#endif
Ingo Molnar5bbcfd92005-07-07 17:57:04 -07004717 /*
4718 * The BKS might be reacquired before we have dropped
4719 * PREEMPT_ACTIVE, which could trigger a second
4720 * cond_resched() call.
4721 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07004722 do {
4723 add_preempt_count(PREEMPT_ACTIVE);
4724 schedule();
4725 sub_preempt_count(PREEMPT_ACTIVE);
4726 } while (need_resched());
4727}
4728
4729int __sched cond_resched(void)
4730{
Ingo Molnar94142322006-12-29 16:48:13 -08004731 if (need_resched() && !(preempt_count() & PREEMPT_ACTIVE) &&
4732 system_state == SYSTEM_RUNNING) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07004733 __cond_resched();
4734 return 1;
4735 }
4736 return 0;
4737}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004738EXPORT_SYMBOL(cond_resched);
4739
4740/*
4741 * cond_resched_lock() - if a reschedule is pending, drop the given lock,
4742 * call schedule, and on return reacquire the lock.
4743 *
4744 * This works OK both with and without CONFIG_PREEMPT. We do strange low-level
4745 * operations here to prevent schedule() from being called twice (once via
4746 * spin_unlock(), once by hand).
4747 */
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07004748int cond_resched_lock(spinlock_t *lock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004749{
Jan Kara6df3cec2005-06-13 15:52:32 -07004750 int ret = 0;
4751
Linus Torvalds1da177e2005-04-16 15:20:36 -07004752 if (need_lockbreak(lock)) {
4753 spin_unlock(lock);
4754 cpu_relax();
Jan Kara6df3cec2005-06-13 15:52:32 -07004755 ret = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004756 spin_lock(lock);
4757 }
Ingo Molnar94142322006-12-29 16:48:13 -08004758 if (need_resched() && system_state == SYSTEM_RUNNING) {
Ingo Molnar8a25d5d2006-07-03 00:24:54 -07004759 spin_release(&lock->dep_map, 1, _THIS_IP_);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004760 _raw_spin_unlock(lock);
4761 preempt_enable_no_resched();
4762 __cond_resched();
Jan Kara6df3cec2005-06-13 15:52:32 -07004763 ret = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004764 spin_lock(lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004765 }
Jan Kara6df3cec2005-06-13 15:52:32 -07004766 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004767}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004768EXPORT_SYMBOL(cond_resched_lock);
4769
4770int __sched cond_resched_softirq(void)
4771{
4772 BUG_ON(!in_softirq());
4773
Ingo Molnar94142322006-12-29 16:48:13 -08004774 if (need_resched() && system_state == SYSTEM_RUNNING) {
Ingo Molnarde30a2b2006-07-03 00:24:42 -07004775 raw_local_irq_disable();
4776 _local_bh_enable();
4777 raw_local_irq_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004778 __cond_resched();
4779 local_bh_disable();
4780 return 1;
4781 }
4782 return 0;
4783}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004784EXPORT_SYMBOL(cond_resched_softirq);
4785
Linus Torvalds1da177e2005-04-16 15:20:36 -07004786/**
4787 * yield - yield the current processor to other threads.
4788 *
Robert P. J. Day72fd4a32007-02-10 01:45:59 -08004789 * This is a shortcut for kernel-space yielding - it marks the
Linus Torvalds1da177e2005-04-16 15:20:36 -07004790 * thread runnable and calls sys_sched_yield().
4791 */
4792void __sched yield(void)
4793{
4794 set_current_state(TASK_RUNNING);
4795 sys_sched_yield();
4796}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004797EXPORT_SYMBOL(yield);
4798
4799/*
4800 * This task is about to go to sleep on IO. Increment rq->nr_iowait so
4801 * that process accounting knows that this is a task in IO wait state.
4802 *
4803 * But don't do that if it is a deliberate, throttling IO wait (this task
4804 * has set its backing_dev_info: the queue against which it should throttle)
4805 */
4806void __sched io_schedule(void)
4807{
Ingo Molnar70b97a72006-07-03 00:25:42 -07004808 struct rq *rq = &__raw_get_cpu_var(runqueues);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004809
Shailabh Nagar0ff92242006-07-14 00:24:37 -07004810 delayacct_blkio_start();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004811 atomic_inc(&rq->nr_iowait);
4812 schedule();
4813 atomic_dec(&rq->nr_iowait);
Shailabh Nagar0ff92242006-07-14 00:24:37 -07004814 delayacct_blkio_end();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004815}
Linus Torvalds1da177e2005-04-16 15:20:36 -07004816EXPORT_SYMBOL(io_schedule);
4817
4818long __sched io_schedule_timeout(long timeout)
4819{
Ingo Molnar70b97a72006-07-03 00:25:42 -07004820 struct rq *rq = &__raw_get_cpu_var(runqueues);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004821 long ret;
4822
Shailabh Nagar0ff92242006-07-14 00:24:37 -07004823 delayacct_blkio_start();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004824 atomic_inc(&rq->nr_iowait);
4825 ret = schedule_timeout(timeout);
4826 atomic_dec(&rq->nr_iowait);
Shailabh Nagar0ff92242006-07-14 00:24:37 -07004827 delayacct_blkio_end();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004828 return ret;
4829}
4830
4831/**
4832 * sys_sched_get_priority_max - return maximum RT priority.
4833 * @policy: scheduling class.
4834 *
4835 * this syscall returns the maximum rt_priority that can be used
4836 * by a given scheduling class.
4837 */
4838asmlinkage long sys_sched_get_priority_max(int policy)
4839{
4840 int ret = -EINVAL;
4841
4842 switch (policy) {
4843 case SCHED_FIFO:
4844 case SCHED_RR:
4845 ret = MAX_USER_RT_PRIO-1;
4846 break;
4847 case SCHED_NORMAL:
Ingo Molnarb0a94992006-01-14 13:20:41 -08004848 case SCHED_BATCH:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004849 ret = 0;
4850 break;
4851 }
4852 return ret;
4853}
4854
4855/**
4856 * sys_sched_get_priority_min - return minimum RT priority.
4857 * @policy: scheduling class.
4858 *
4859 * this syscall returns the minimum rt_priority that can be used
4860 * by a given scheduling class.
4861 */
4862asmlinkage long sys_sched_get_priority_min(int policy)
4863{
4864 int ret = -EINVAL;
4865
4866 switch (policy) {
4867 case SCHED_FIFO:
4868 case SCHED_RR:
4869 ret = 1;
4870 break;
4871 case SCHED_NORMAL:
Ingo Molnarb0a94992006-01-14 13:20:41 -08004872 case SCHED_BATCH:
Linus Torvalds1da177e2005-04-16 15:20:36 -07004873 ret = 0;
4874 }
4875 return ret;
4876}
4877
4878/**
4879 * sys_sched_rr_get_interval - return the default timeslice of a process.
4880 * @pid: pid of the process.
4881 * @interval: userspace pointer to the timeslice value.
4882 *
4883 * this syscall writes the default timeslice value of a given process
4884 * into the user-space timespec buffer. A value of '0' means infinity.
4885 */
4886asmlinkage
4887long sys_sched_rr_get_interval(pid_t pid, struct timespec __user *interval)
4888{
Ingo Molnar36c8b582006-07-03 00:25:41 -07004889 struct task_struct *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004890 int retval = -EINVAL;
4891 struct timespec t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004892
4893 if (pid < 0)
4894 goto out_nounlock;
4895
4896 retval = -ESRCH;
4897 read_lock(&tasklist_lock);
4898 p = find_process_by_pid(pid);
4899 if (!p)
4900 goto out_unlock;
4901
4902 retval = security_task_getscheduler(p);
4903 if (retval)
4904 goto out_unlock;
4905
Peter Williamsb78709c2006-06-26 16:58:00 +10004906 jiffies_to_timespec(p->policy == SCHED_FIFO ?
Linus Torvalds1da177e2005-04-16 15:20:36 -07004907 0 : task_timeslice(p), &t);
4908 read_unlock(&tasklist_lock);
4909 retval = copy_to_user(interval, &t, sizeof(t)) ? -EFAULT : 0;
4910out_nounlock:
4911 return retval;
4912out_unlock:
4913 read_unlock(&tasklist_lock);
4914 return retval;
4915}
4916
Andreas Mohr2ed6e342006-07-10 04:43:52 -07004917static const char stat_nam[] = "RSDTtZX";
Ingo Molnar36c8b582006-07-03 00:25:41 -07004918
4919static void show_task(struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004920{
Linus Torvalds1da177e2005-04-16 15:20:36 -07004921 unsigned long free = 0;
Ingo Molnar36c8b582006-07-03 00:25:41 -07004922 unsigned state;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004923
Linus Torvalds1da177e2005-04-16 15:20:36 -07004924 state = p->state ? __ffs(p->state) + 1 : 0;
Andreas Mohr2ed6e342006-07-10 04:43:52 -07004925 printk("%-13.13s %c", p->comm,
4926 state < sizeof(stat_nam) - 1 ? stat_nam[state] : '?');
Linus Torvalds1da177e2005-04-16 15:20:36 -07004927#if (BITS_PER_LONG == 32)
4928 if (state == TASK_RUNNING)
4929 printk(" running ");
4930 else
4931 printk(" %08lX ", thread_saved_pc(p));
4932#else
4933 if (state == TASK_RUNNING)
4934 printk(" running task ");
4935 else
4936 printk(" %016lx ", thread_saved_pc(p));
4937#endif
4938#ifdef CONFIG_DEBUG_STACK_USAGE
4939 {
Al Viro10ebffd2005-11-13 16:06:56 -08004940 unsigned long *n = end_of_stack(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004941 while (!*n)
4942 n++;
Al Viro10ebffd2005-11-13 16:06:56 -08004943 free = (unsigned long)n - (unsigned long)end_of_stack(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004944 }
4945#endif
Ingo Molnar35f6f752007-04-06 21:18:06 +02004946 printk("%5lu %5d %6d", free, p->pid, p->parent->pid);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004947 if (!p->mm)
4948 printk(" (L-TLB)\n");
4949 else
4950 printk(" (NOTLB)\n");
4951
4952 if (state != TASK_RUNNING)
4953 show_stack(p, NULL);
4954}
4955
Ingo Molnare59e2ae2006-12-06 20:35:59 -08004956void show_state_filter(unsigned long state_filter)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004957{
Ingo Molnar36c8b582006-07-03 00:25:41 -07004958 struct task_struct *g, *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07004959
4960#if (BITS_PER_LONG == 32)
4961 printk("\n"
Chris Caputo301827a2006-12-06 20:39:11 -08004962 " free sibling\n");
4963 printk(" task PC stack pid father child younger older\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07004964#else
4965 printk("\n"
Chris Caputo301827a2006-12-06 20:39:11 -08004966 " free sibling\n");
4967 printk(" task PC stack pid father child younger older\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07004968#endif
4969 read_lock(&tasklist_lock);
4970 do_each_thread(g, p) {
4971 /*
4972 * reset the NMI-timeout, listing all files on a slow
4973 * console might take alot of time:
4974 */
4975 touch_nmi_watchdog();
Ingo Molnar39bc89f2007-04-25 20:50:03 -07004976 if (!state_filter || (p->state & state_filter))
Ingo Molnare59e2ae2006-12-06 20:35:59 -08004977 show_task(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07004978 } while_each_thread(g, p);
4979
Jeremy Fitzhardinge04c91672007-05-08 00:28:05 -07004980 touch_all_softlockup_watchdogs();
4981
Linus Torvalds1da177e2005-04-16 15:20:36 -07004982 read_unlock(&tasklist_lock);
Ingo Molnare59e2ae2006-12-06 20:35:59 -08004983 /*
4984 * Only show locks if all tasks are dumped:
4985 */
4986 if (state_filter == -1)
4987 debug_show_all_locks();
Linus Torvalds1da177e2005-04-16 15:20:36 -07004988}
4989
Ingo Molnarf340c0d2005-06-28 16:40:42 +02004990/**
4991 * init_idle - set up an idle thread for a given CPU
4992 * @idle: task in question
4993 * @cpu: cpu the idle task belongs to
4994 *
4995 * NOTE: this function does not set the idle thread's NEED_RESCHED
4996 * flag, to make booting more robust.
4997 */
Nick Piggin5c1e1762006-10-03 01:14:04 -07004998void __cpuinit init_idle(struct task_struct *idle, int cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07004999{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005000 struct rq *rq = cpu_rq(cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005001 unsigned long flags;
5002
Ingo Molnar81c29a82006-03-07 21:55:27 -08005003 idle->timestamp = sched_clock();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005004 idle->sleep_avg = 0;
5005 idle->array = NULL;
Ingo Molnarb29739f2006-06-27 02:54:51 -07005006 idle->prio = idle->normal_prio = MAX_PRIO;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005007 idle->state = TASK_RUNNING;
5008 idle->cpus_allowed = cpumask_of_cpu(cpu);
5009 set_task_cpu(idle, cpu);
5010
5011 spin_lock_irqsave(&rq->lock, flags);
5012 rq->curr = rq->idle = idle;
Nick Piggin4866cde2005-06-25 14:57:23 -07005013#if defined(CONFIG_SMP) && defined(__ARCH_WANT_UNLOCKED_CTXSW)
5014 idle->oncpu = 1;
5015#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07005016 spin_unlock_irqrestore(&rq->lock, flags);
5017
5018 /* Set the preempt count _outside_ the spinlocks! */
5019#if defined(CONFIG_PREEMPT) && !defined(CONFIG_PREEMPT_BKL)
Al Viroa1261f52005-11-13 16:06:55 -08005020 task_thread_info(idle)->preempt_count = (idle->lock_depth >= 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005021#else
Al Viroa1261f52005-11-13 16:06:55 -08005022 task_thread_info(idle)->preempt_count = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005023#endif
5024}
5025
5026/*
5027 * In a system that switches off the HZ timer nohz_cpu_mask
5028 * indicates which cpus entered this state. This is used
5029 * in the rcu update to wait only for active cpus. For system
5030 * which do not switch off the HZ timer nohz_cpu_mask should
5031 * always be CPU_MASK_NONE.
5032 */
5033cpumask_t nohz_cpu_mask = CPU_MASK_NONE;
5034
5035#ifdef CONFIG_SMP
5036/*
5037 * This is how migration works:
5038 *
Ingo Molnar70b97a72006-07-03 00:25:42 -07005039 * 1) we queue a struct migration_req structure in the source CPU's
Linus Torvalds1da177e2005-04-16 15:20:36 -07005040 * runqueue and wake up that CPU's migration thread.
5041 * 2) we down() the locked semaphore => thread blocks.
5042 * 3) migration thread wakes up (implicitly it forces the migrated
5043 * thread off the CPU)
5044 * 4) it gets the migration request and checks whether the migrated
5045 * task is still in the wrong runqueue.
5046 * 5) if it's in the wrong runqueue then the migration thread removes
5047 * it and puts it into the right queue.
5048 * 6) migration thread up()s the semaphore.
5049 * 7) we wake up and the migration is done.
5050 */
5051
5052/*
5053 * Change a given task's CPU affinity. Migrate the thread to a
5054 * proper CPU and schedule it away if the CPU it's executing on
5055 * is removed from the allowed bitmask.
5056 *
5057 * NOTE: the caller must have a valid reference to the task, the
5058 * task must not exit() & deallocate itself prematurely. The
5059 * call is not atomic; no spinlocks may be held.
5060 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07005061int set_cpus_allowed(struct task_struct *p, cpumask_t new_mask)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005062{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005063 struct migration_req req;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005064 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07005065 struct rq *rq;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005066 int ret = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005067
5068 rq = task_rq_lock(p, &flags);
5069 if (!cpus_intersects(new_mask, cpu_online_map)) {
5070 ret = -EINVAL;
5071 goto out;
5072 }
5073
5074 p->cpus_allowed = new_mask;
5075 /* Can the task run on the task's current CPU? If so, we're done */
5076 if (cpu_isset(task_cpu(p), new_mask))
5077 goto out;
5078
5079 if (migrate_task(p, any_online_cpu(new_mask), &req)) {
5080 /* Need help from migration thread: drop lock and wait. */
5081 task_rq_unlock(rq, &flags);
5082 wake_up_process(rq->migration_thread);
5083 wait_for_completion(&req.done);
5084 tlb_migrate_finish(p->mm);
5085 return 0;
5086 }
5087out:
5088 task_rq_unlock(rq, &flags);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005089
Linus Torvalds1da177e2005-04-16 15:20:36 -07005090 return ret;
5091}
Linus Torvalds1da177e2005-04-16 15:20:36 -07005092EXPORT_SYMBOL_GPL(set_cpus_allowed);
5093
5094/*
5095 * Move (not current) task off this cpu, onto dest cpu. We're doing
5096 * this because either it can't run here any more (set_cpus_allowed()
5097 * away from this CPU, or CPU going down), or because we're
5098 * attempting to rebalance this task on exec (sched_exec).
5099 *
5100 * So we race with normal scheduler movements, but that's OK, as long
5101 * as the task is no longer on this CPU.
Kirill Korotaevefc30812006-06-27 02:54:32 -07005102 *
5103 * Returns non-zero if task was successfully migrated.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005104 */
Kirill Korotaevefc30812006-06-27 02:54:32 -07005105static int __migrate_task(struct task_struct *p, int src_cpu, int dest_cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005106{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005107 struct rq *rq_dest, *rq_src;
Kirill Korotaevefc30812006-06-27 02:54:32 -07005108 int ret = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005109
5110 if (unlikely(cpu_is_offline(dest_cpu)))
Kirill Korotaevefc30812006-06-27 02:54:32 -07005111 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005112
5113 rq_src = cpu_rq(src_cpu);
5114 rq_dest = cpu_rq(dest_cpu);
5115
5116 double_rq_lock(rq_src, rq_dest);
5117 /* Already moved. */
5118 if (task_cpu(p) != src_cpu)
5119 goto out;
5120 /* Affinity changed (again). */
5121 if (!cpu_isset(dest_cpu, p->cpus_allowed))
5122 goto out;
5123
5124 set_task_cpu(p, dest_cpu);
5125 if (p->array) {
5126 /*
5127 * Sync timestamp with rq_dest's before activating.
5128 * The same thing could be achieved by doing this step
5129 * afterwards, and pretending it was a local activate.
5130 * This way is cleaner and logically correct.
5131 */
Mike Galbraithb18ec802006-12-10 02:20:31 -08005132 p->timestamp = p->timestamp - rq_src->most_recent_timestamp
5133 + rq_dest->most_recent_timestamp;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005134 deactivate_task(p, rq_src);
Peter Williams0a565f72006-07-10 04:43:51 -07005135 __activate_task(p, rq_dest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005136 if (TASK_PREEMPTS_CURR(p, rq_dest))
5137 resched_task(rq_dest->curr);
5138 }
Kirill Korotaevefc30812006-06-27 02:54:32 -07005139 ret = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005140out:
5141 double_rq_unlock(rq_src, rq_dest);
Kirill Korotaevefc30812006-06-27 02:54:32 -07005142 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005143}
5144
5145/*
5146 * migration_thread - this is a highprio system thread that performs
5147 * thread migration by bumping thread off CPU then 'pushing' onto
5148 * another runqueue.
5149 */
Ingo Molnar95cdf3b2005-09-10 00:26:11 -07005150static int migration_thread(void *data)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005151{
Linus Torvalds1da177e2005-04-16 15:20:36 -07005152 int cpu = (long)data;
Ingo Molnar70b97a72006-07-03 00:25:42 -07005153 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005154
5155 rq = cpu_rq(cpu);
5156 BUG_ON(rq->migration_thread != current);
5157
5158 set_current_state(TASK_INTERRUPTIBLE);
5159 while (!kthread_should_stop()) {
Ingo Molnar70b97a72006-07-03 00:25:42 -07005160 struct migration_req *req;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005161 struct list_head *head;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005162
Christoph Lameter3e1d1d22005-06-24 23:13:50 -07005163 try_to_freeze();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005164
5165 spin_lock_irq(&rq->lock);
5166
5167 if (cpu_is_offline(cpu)) {
5168 spin_unlock_irq(&rq->lock);
5169 goto wait_to_die;
5170 }
5171
5172 if (rq->active_balance) {
5173 active_load_balance(rq, cpu);
5174 rq->active_balance = 0;
5175 }
5176
5177 head = &rq->migration_queue;
5178
5179 if (list_empty(head)) {
5180 spin_unlock_irq(&rq->lock);
5181 schedule();
5182 set_current_state(TASK_INTERRUPTIBLE);
5183 continue;
5184 }
Ingo Molnar70b97a72006-07-03 00:25:42 -07005185 req = list_entry(head->next, struct migration_req, list);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005186 list_del_init(head->next);
5187
Nick Piggin674311d2005-06-25 14:57:27 -07005188 spin_unlock(&rq->lock);
5189 __migrate_task(req->task, cpu, req->dest_cpu);
5190 local_irq_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -07005191
5192 complete(&req->done);
5193 }
5194 __set_current_state(TASK_RUNNING);
5195 return 0;
5196
5197wait_to_die:
5198 /* Wait for kthread_stop */
5199 set_current_state(TASK_INTERRUPTIBLE);
5200 while (!kthread_should_stop()) {
5201 schedule();
5202 set_current_state(TASK_INTERRUPTIBLE);
5203 }
5204 __set_current_state(TASK_RUNNING);
5205 return 0;
5206}
5207
5208#ifdef CONFIG_HOTPLUG_CPU
Kirill Korotaev054b9102006-12-10 02:20:11 -08005209/*
5210 * Figure out where task on dead CPU should go, use force if neccessary.
5211 * NOTE: interrupts should be disabled by the caller
5212 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005213static void move_task_off_dead_cpu(int dead_cpu, struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005214{
Kirill Korotaevefc30812006-06-27 02:54:32 -07005215 unsigned long flags;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005216 cpumask_t mask;
Ingo Molnar70b97a72006-07-03 00:25:42 -07005217 struct rq *rq;
5218 int dest_cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005219
Kirill Korotaevefc30812006-06-27 02:54:32 -07005220restart:
Linus Torvalds1da177e2005-04-16 15:20:36 -07005221 /* On same node? */
5222 mask = node_to_cpumask(cpu_to_node(dead_cpu));
Ingo Molnar48f24c42006-07-03 00:25:40 -07005223 cpus_and(mask, mask, p->cpus_allowed);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005224 dest_cpu = any_online_cpu(mask);
5225
5226 /* On any allowed CPU? */
5227 if (dest_cpu == NR_CPUS)
Ingo Molnar48f24c42006-07-03 00:25:40 -07005228 dest_cpu = any_online_cpu(p->cpus_allowed);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005229
5230 /* No more Mr. Nice Guy. */
5231 if (dest_cpu == NR_CPUS) {
Ingo Molnar48f24c42006-07-03 00:25:40 -07005232 rq = task_rq_lock(p, &flags);
5233 cpus_setall(p->cpus_allowed);
5234 dest_cpu = any_online_cpu(p->cpus_allowed);
Kirill Korotaevefc30812006-06-27 02:54:32 -07005235 task_rq_unlock(rq, &flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005236
5237 /*
5238 * Don't tell them about moving exiting tasks or
5239 * kernel threads (both mm NULL), since they never
5240 * leave kernel.
5241 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005242 if (p->mm && printk_ratelimit())
Linus Torvalds1da177e2005-04-16 15:20:36 -07005243 printk(KERN_INFO "process %d (%s) no "
5244 "longer affine to cpu%d\n",
Ingo Molnar48f24c42006-07-03 00:25:40 -07005245 p->pid, p->comm, dead_cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005246 }
Ingo Molnar48f24c42006-07-03 00:25:40 -07005247 if (!__migrate_task(p, dead_cpu, dest_cpu))
Kirill Korotaevefc30812006-06-27 02:54:32 -07005248 goto restart;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005249}
5250
5251/*
5252 * While a dead CPU has no uninterruptible tasks queued at this point,
5253 * it might still have a nonzero ->nr_uninterruptible counter, because
5254 * for performance reasons the counter is not stricly tracking tasks to
5255 * their home CPUs. So we just add the counter to another CPU's counter,
5256 * to keep the global sum constant after CPU-down:
5257 */
Ingo Molnar70b97a72006-07-03 00:25:42 -07005258static void migrate_nr_uninterruptible(struct rq *rq_src)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005259{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005260 struct rq *rq_dest = cpu_rq(any_online_cpu(CPU_MASK_ALL));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005261 unsigned long flags;
5262
5263 local_irq_save(flags);
5264 double_rq_lock(rq_src, rq_dest);
5265 rq_dest->nr_uninterruptible += rq_src->nr_uninterruptible;
5266 rq_src->nr_uninterruptible = 0;
5267 double_rq_unlock(rq_src, rq_dest);
5268 local_irq_restore(flags);
5269}
5270
5271/* Run through task list and migrate tasks from the dead cpu. */
5272static void migrate_live_tasks(int src_cpu)
5273{
Ingo Molnar48f24c42006-07-03 00:25:40 -07005274 struct task_struct *p, *t;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005275
5276 write_lock_irq(&tasklist_lock);
5277
Ingo Molnar48f24c42006-07-03 00:25:40 -07005278 do_each_thread(t, p) {
5279 if (p == current)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005280 continue;
5281
Ingo Molnar48f24c42006-07-03 00:25:40 -07005282 if (task_cpu(p) == src_cpu)
5283 move_task_off_dead_cpu(src_cpu, p);
5284 } while_each_thread(t, p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005285
5286 write_unlock_irq(&tasklist_lock);
5287}
5288
5289/* Schedules idle task to be the next runnable task on current CPU.
5290 * It does so by boosting its priority to highest possible and adding it to
Ingo Molnar48f24c42006-07-03 00:25:40 -07005291 * the _front_ of the runqueue. Used by CPU offline code.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005292 */
5293void sched_idle_next(void)
5294{
Ingo Molnar48f24c42006-07-03 00:25:40 -07005295 int this_cpu = smp_processor_id();
Ingo Molnar70b97a72006-07-03 00:25:42 -07005296 struct rq *rq = cpu_rq(this_cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005297 struct task_struct *p = rq->idle;
5298 unsigned long flags;
5299
5300 /* cpu has to be offline */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005301 BUG_ON(cpu_online(this_cpu));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005302
Ingo Molnar48f24c42006-07-03 00:25:40 -07005303 /*
5304 * Strictly not necessary since rest of the CPUs are stopped by now
5305 * and interrupts disabled on the current cpu.
Linus Torvalds1da177e2005-04-16 15:20:36 -07005306 */
5307 spin_lock_irqsave(&rq->lock, flags);
5308
5309 __setscheduler(p, SCHED_FIFO, MAX_RT_PRIO-1);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005310
5311 /* Add idle task to the _front_ of its priority queue: */
Linus Torvalds1da177e2005-04-16 15:20:36 -07005312 __activate_idle_task(p, rq);
5313
5314 spin_unlock_irqrestore(&rq->lock, flags);
5315}
5316
Ingo Molnar48f24c42006-07-03 00:25:40 -07005317/*
5318 * Ensures that the idle task is using init_mm right before its cpu goes
Linus Torvalds1da177e2005-04-16 15:20:36 -07005319 * offline.
5320 */
5321void idle_task_exit(void)
5322{
5323 struct mm_struct *mm = current->active_mm;
5324
5325 BUG_ON(cpu_online(smp_processor_id()));
5326
5327 if (mm != &init_mm)
5328 switch_mm(mm, &init_mm, current);
5329 mmdrop(mm);
5330}
5331
Kirill Korotaev054b9102006-12-10 02:20:11 -08005332/* called under rq->lock with disabled interrupts */
Ingo Molnar36c8b582006-07-03 00:25:41 -07005333static void migrate_dead(unsigned int dead_cpu, struct task_struct *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005334{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005335 struct rq *rq = cpu_rq(dead_cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005336
5337 /* Must be exiting, otherwise would be on tasklist. */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005338 BUG_ON(p->exit_state != EXIT_ZOMBIE && p->exit_state != EXIT_DEAD);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005339
5340 /* Cannot have done final schedule yet: would have vanished. */
Oleg Nesterovc394cc92006-09-29 02:01:11 -07005341 BUG_ON(p->state == TASK_DEAD);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005342
Ingo Molnar48f24c42006-07-03 00:25:40 -07005343 get_task_struct(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005344
5345 /*
5346 * Drop lock around migration; if someone else moves it,
5347 * that's OK. No task can be added to this CPU, so iteration is
5348 * fine.
Kirill Korotaev054b9102006-12-10 02:20:11 -08005349 * NOTE: interrupts should be left disabled --dev@
Linus Torvalds1da177e2005-04-16 15:20:36 -07005350 */
Kirill Korotaev054b9102006-12-10 02:20:11 -08005351 spin_unlock(&rq->lock);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005352 move_task_off_dead_cpu(dead_cpu, p);
Kirill Korotaev054b9102006-12-10 02:20:11 -08005353 spin_lock(&rq->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005354
Ingo Molnar48f24c42006-07-03 00:25:40 -07005355 put_task_struct(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005356}
5357
5358/* release_task() removes task from tasklist, so we won't find dead tasks. */
5359static void migrate_dead_tasks(unsigned int dead_cpu)
5360{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005361 struct rq *rq = cpu_rq(dead_cpu);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005362 unsigned int arr, i;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005363
5364 for (arr = 0; arr < 2; arr++) {
5365 for (i = 0; i < MAX_PRIO; i++) {
5366 struct list_head *list = &rq->arrays[arr].queue[i];
Ingo Molnar48f24c42006-07-03 00:25:40 -07005367
Linus Torvalds1da177e2005-04-16 15:20:36 -07005368 while (!list_empty(list))
Ingo Molnar36c8b582006-07-03 00:25:41 -07005369 migrate_dead(dead_cpu, list_entry(list->next,
5370 struct task_struct, run_list));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005371 }
5372 }
5373}
5374#endif /* CONFIG_HOTPLUG_CPU */
5375
5376/*
5377 * migration_call - callback that gets triggered when a CPU is added.
5378 * Here we can start up the necessary migration thread for the new CPU.
5379 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005380static int __cpuinit
5381migration_call(struct notifier_block *nfb, unsigned long action, void *hcpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005382{
Linus Torvalds1da177e2005-04-16 15:20:36 -07005383 struct task_struct *p;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005384 int cpu = (long)hcpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005385 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07005386 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005387
5388 switch (action) {
5389 case CPU_UP_PREPARE:
5390 p = kthread_create(migration_thread, hcpu, "migration/%d",cpu);
5391 if (IS_ERR(p))
5392 return NOTIFY_BAD;
5393 p->flags |= PF_NOFREEZE;
5394 kthread_bind(p, cpu);
5395 /* Must be high prio: stop_machine expects to yield to it. */
5396 rq = task_rq_lock(p, &flags);
5397 __setscheduler(p, SCHED_FIFO, MAX_RT_PRIO-1);
5398 task_rq_unlock(rq, &flags);
5399 cpu_rq(cpu)->migration_thread = p;
5400 break;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005401
Linus Torvalds1da177e2005-04-16 15:20:36 -07005402 case CPU_ONLINE:
5403 /* Strictly unneccessary, as first user will wake it. */
5404 wake_up_process(cpu_rq(cpu)->migration_thread);
5405 break;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005406
Linus Torvalds1da177e2005-04-16 15:20:36 -07005407#ifdef CONFIG_HOTPLUG_CPU
5408 case CPU_UP_CANCELED:
Heiko Carstensfc75cdf2006-06-25 05:49:10 -07005409 if (!cpu_rq(cpu)->migration_thread)
5410 break;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005411 /* Unbind it from offline cpu so it can run. Fall thru. */
Heiko Carstensa4c4af72005-11-07 00:58:38 -08005412 kthread_bind(cpu_rq(cpu)->migration_thread,
5413 any_online_cpu(cpu_online_map));
Linus Torvalds1da177e2005-04-16 15:20:36 -07005414 kthread_stop(cpu_rq(cpu)->migration_thread);
5415 cpu_rq(cpu)->migration_thread = NULL;
5416 break;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005417
Linus Torvalds1da177e2005-04-16 15:20:36 -07005418 case CPU_DEAD:
5419 migrate_live_tasks(cpu);
5420 rq = cpu_rq(cpu);
5421 kthread_stop(rq->migration_thread);
5422 rq->migration_thread = NULL;
5423 /* Idle task back to normal (off runqueue, low prio) */
5424 rq = task_rq_lock(rq->idle, &flags);
5425 deactivate_task(rq->idle, rq);
5426 rq->idle->static_prio = MAX_PRIO;
5427 __setscheduler(rq->idle, SCHED_NORMAL, 0);
5428 migrate_dead_tasks(cpu);
5429 task_rq_unlock(rq, &flags);
5430 migrate_nr_uninterruptible(rq);
5431 BUG_ON(rq->nr_running != 0);
5432
5433 /* No need to migrate the tasks: it was best-effort if
5434 * they didn't do lock_cpu_hotplug(). Just wake up
5435 * the requestors. */
5436 spin_lock_irq(&rq->lock);
5437 while (!list_empty(&rq->migration_queue)) {
Ingo Molnar70b97a72006-07-03 00:25:42 -07005438 struct migration_req *req;
5439
Linus Torvalds1da177e2005-04-16 15:20:36 -07005440 req = list_entry(rq->migration_queue.next,
Ingo Molnar70b97a72006-07-03 00:25:42 -07005441 struct migration_req, list);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005442 list_del_init(&req->list);
5443 complete(&req->done);
5444 }
5445 spin_unlock_irq(&rq->lock);
5446 break;
5447#endif
5448 }
5449 return NOTIFY_OK;
5450}
5451
5452/* Register at highest priority so that task migration (migrate_all_tasks)
5453 * happens before everything else.
5454 */
Chandra Seetharaman26c21432006-06-27 02:54:10 -07005455static struct notifier_block __cpuinitdata migration_notifier = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07005456 .notifier_call = migration_call,
5457 .priority = 10
5458};
5459
5460int __init migration_init(void)
5461{
5462 void *cpu = (void *)(long)smp_processor_id();
Akinobu Mita07dccf32006-09-29 02:00:22 -07005463 int err;
Ingo Molnar48f24c42006-07-03 00:25:40 -07005464
5465 /* Start one for the boot CPU: */
Akinobu Mita07dccf32006-09-29 02:00:22 -07005466 err = migration_call(&migration_notifier, CPU_UP_PREPARE, cpu);
5467 BUG_ON(err == NOTIFY_BAD);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005468 migration_call(&migration_notifier, CPU_ONLINE, cpu);
5469 register_cpu_notifier(&migration_notifier);
Ingo Molnar48f24c42006-07-03 00:25:40 -07005470
Linus Torvalds1da177e2005-04-16 15:20:36 -07005471 return 0;
5472}
5473#endif
5474
5475#ifdef CONFIG_SMP
Christoph Lameter476f3532007-05-06 14:48:58 -07005476
5477/* Number of possible processor ids */
5478int nr_cpu_ids __read_mostly = NR_CPUS;
5479EXPORT_SYMBOL(nr_cpu_ids);
5480
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07005481#undef SCHED_DOMAIN_DEBUG
Linus Torvalds1da177e2005-04-16 15:20:36 -07005482#ifdef SCHED_DOMAIN_DEBUG
5483static void sched_domain_debug(struct sched_domain *sd, int cpu)
5484{
5485 int level = 0;
5486
Nick Piggin41c7ce92005-06-25 14:57:24 -07005487 if (!sd) {
5488 printk(KERN_DEBUG "CPU%d attaching NULL sched-domain.\n", cpu);
5489 return;
5490 }
5491
Linus Torvalds1da177e2005-04-16 15:20:36 -07005492 printk(KERN_DEBUG "CPU%d attaching sched-domain:\n", cpu);
5493
5494 do {
5495 int i;
5496 char str[NR_CPUS];
5497 struct sched_group *group = sd->groups;
5498 cpumask_t groupmask;
5499
5500 cpumask_scnprintf(str, NR_CPUS, sd->span);
5501 cpus_clear(groupmask);
5502
5503 printk(KERN_DEBUG);
5504 for (i = 0; i < level + 1; i++)
5505 printk(" ");
5506 printk("domain %d: ", level);
5507
5508 if (!(sd->flags & SD_LOAD_BALANCE)) {
5509 printk("does not load-balance\n");
5510 if (sd->parent)
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005511 printk(KERN_ERR "ERROR: !SD_LOAD_BALANCE domain"
5512 " has parent");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005513 break;
5514 }
5515
5516 printk("span %s\n", str);
5517
5518 if (!cpu_isset(cpu, sd->span))
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005519 printk(KERN_ERR "ERROR: domain->span does not contain "
5520 "CPU%d\n", cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005521 if (!cpu_isset(cpu, group->cpumask))
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005522 printk(KERN_ERR "ERROR: domain->groups does not contain"
5523 " CPU%d\n", cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005524
5525 printk(KERN_DEBUG);
5526 for (i = 0; i < level + 2; i++)
5527 printk(" ");
5528 printk("groups:");
5529 do {
5530 if (!group) {
5531 printk("\n");
5532 printk(KERN_ERR "ERROR: group is NULL\n");
5533 break;
5534 }
5535
Eric Dumazet5517d862007-05-08 00:32:57 -07005536 if (!group->__cpu_power) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07005537 printk("\n");
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005538 printk(KERN_ERR "ERROR: domain->cpu_power not "
5539 "set\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005540 }
5541
5542 if (!cpus_weight(group->cpumask)) {
5543 printk("\n");
5544 printk(KERN_ERR "ERROR: empty group\n");
5545 }
5546
5547 if (cpus_intersects(groupmask, group->cpumask)) {
5548 printk("\n");
5549 printk(KERN_ERR "ERROR: repeated CPUs\n");
5550 }
5551
5552 cpus_or(groupmask, groupmask, group->cpumask);
5553
5554 cpumask_scnprintf(str, NR_CPUS, group->cpumask);
5555 printk(" %s", str);
5556
5557 group = group->next;
5558 } while (group != sd->groups);
5559 printk("\n");
5560
5561 if (!cpus_equal(sd->span, groupmask))
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005562 printk(KERN_ERR "ERROR: groups don't span "
5563 "domain->span\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005564
5565 level++;
5566 sd = sd->parent;
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005567 if (!sd)
5568 continue;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005569
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005570 if (!cpus_subset(groupmask, sd->span))
5571 printk(KERN_ERR "ERROR: parent span is not a superset "
5572 "of domain->span\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -07005573
5574 } while (sd);
5575}
5576#else
Ingo Molnar48f24c42006-07-03 00:25:40 -07005577# define sched_domain_debug(sd, cpu) do { } while (0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005578#endif
5579
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07005580static int sd_degenerate(struct sched_domain *sd)
Suresh Siddha245af2c2005-06-25 14:57:25 -07005581{
5582 if (cpus_weight(sd->span) == 1)
5583 return 1;
5584
5585 /* Following flags need at least 2 groups */
5586 if (sd->flags & (SD_LOAD_BALANCE |
5587 SD_BALANCE_NEWIDLE |
5588 SD_BALANCE_FORK |
Siddha, Suresh B89c47102006-10-03 01:14:09 -07005589 SD_BALANCE_EXEC |
5590 SD_SHARE_CPUPOWER |
5591 SD_SHARE_PKG_RESOURCES)) {
Suresh Siddha245af2c2005-06-25 14:57:25 -07005592 if (sd->groups != sd->groups->next)
5593 return 0;
5594 }
5595
5596 /* Following flags don't use groups */
5597 if (sd->flags & (SD_WAKE_IDLE |
5598 SD_WAKE_AFFINE |
5599 SD_WAKE_BALANCE))
5600 return 0;
5601
5602 return 1;
5603}
5604
Ingo Molnar48f24c42006-07-03 00:25:40 -07005605static int
5606sd_parent_degenerate(struct sched_domain *sd, struct sched_domain *parent)
Suresh Siddha245af2c2005-06-25 14:57:25 -07005607{
5608 unsigned long cflags = sd->flags, pflags = parent->flags;
5609
5610 if (sd_degenerate(parent))
5611 return 1;
5612
5613 if (!cpus_equal(sd->span, parent->span))
5614 return 0;
5615
5616 /* Does parent contain flags not in child? */
5617 /* WAKE_BALANCE is a subset of WAKE_AFFINE */
5618 if (cflags & SD_WAKE_AFFINE)
5619 pflags &= ~SD_WAKE_BALANCE;
5620 /* Flags needing groups don't count if only 1 group in parent */
5621 if (parent->groups == parent->groups->next) {
5622 pflags &= ~(SD_LOAD_BALANCE |
5623 SD_BALANCE_NEWIDLE |
5624 SD_BALANCE_FORK |
Siddha, Suresh B89c47102006-10-03 01:14:09 -07005625 SD_BALANCE_EXEC |
5626 SD_SHARE_CPUPOWER |
5627 SD_SHARE_PKG_RESOURCES);
Suresh Siddha245af2c2005-06-25 14:57:25 -07005628 }
5629 if (~cflags & pflags)
5630 return 0;
5631
5632 return 1;
5633}
5634
Linus Torvalds1da177e2005-04-16 15:20:36 -07005635/*
5636 * Attach the domain 'sd' to 'cpu' as its base domain. Callers must
5637 * hold the hotplug lock.
5638 */
John Hawkes9c1cfda2005-09-06 15:18:14 -07005639static void cpu_attach_domain(struct sched_domain *sd, int cpu)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005640{
Ingo Molnar70b97a72006-07-03 00:25:42 -07005641 struct rq *rq = cpu_rq(cpu);
Suresh Siddha245af2c2005-06-25 14:57:25 -07005642 struct sched_domain *tmp;
5643
5644 /* Remove the sched domains which do not contribute to scheduling. */
5645 for (tmp = sd; tmp; tmp = tmp->parent) {
5646 struct sched_domain *parent = tmp->parent;
5647 if (!parent)
5648 break;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07005649 if (sd_parent_degenerate(tmp, parent)) {
Suresh Siddha245af2c2005-06-25 14:57:25 -07005650 tmp->parent = parent->parent;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07005651 if (parent->parent)
5652 parent->parent->child = tmp;
5653 }
Suresh Siddha245af2c2005-06-25 14:57:25 -07005654 }
5655
Siddha, Suresh B1a848872006-10-03 01:14:08 -07005656 if (sd && sd_degenerate(sd)) {
Suresh Siddha245af2c2005-06-25 14:57:25 -07005657 sd = sd->parent;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07005658 if (sd)
5659 sd->child = NULL;
5660 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07005661
5662 sched_domain_debug(sd, cpu);
5663
Nick Piggin674311d2005-06-25 14:57:27 -07005664 rcu_assign_pointer(rq->sd, sd);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005665}
5666
5667/* cpus with isolated domains */
Tim Chen67af63a2006-12-22 01:07:50 -08005668static cpumask_t cpu_isolated_map = CPU_MASK_NONE;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005669
5670/* Setup the mask of cpus configured for isolated domains */
5671static int __init isolated_cpu_setup(char *str)
5672{
5673 int ints[NR_CPUS], i;
5674
5675 str = get_options(str, ARRAY_SIZE(ints), ints);
5676 cpus_clear(cpu_isolated_map);
5677 for (i = 1; i <= ints[0]; i++)
5678 if (ints[i] < NR_CPUS)
5679 cpu_set(ints[i], cpu_isolated_map);
5680 return 1;
5681}
5682
5683__setup ("isolcpus=", isolated_cpu_setup);
5684
5685/*
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08005686 * init_sched_build_groups takes the cpumask we wish to span, and a pointer
5687 * to a function which identifies what group(along with sched group) a CPU
5688 * belongs to. The return value of group_fn must be a >= 0 and < NR_CPUS
5689 * (due to the fact that we keep track of groups covered with a cpumask_t).
Linus Torvalds1da177e2005-04-16 15:20:36 -07005690 *
5691 * init_sched_build_groups will build a circular linked list of the groups
5692 * covered by the given span, and will set each group's ->cpumask correctly,
5693 * and ->cpu_power to 0.
5694 */
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07005695static void
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08005696init_sched_build_groups(cpumask_t span, const cpumask_t *cpu_map,
5697 int (*group_fn)(int cpu, const cpumask_t *cpu_map,
5698 struct sched_group **sg))
Linus Torvalds1da177e2005-04-16 15:20:36 -07005699{
5700 struct sched_group *first = NULL, *last = NULL;
5701 cpumask_t covered = CPU_MASK_NONE;
5702 int i;
5703
5704 for_each_cpu_mask(i, span) {
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08005705 struct sched_group *sg;
5706 int group = group_fn(i, cpu_map, &sg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07005707 int j;
5708
5709 if (cpu_isset(i, covered))
5710 continue;
5711
5712 sg->cpumask = CPU_MASK_NONE;
Eric Dumazet5517d862007-05-08 00:32:57 -07005713 sg->__cpu_power = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07005714
5715 for_each_cpu_mask(j, span) {
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08005716 if (group_fn(j, cpu_map, NULL) != group)
Linus Torvalds1da177e2005-04-16 15:20:36 -07005717 continue;
5718
5719 cpu_set(j, covered);
5720 cpu_set(j, sg->cpumask);
5721 }
5722 if (!first)
5723 first = sg;
5724 if (last)
5725 last->next = sg;
5726 last = sg;
5727 }
5728 last->next = first;
5729}
5730
John Hawkes9c1cfda2005-09-06 15:18:14 -07005731#define SD_NODES_PER_DOMAIN 16
Linus Torvalds1da177e2005-04-16 15:20:36 -07005732
akpm@osdl.org198e2f12006-01-12 01:05:30 -08005733/*
5734 * Self-tuning task migration cost measurement between source and target CPUs.
5735 *
5736 * This is done by measuring the cost of manipulating buffers of varying
5737 * sizes. For a given buffer-size here are the steps that are taken:
5738 *
5739 * 1) the source CPU reads+dirties a shared buffer
5740 * 2) the target CPU reads+dirties the same shared buffer
5741 *
5742 * We measure how long they take, in the following 4 scenarios:
5743 *
5744 * - source: CPU1, target: CPU2 | cost1
5745 * - source: CPU2, target: CPU1 | cost2
5746 * - source: CPU1, target: CPU1 | cost3
5747 * - source: CPU2, target: CPU2 | cost4
5748 *
5749 * We then calculate the cost3+cost4-cost1-cost2 difference - this is
5750 * the cost of migration.
5751 *
5752 * We then start off from a small buffer-size and iterate up to larger
5753 * buffer sizes, in 5% steps - measuring each buffer-size separately, and
5754 * doing a maximum search for the cost. (The maximum cost for a migration
5755 * normally occurs when the working set size is around the effective cache
5756 * size.)
5757 */
5758#define SEARCH_SCOPE 2
5759#define MIN_CACHE_SIZE (64*1024U)
5760#define DEFAULT_CACHE_SIZE (5*1024*1024U)
Ingo Molnar70b4d632006-01-30 20:24:38 +01005761#define ITERATIONS 1
akpm@osdl.org198e2f12006-01-12 01:05:30 -08005762#define SIZE_THRESH 130
5763#define COST_THRESH 130
5764
5765/*
5766 * The migration cost is a function of 'domain distance'. Domain
5767 * distance is the number of steps a CPU has to iterate down its
5768 * domain tree to share a domain with the other CPU. The farther
5769 * two CPUs are from each other, the larger the distance gets.
5770 *
5771 * Note that we use the distance only to cache measurement results,
5772 * the distance value is not used numerically otherwise. When two
5773 * CPUs have the same distance it is assumed that the migration
5774 * cost is the same. (this is a simplification but quite practical)
5775 */
5776#define MAX_DOMAIN_DISTANCE 32
5777
5778static unsigned long long migration_cost[MAX_DOMAIN_DISTANCE] =
Ingo Molnar4bbf39c2006-02-17 13:52:44 -08005779 { [ 0 ... MAX_DOMAIN_DISTANCE-1 ] =
5780/*
5781 * Architectures may override the migration cost and thus avoid
5782 * boot-time calibration. Unit is nanoseconds. Mostly useful for
5783 * virtualized hardware:
5784 */
5785#ifdef CONFIG_DEFAULT_MIGRATION_COST
5786 CONFIG_DEFAULT_MIGRATION_COST
5787#else
5788 -1LL
5789#endif
5790};
akpm@osdl.org198e2f12006-01-12 01:05:30 -08005791
5792/*
5793 * Allow override of migration cost - in units of microseconds.
5794 * E.g. migration_cost=1000,2000,3000 will set up a level-1 cost
5795 * of 1 msec, level-2 cost of 2 msecs and level3 cost of 3 msecs:
5796 */
5797static int __init migration_cost_setup(char *str)
5798{
5799 int ints[MAX_DOMAIN_DISTANCE+1], i;
5800
5801 str = get_options(str, ARRAY_SIZE(ints), ints);
5802
5803 printk("#ints: %d\n", ints[0]);
5804 for (i = 1; i <= ints[0]; i++) {
5805 migration_cost[i-1] = (unsigned long long)ints[i]*1000;
5806 printk("migration_cost[%d]: %Ld\n", i-1, migration_cost[i-1]);
5807 }
5808 return 1;
5809}
5810
5811__setup ("migration_cost=", migration_cost_setup);
5812
5813/*
5814 * Global multiplier (divisor) for migration-cutoff values,
5815 * in percentiles. E.g. use a value of 150 to get 1.5 times
5816 * longer cache-hot cutoff times.
5817 *
5818 * (We scale it from 100 to 128 to long long handling easier.)
5819 */
5820
5821#define MIGRATION_FACTOR_SCALE 128
5822
5823static unsigned int migration_factor = MIGRATION_FACTOR_SCALE;
5824
5825static int __init setup_migration_factor(char *str)
5826{
5827 get_option(&str, &migration_factor);
5828 migration_factor = migration_factor * MIGRATION_FACTOR_SCALE / 100;
5829 return 1;
5830}
5831
5832__setup("migration_factor=", setup_migration_factor);
5833
5834/*
5835 * Estimated distance of two CPUs, measured via the number of domains
5836 * we have to pass for the two CPUs to be in the same span:
5837 */
5838static unsigned long domain_distance(int cpu1, int cpu2)
5839{
5840 unsigned long distance = 0;
5841 struct sched_domain *sd;
5842
5843 for_each_domain(cpu1, sd) {
5844 WARN_ON(!cpu_isset(cpu1, sd->span));
5845 if (cpu_isset(cpu2, sd->span))
5846 return distance;
5847 distance++;
5848 }
5849 if (distance >= MAX_DOMAIN_DISTANCE) {
5850 WARN_ON(1);
5851 distance = MAX_DOMAIN_DISTANCE-1;
5852 }
5853
5854 return distance;
5855}
5856
5857static unsigned int migration_debug;
5858
5859static int __init setup_migration_debug(char *str)
5860{
5861 get_option(&str, &migration_debug);
5862 return 1;
5863}
5864
5865__setup("migration_debug=", setup_migration_debug);
5866
5867/*
5868 * Maximum cache-size that the scheduler should try to measure.
5869 * Architectures with larger caches should tune this up during
5870 * bootup. Gets used in the domain-setup code (i.e. during SMP
5871 * bootup).
5872 */
5873unsigned int max_cache_size;
5874
5875static int __init setup_max_cache_size(char *str)
5876{
5877 get_option(&str, &max_cache_size);
5878 return 1;
5879}
5880
5881__setup("max_cache_size=", setup_max_cache_size);
5882
5883/*
5884 * Dirty a big buffer in a hard-to-predict (for the L2 cache) way. This
5885 * is the operation that is timed, so we try to generate unpredictable
5886 * cachemisses that still end up filling the L2 cache:
5887 */
5888static void touch_cache(void *__cache, unsigned long __size)
5889{
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08005890 unsigned long size = __size / sizeof(long);
5891 unsigned long chunk1 = size / 3;
5892 unsigned long chunk2 = 2 * size / 3;
akpm@osdl.org198e2f12006-01-12 01:05:30 -08005893 unsigned long *cache = __cache;
5894 int i;
5895
5896 for (i = 0; i < size/6; i += 8) {
5897 switch (i % 6) {
5898 case 0: cache[i]++;
5899 case 1: cache[size-1-i]++;
5900 case 2: cache[chunk1-i]++;
5901 case 3: cache[chunk1+i]++;
5902 case 4: cache[chunk2-i]++;
5903 case 5: cache[chunk2+i]++;
5904 }
5905 }
5906}
5907
5908/*
5909 * Measure the cache-cost of one task migration. Returns in units of nsec.
5910 */
Ingo Molnar48f24c42006-07-03 00:25:40 -07005911static unsigned long long
5912measure_one(void *cache, unsigned long size, int source, int target)
akpm@osdl.org198e2f12006-01-12 01:05:30 -08005913{
5914 cpumask_t mask, saved_mask;
5915 unsigned long long t0, t1, t2, t3, cost;
5916
5917 saved_mask = current->cpus_allowed;
5918
5919 /*
5920 * Flush source caches to RAM and invalidate them:
5921 */
5922 sched_cacheflush();
5923
5924 /*
5925 * Migrate to the source CPU:
5926 */
5927 mask = cpumask_of_cpu(source);
5928 set_cpus_allowed(current, mask);
5929 WARN_ON(smp_processor_id() != source);
5930
5931 /*
5932 * Dirty the working set:
5933 */
5934 t0 = sched_clock();
5935 touch_cache(cache, size);
5936 t1 = sched_clock();
5937
5938 /*
5939 * Migrate to the target CPU, dirty the L2 cache and access
5940 * the shared buffer. (which represents the working set
5941 * of a migrated task.)
5942 */
5943 mask = cpumask_of_cpu(target);
5944 set_cpus_allowed(current, mask);
5945 WARN_ON(smp_processor_id() != target);
5946
5947 t2 = sched_clock();
5948 touch_cache(cache, size);
5949 t3 = sched_clock();
5950
5951 cost = t1-t0 + t3-t2;
5952
5953 if (migration_debug >= 2)
5954 printk("[%d->%d]: %8Ld %8Ld %8Ld => %10Ld.\n",
5955 source, target, t1-t0, t1-t0, t3-t2, cost);
5956 /*
5957 * Flush target caches to RAM and invalidate them:
5958 */
5959 sched_cacheflush();
5960
5961 set_cpus_allowed(current, saved_mask);
5962
5963 return cost;
5964}
5965
5966/*
5967 * Measure a series of task migrations and return the average
5968 * result. Since this code runs early during bootup the system
5969 * is 'undisturbed' and the average latency makes sense.
5970 *
5971 * The algorithm in essence auto-detects the relevant cache-size,
5972 * so it will properly detect different cachesizes for different
5973 * cache-hierarchies, depending on how the CPUs are connected.
5974 *
5975 * Architectures can prime the upper limit of the search range via
5976 * max_cache_size, otherwise the search range defaults to 20MB...64K.
5977 */
5978static unsigned long long
5979measure_cost(int cpu1, int cpu2, void *cache, unsigned int size)
5980{
5981 unsigned long long cost1, cost2;
5982 int i;
5983
5984 /*
5985 * Measure the migration cost of 'size' bytes, over an
5986 * average of 10 runs:
5987 *
5988 * (We perturb the cache size by a small (0..4k)
5989 * value to compensate size/alignment related artifacts.
5990 * We also subtract the cost of the operation done on
5991 * the same CPU.)
5992 */
5993 cost1 = 0;
5994
5995 /*
5996 * dry run, to make sure we start off cache-cold on cpu1,
5997 * and to get any vmalloc pagefaults in advance:
5998 */
5999 measure_one(cache, size, cpu1, cpu2);
6000 for (i = 0; i < ITERATIONS; i++)
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006001 cost1 += measure_one(cache, size - i * 1024, cpu1, cpu2);
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006002
6003 measure_one(cache, size, cpu2, cpu1);
6004 for (i = 0; i < ITERATIONS; i++)
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006005 cost1 += measure_one(cache, size - i * 1024, cpu2, cpu1);
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006006
6007 /*
6008 * (We measure the non-migrating [cached] cost on both
6009 * cpu1 and cpu2, to handle CPUs with different speeds)
6010 */
6011 cost2 = 0;
6012
6013 measure_one(cache, size, cpu1, cpu1);
6014 for (i = 0; i < ITERATIONS; i++)
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006015 cost2 += measure_one(cache, size - i * 1024, cpu1, cpu1);
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006016
6017 measure_one(cache, size, cpu2, cpu2);
6018 for (i = 0; i < ITERATIONS; i++)
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006019 cost2 += measure_one(cache, size - i * 1024, cpu2, cpu2);
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006020
6021 /*
6022 * Get the per-iteration migration cost:
6023 */
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006024 do_div(cost1, 2 * ITERATIONS);
6025 do_div(cost2, 2 * ITERATIONS);
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006026
6027 return cost1 - cost2;
6028}
6029
6030static unsigned long long measure_migration_cost(int cpu1, int cpu2)
6031{
6032 unsigned long long max_cost = 0, fluct = 0, avg_fluct = 0;
6033 unsigned int max_size, size, size_found = 0;
6034 long long cost = 0, prev_cost;
6035 void *cache;
6036
6037 /*
6038 * Search from max_cache_size*5 down to 64K - the real relevant
6039 * cachesize has to lie somewhere inbetween.
6040 */
6041 if (max_cache_size) {
6042 max_size = max(max_cache_size * SEARCH_SCOPE, MIN_CACHE_SIZE);
6043 size = max(max_cache_size / SEARCH_SCOPE, MIN_CACHE_SIZE);
6044 } else {
6045 /*
6046 * Since we have no estimation about the relevant
6047 * search range
6048 */
6049 max_size = DEFAULT_CACHE_SIZE * SEARCH_SCOPE;
6050 size = MIN_CACHE_SIZE;
6051 }
6052
6053 if (!cpu_online(cpu1) || !cpu_online(cpu2)) {
6054 printk("cpu %d and %d not both online!\n", cpu1, cpu2);
6055 return 0;
6056 }
6057
6058 /*
6059 * Allocate the working set:
6060 */
6061 cache = vmalloc(max_size);
6062 if (!cache) {
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006063 printk("could not vmalloc %d bytes for cache!\n", 2 * max_size);
Andreas Mohr2ed6e342006-07-10 04:43:52 -07006064 return 1000000; /* return 1 msec on very small boxen */
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006065 }
6066
6067 while (size <= max_size) {
6068 prev_cost = cost;
6069 cost = measure_cost(cpu1, cpu2, cache, size);
6070
6071 /*
6072 * Update the max:
6073 */
6074 if (cost > 0) {
6075 if (max_cost < cost) {
6076 max_cost = cost;
6077 size_found = size;
6078 }
6079 }
6080 /*
6081 * Calculate average fluctuation, we use this to prevent
6082 * noise from triggering an early break out of the loop:
6083 */
6084 fluct = abs(cost - prev_cost);
6085 avg_fluct = (avg_fluct + fluct)/2;
6086
6087 if (migration_debug)
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006088 printk("-> [%d][%d][%7d] %3ld.%ld [%3ld.%ld] (%ld): "
6089 "(%8Ld %8Ld)\n",
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006090 cpu1, cpu2, size,
6091 (long)cost / 1000000,
6092 ((long)cost / 100000) % 10,
6093 (long)max_cost / 1000000,
6094 ((long)max_cost / 100000) % 10,
6095 domain_distance(cpu1, cpu2),
6096 cost, avg_fluct);
6097
6098 /*
6099 * If we iterated at least 20% past the previous maximum,
6100 * and the cost has dropped by more than 20% already,
6101 * (taking fluctuations into account) then we assume to
6102 * have found the maximum and break out of the loop early:
6103 */
6104 if (size_found && (size*100 > size_found*SIZE_THRESH))
6105 if (cost+avg_fluct <= 0 ||
6106 max_cost*100 > (cost+avg_fluct)*COST_THRESH) {
6107
6108 if (migration_debug)
6109 printk("-> found max.\n");
6110 break;
6111 }
6112 /*
Ingo Molnar70b4d632006-01-30 20:24:38 +01006113 * Increase the cachesize in 10% steps:
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006114 */
Ingo Molnar70b4d632006-01-30 20:24:38 +01006115 size = size * 10 / 9;
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006116 }
6117
6118 if (migration_debug)
6119 printk("[%d][%d] working set size found: %d, cost: %Ld\n",
6120 cpu1, cpu2, size_found, max_cost);
6121
6122 vfree(cache);
6123
6124 /*
6125 * A task is considered 'cache cold' if at least 2 times
6126 * the worst-case cost of migration has passed.
6127 *
6128 * (this limit is only listened to if the load-balancing
6129 * situation is 'nice' - if there is a large imbalance we
6130 * ignore it for the sake of CPU utilization and
6131 * processing fairness.)
6132 */
6133 return 2 * max_cost * migration_factor / MIGRATION_FACTOR_SCALE;
6134}
6135
6136static void calibrate_migration_costs(const cpumask_t *cpu_map)
6137{
6138 int cpu1 = -1, cpu2 = -1, cpu, orig_cpu = raw_smp_processor_id();
6139 unsigned long j0, j1, distance, max_distance = 0;
6140 struct sched_domain *sd;
6141
6142 j0 = jiffies;
6143
6144 /*
6145 * First pass - calculate the cacheflush times:
6146 */
6147 for_each_cpu_mask(cpu1, *cpu_map) {
6148 for_each_cpu_mask(cpu2, *cpu_map) {
6149 if (cpu1 == cpu2)
6150 continue;
6151 distance = domain_distance(cpu1, cpu2);
6152 max_distance = max(max_distance, distance);
6153 /*
6154 * No result cached yet?
6155 */
6156 if (migration_cost[distance] == -1LL)
6157 migration_cost[distance] =
6158 measure_migration_cost(cpu1, cpu2);
6159 }
6160 }
6161 /*
6162 * Second pass - update the sched domain hierarchy with
6163 * the new cache-hot-time estimations:
6164 */
6165 for_each_cpu_mask(cpu, *cpu_map) {
6166 distance = 0;
6167 for_each_domain(cpu, sd) {
6168 sd->cache_hot_time = migration_cost[distance];
6169 distance++;
6170 }
6171 }
6172 /*
6173 * Print the matrix:
6174 */
6175 if (migration_debug)
6176 printk("migration: max_cache_size: %d, cpu: %d MHz:\n",
6177 max_cache_size,
6178#ifdef CONFIG_X86
6179 cpu_khz/1000
6180#else
6181 -1
6182#endif
6183 );
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006184 if (system_state == SYSTEM_BOOTING && num_online_cpus() > 1) {
6185 printk("migration_cost=");
6186 for (distance = 0; distance <= max_distance; distance++) {
6187 if (distance)
6188 printk(",");
6189 printk("%ld", (long)migration_cost[distance] / 1000);
Chuck Ebbertbd576c92006-02-04 23:27:42 -08006190 }
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006191 printk("\n");
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006192 }
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006193 j1 = jiffies;
6194 if (migration_debug)
Miguel Ojeda Sandonis33859f72006-12-10 02:20:38 -08006195 printk("migration: %ld seconds\n", (j1-j0) / HZ);
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006196
6197 /*
6198 * Move back to the original CPU. NUMA-Q gets confused
6199 * if we migrate to another quad during bootup.
6200 */
6201 if (raw_smp_processor_id() != orig_cpu) {
6202 cpumask_t mask = cpumask_of_cpu(orig_cpu),
6203 saved_mask = current->cpus_allowed;
6204
6205 set_cpus_allowed(current, mask);
6206 set_cpus_allowed(current, saved_mask);
6207 }
6208}
6209
John Hawkes9c1cfda2005-09-06 15:18:14 -07006210#ifdef CONFIG_NUMA
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006211
John Hawkes9c1cfda2005-09-06 15:18:14 -07006212/**
6213 * find_next_best_node - find the next node to include in a sched_domain
6214 * @node: node whose sched_domain we're building
6215 * @used_nodes: nodes already in the sched_domain
6216 *
6217 * Find the next node to include in a given scheduling domain. Simply
6218 * finds the closest node not already in the @used_nodes map.
6219 *
6220 * Should use nodemask_t.
6221 */
6222static int find_next_best_node(int node, unsigned long *used_nodes)
6223{
6224 int i, n, val, min_val, best_node = 0;
6225
6226 min_val = INT_MAX;
6227
6228 for (i = 0; i < MAX_NUMNODES; i++) {
6229 /* Start at @node */
6230 n = (node + i) % MAX_NUMNODES;
6231
6232 if (!nr_cpus_node(n))
6233 continue;
6234
6235 /* Skip already used nodes */
6236 if (test_bit(n, used_nodes))
6237 continue;
6238
6239 /* Simple min distance search */
6240 val = node_distance(node, n);
6241
6242 if (val < min_val) {
6243 min_val = val;
6244 best_node = n;
6245 }
6246 }
6247
6248 set_bit(best_node, used_nodes);
6249 return best_node;
6250}
6251
6252/**
6253 * sched_domain_node_span - get a cpumask for a node's sched_domain
6254 * @node: node whose cpumask we're constructing
6255 * @size: number of nodes to include in this span
6256 *
6257 * Given a node, construct a good cpumask for its sched_domain to span. It
6258 * should be one that prevents unnecessary balancing, but also spreads tasks
6259 * out optimally.
6260 */
6261static cpumask_t sched_domain_node_span(int node)
6262{
John Hawkes9c1cfda2005-09-06 15:18:14 -07006263 DECLARE_BITMAP(used_nodes, MAX_NUMNODES);
Ingo Molnar48f24c42006-07-03 00:25:40 -07006264 cpumask_t span, nodemask;
6265 int i;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006266
6267 cpus_clear(span);
6268 bitmap_zero(used_nodes, MAX_NUMNODES);
6269
6270 nodemask = node_to_cpumask(node);
6271 cpus_or(span, span, nodemask);
6272 set_bit(node, used_nodes);
6273
6274 for (i = 1; i < SD_NODES_PER_DOMAIN; i++) {
6275 int next_node = find_next_best_node(node, used_nodes);
Ingo Molnar48f24c42006-07-03 00:25:40 -07006276
John Hawkes9c1cfda2005-09-06 15:18:14 -07006277 nodemask = node_to_cpumask(next_node);
6278 cpus_or(span, span, nodemask);
6279 }
6280
6281 return span;
6282}
6283#endif
6284
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006285int sched_smt_power_savings = 0, sched_mc_power_savings = 0;
Ingo Molnar48f24c42006-07-03 00:25:40 -07006286
John Hawkes9c1cfda2005-09-06 15:18:14 -07006287/*
Ingo Molnar48f24c42006-07-03 00:25:40 -07006288 * SMT sched-domains:
John Hawkes9c1cfda2005-09-06 15:18:14 -07006289 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07006290#ifdef CONFIG_SCHED_SMT
6291static DEFINE_PER_CPU(struct sched_domain, cpu_domains);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006292static DEFINE_PER_CPU(struct sched_group, sched_group_cpus);
Ingo Molnar48f24c42006-07-03 00:25:40 -07006293
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006294static int cpu_to_cpu_group(int cpu, const cpumask_t *cpu_map,
6295 struct sched_group **sg)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006296{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006297 if (sg)
6298 *sg = &per_cpu(sched_group_cpus, cpu);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006299 return cpu;
6300}
6301#endif
6302
Ingo Molnar48f24c42006-07-03 00:25:40 -07006303/*
6304 * multi-core sched-domains:
6305 */
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006306#ifdef CONFIG_SCHED_MC
6307static DEFINE_PER_CPU(struct sched_domain, core_domains);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006308static DEFINE_PER_CPU(struct sched_group, sched_group_core);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006309#endif
6310
6311#if defined(CONFIG_SCHED_MC) && defined(CONFIG_SCHED_SMT)
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006312static int cpu_to_core_group(int cpu, const cpumask_t *cpu_map,
6313 struct sched_group **sg)
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006314{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006315 int group;
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006316 cpumask_t mask = cpu_sibling_map[cpu];
6317 cpus_and(mask, mask, *cpu_map);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006318 group = first_cpu(mask);
6319 if (sg)
6320 *sg = &per_cpu(sched_group_core, group);
6321 return group;
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006322}
6323#elif defined(CONFIG_SCHED_MC)
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006324static int cpu_to_core_group(int cpu, const cpumask_t *cpu_map,
6325 struct sched_group **sg)
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006326{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006327 if (sg)
6328 *sg = &per_cpu(sched_group_core, cpu);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006329 return cpu;
6330}
6331#endif
6332
Linus Torvalds1da177e2005-04-16 15:20:36 -07006333static DEFINE_PER_CPU(struct sched_domain, phys_domains);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006334static DEFINE_PER_CPU(struct sched_group, sched_group_phys);
Ingo Molnar48f24c42006-07-03 00:25:40 -07006335
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006336static int cpu_to_phys_group(int cpu, const cpumask_t *cpu_map,
6337 struct sched_group **sg)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006338{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006339 int group;
Ingo Molnar48f24c42006-07-03 00:25:40 -07006340#ifdef CONFIG_SCHED_MC
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006341 cpumask_t mask = cpu_coregroup_map(cpu);
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006342 cpus_and(mask, mask, *cpu_map);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006343 group = first_cpu(mask);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006344#elif defined(CONFIG_SCHED_SMT)
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006345 cpumask_t mask = cpu_sibling_map[cpu];
6346 cpus_and(mask, mask, *cpu_map);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006347 group = first_cpu(mask);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006348#else
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006349 group = cpu;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006350#endif
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006351 if (sg)
6352 *sg = &per_cpu(sched_group_phys, group);
6353 return group;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006354}
6355
6356#ifdef CONFIG_NUMA
John Hawkes9c1cfda2005-09-06 15:18:14 -07006357/*
6358 * The init_sched_build_groups can't handle what we want to do with node
6359 * groups, so roll our own. Now each node has its own list of groups which
6360 * gets dynamically allocated.
6361 */
Linus Torvalds1da177e2005-04-16 15:20:36 -07006362static DEFINE_PER_CPU(struct sched_domain, node_domains);
John Hawkesd1b55132005-09-06 15:18:14 -07006363static struct sched_group **sched_group_nodes_bycpu[NR_CPUS];
John Hawkes9c1cfda2005-09-06 15:18:14 -07006364
6365static DEFINE_PER_CPU(struct sched_domain, allnodes_domains);
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006366static DEFINE_PER_CPU(struct sched_group, sched_group_allnodes);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006367
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006368static int cpu_to_allnodes_group(int cpu, const cpumask_t *cpu_map,
6369 struct sched_group **sg)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006370{
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006371 cpumask_t nodemask = node_to_cpumask(cpu_to_node(cpu));
6372 int group;
6373
6374 cpus_and(nodemask, nodemask, *cpu_map);
6375 group = first_cpu(nodemask);
6376
6377 if (sg)
6378 *sg = &per_cpu(sched_group_allnodes, group);
6379 return group;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006380}
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006381
Siddha, Suresh B08069032006-03-27 01:15:23 -08006382static void init_numa_sched_groups_power(struct sched_group *group_head)
6383{
6384 struct sched_group *sg = group_head;
6385 int j;
6386
6387 if (!sg)
6388 return;
6389next_sg:
6390 for_each_cpu_mask(j, sg->cpumask) {
6391 struct sched_domain *sd;
6392
6393 sd = &per_cpu(phys_domains, j);
6394 if (j != first_cpu(sd->groups->cpumask)) {
6395 /*
6396 * Only add "power" once for each
6397 * physical package.
6398 */
6399 continue;
6400 }
6401
Eric Dumazet5517d862007-05-08 00:32:57 -07006402 sg_inc_cpu_power(sg, sd->groups->__cpu_power);
Siddha, Suresh B08069032006-03-27 01:15:23 -08006403 }
6404 sg = sg->next;
6405 if (sg != group_head)
6406 goto next_sg;
6407}
Linus Torvalds1da177e2005-04-16 15:20:36 -07006408#endif
6409
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006410#ifdef CONFIG_NUMA
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006411/* Free memory allocated for various sched_group structures */
6412static void free_sched_groups(const cpumask_t *cpu_map)
6413{
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006414 int cpu, i;
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006415
6416 for_each_cpu_mask(cpu, *cpu_map) {
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006417 struct sched_group **sched_group_nodes
6418 = sched_group_nodes_bycpu[cpu];
6419
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006420 if (!sched_group_nodes)
6421 continue;
6422
6423 for (i = 0; i < MAX_NUMNODES; i++) {
6424 cpumask_t nodemask = node_to_cpumask(i);
6425 struct sched_group *oldsg, *sg = sched_group_nodes[i];
6426
6427 cpus_and(nodemask, nodemask, *cpu_map);
6428 if (cpus_empty(nodemask))
6429 continue;
6430
6431 if (sg == NULL)
6432 continue;
6433 sg = sg->next;
6434next_sg:
6435 oldsg = sg;
6436 sg = sg->next;
6437 kfree(oldsg);
6438 if (oldsg != sched_group_nodes[i])
6439 goto next_sg;
6440 }
6441 kfree(sched_group_nodes);
6442 sched_group_nodes_bycpu[cpu] = NULL;
6443 }
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006444}
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006445#else
6446static void free_sched_groups(const cpumask_t *cpu_map)
6447{
6448}
6449#endif
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006450
Linus Torvalds1da177e2005-04-16 15:20:36 -07006451/*
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006452 * Initialize sched groups cpu_power.
6453 *
6454 * cpu_power indicates the capacity of sched group, which is used while
6455 * distributing the load between different sched groups in a sched domain.
6456 * Typically cpu_power for all the groups in a sched domain will be same unless
6457 * there are asymmetries in the topology. If there are asymmetries, group
6458 * having more cpu_power will pickup more load compared to the group having
6459 * less cpu_power.
6460 *
6461 * cpu_power will be a multiple of SCHED_LOAD_SCALE. This multiple represents
6462 * the maximum number of tasks a group can handle in the presence of other idle
6463 * or lightly loaded groups in the same sched domain.
6464 */
6465static void init_sched_groups_power(int cpu, struct sched_domain *sd)
6466{
6467 struct sched_domain *child;
6468 struct sched_group *group;
6469
6470 WARN_ON(!sd || !sd->groups);
6471
6472 if (cpu != first_cpu(sd->groups->cpumask))
6473 return;
6474
6475 child = sd->child;
6476
Eric Dumazet5517d862007-05-08 00:32:57 -07006477 sd->groups->__cpu_power = 0;
6478
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006479 /*
6480 * For perf policy, if the groups in child domain share resources
6481 * (for example cores sharing some portions of the cache hierarchy
6482 * or SMT), then set this domain groups cpu_power such that each group
6483 * can handle only one task, when there are other idle groups in the
6484 * same sched domain.
6485 */
6486 if (!child || (!(sd->flags & SD_POWERSAVINGS_BALANCE) &&
6487 (child->flags &
6488 (SD_SHARE_CPUPOWER | SD_SHARE_PKG_RESOURCES)))) {
Eric Dumazet5517d862007-05-08 00:32:57 -07006489 sg_inc_cpu_power(sd->groups, SCHED_LOAD_SCALE);
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006490 return;
6491 }
6492
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006493 /*
6494 * add cpu_power of each child group to this groups cpu_power
6495 */
6496 group = child->groups;
6497 do {
Eric Dumazet5517d862007-05-08 00:32:57 -07006498 sg_inc_cpu_power(sd->groups, group->__cpu_power);
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006499 group = group->next;
6500 } while (group != child->groups);
6501}
6502
6503/*
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006504 * Build sched domains for a given set of cpus and attach the sched domains
6505 * to the individual cpus
Linus Torvalds1da177e2005-04-16 15:20:36 -07006506 */
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006507static int build_sched_domains(const cpumask_t *cpu_map)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006508{
6509 int i;
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006510 struct sched_domain *sd;
John Hawkesd1b55132005-09-06 15:18:14 -07006511#ifdef CONFIG_NUMA
6512 struct sched_group **sched_group_nodes = NULL;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006513 int sd_allnodes = 0;
John Hawkesd1b55132005-09-06 15:18:14 -07006514
6515 /*
6516 * Allocate the per-node list of sched groups
6517 */
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006518 sched_group_nodes = kzalloc(sizeof(struct sched_group*)*MAX_NUMNODES,
Srivatsa Vaddagirid3a5aa92006-06-27 02:54:39 -07006519 GFP_KERNEL);
John Hawkesd1b55132005-09-06 15:18:14 -07006520 if (!sched_group_nodes) {
6521 printk(KERN_WARNING "Can not alloc sched group node list\n");
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006522 return -ENOMEM;
John Hawkesd1b55132005-09-06 15:18:14 -07006523 }
6524 sched_group_nodes_bycpu[first_cpu(*cpu_map)] = sched_group_nodes;
6525#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07006526
6527 /*
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006528 * Set up domains for cpus specified by the cpu_map.
Linus Torvalds1da177e2005-04-16 15:20:36 -07006529 */
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006530 for_each_cpu_mask(i, *cpu_map) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07006531 struct sched_domain *sd = NULL, *p;
6532 cpumask_t nodemask = node_to_cpumask(cpu_to_node(i));
6533
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006534 cpus_and(nodemask, nodemask, *cpu_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006535
6536#ifdef CONFIG_NUMA
John Hawkesd1b55132005-09-06 15:18:14 -07006537 if (cpus_weight(*cpu_map)
John Hawkes9c1cfda2005-09-06 15:18:14 -07006538 > SD_NODES_PER_DOMAIN*cpus_weight(nodemask)) {
6539 sd = &per_cpu(allnodes_domains, i);
6540 *sd = SD_ALLNODES_INIT;
6541 sd->span = *cpu_map;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006542 cpu_to_allnodes_group(i, cpu_map, &sd->groups);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006543 p = sd;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006544 sd_allnodes = 1;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006545 } else
6546 p = NULL;
6547
Linus Torvalds1da177e2005-04-16 15:20:36 -07006548 sd = &per_cpu(node_domains, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006549 *sd = SD_NODE_INIT;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006550 sd->span = sched_domain_node_span(cpu_to_node(i));
6551 sd->parent = p;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006552 if (p)
6553 p->child = sd;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006554 cpus_and(sd->span, sd->span, *cpu_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006555#endif
6556
6557 p = sd;
6558 sd = &per_cpu(phys_domains, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006559 *sd = SD_CPU_INIT;
6560 sd->span = nodemask;
6561 sd->parent = p;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006562 if (p)
6563 p->child = sd;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006564 cpu_to_phys_group(i, cpu_map, &sd->groups);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006565
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006566#ifdef CONFIG_SCHED_MC
6567 p = sd;
6568 sd = &per_cpu(core_domains, i);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006569 *sd = SD_MC_INIT;
6570 sd->span = cpu_coregroup_map(i);
6571 cpus_and(sd->span, sd->span, *cpu_map);
6572 sd->parent = p;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006573 p->child = sd;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006574 cpu_to_core_group(i, cpu_map, &sd->groups);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006575#endif
6576
Linus Torvalds1da177e2005-04-16 15:20:36 -07006577#ifdef CONFIG_SCHED_SMT
6578 p = sd;
6579 sd = &per_cpu(cpu_domains, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006580 *sd = SD_SIBLING_INIT;
6581 sd->span = cpu_sibling_map[i];
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006582 cpus_and(sd->span, sd->span, *cpu_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006583 sd->parent = p;
Siddha, Suresh B1a848872006-10-03 01:14:08 -07006584 p->child = sd;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006585 cpu_to_cpu_group(i, cpu_map, &sd->groups);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006586#endif
6587 }
6588
6589#ifdef CONFIG_SCHED_SMT
6590 /* Set up CPU (sibling) groups */
John Hawkes9c1cfda2005-09-06 15:18:14 -07006591 for_each_cpu_mask(i, *cpu_map) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07006592 cpumask_t this_sibling_map = cpu_sibling_map[i];
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006593 cpus_and(this_sibling_map, this_sibling_map, *cpu_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006594 if (i != first_cpu(this_sibling_map))
6595 continue;
6596
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006597 init_sched_build_groups(this_sibling_map, cpu_map, &cpu_to_cpu_group);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006598 }
6599#endif
6600
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006601#ifdef CONFIG_SCHED_MC
6602 /* Set up multi-core groups */
6603 for_each_cpu_mask(i, *cpu_map) {
6604 cpumask_t this_core_map = cpu_coregroup_map(i);
6605 cpus_and(this_core_map, this_core_map, *cpu_map);
6606 if (i != first_cpu(this_core_map))
6607 continue;
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006608 init_sched_build_groups(this_core_map, cpu_map, &cpu_to_core_group);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006609 }
6610#endif
6611
6612
Linus Torvalds1da177e2005-04-16 15:20:36 -07006613 /* Set up physical groups */
6614 for (i = 0; i < MAX_NUMNODES; i++) {
6615 cpumask_t nodemask = node_to_cpumask(i);
6616
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006617 cpus_and(nodemask, nodemask, *cpu_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006618 if (cpus_empty(nodemask))
6619 continue;
6620
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006621 init_sched_build_groups(nodemask, cpu_map, &cpu_to_phys_group);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006622 }
6623
6624#ifdef CONFIG_NUMA
6625 /* Set up node groups */
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006626 if (sd_allnodes)
6627 init_sched_build_groups(*cpu_map, cpu_map, &cpu_to_allnodes_group);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006628
6629 for (i = 0; i < MAX_NUMNODES; i++) {
6630 /* Set up node groups */
6631 struct sched_group *sg, *prev;
6632 cpumask_t nodemask = node_to_cpumask(i);
6633 cpumask_t domainspan;
6634 cpumask_t covered = CPU_MASK_NONE;
6635 int j;
6636
6637 cpus_and(nodemask, nodemask, *cpu_map);
John Hawkesd1b55132005-09-06 15:18:14 -07006638 if (cpus_empty(nodemask)) {
6639 sched_group_nodes[i] = NULL;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006640 continue;
John Hawkesd1b55132005-09-06 15:18:14 -07006641 }
John Hawkes9c1cfda2005-09-06 15:18:14 -07006642
6643 domainspan = sched_domain_node_span(i);
6644 cpus_and(domainspan, domainspan, *cpu_map);
6645
Srivatsa Vaddagiri15f0b672006-06-27 02:54:40 -07006646 sg = kmalloc_node(sizeof(struct sched_group), GFP_KERNEL, i);
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006647 if (!sg) {
6648 printk(KERN_WARNING "Can not alloc domain group for "
6649 "node %d\n", i);
6650 goto error;
6651 }
John Hawkes9c1cfda2005-09-06 15:18:14 -07006652 sched_group_nodes[i] = sg;
6653 for_each_cpu_mask(j, nodemask) {
6654 struct sched_domain *sd;
6655 sd = &per_cpu(node_domains, j);
6656 sd->groups = sg;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006657 }
Eric Dumazet5517d862007-05-08 00:32:57 -07006658 sg->__cpu_power = 0;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006659 sg->cpumask = nodemask;
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006660 sg->next = sg;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006661 cpus_or(covered, covered, nodemask);
6662 prev = sg;
6663
6664 for (j = 0; j < MAX_NUMNODES; j++) {
6665 cpumask_t tmp, notcovered;
6666 int n = (i + j) % MAX_NUMNODES;
6667
6668 cpus_complement(notcovered, covered);
6669 cpus_and(tmp, notcovered, *cpu_map);
6670 cpus_and(tmp, tmp, domainspan);
6671 if (cpus_empty(tmp))
6672 break;
6673
6674 nodemask = node_to_cpumask(n);
6675 cpus_and(tmp, tmp, nodemask);
6676 if (cpus_empty(tmp))
6677 continue;
6678
Srivatsa Vaddagiri15f0b672006-06-27 02:54:40 -07006679 sg = kmalloc_node(sizeof(struct sched_group),
6680 GFP_KERNEL, i);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006681 if (!sg) {
6682 printk(KERN_WARNING
6683 "Can not alloc domain group for node %d\n", j);
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006684 goto error;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006685 }
Eric Dumazet5517d862007-05-08 00:32:57 -07006686 sg->__cpu_power = 0;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006687 sg->cpumask = tmp;
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006688 sg->next = prev->next;
John Hawkes9c1cfda2005-09-06 15:18:14 -07006689 cpus_or(covered, covered, tmp);
6690 prev->next = sg;
6691 prev = sg;
6692 }
John Hawkes9c1cfda2005-09-06 15:18:14 -07006693 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07006694#endif
6695
6696 /* Calculate CPU power for physical packages and nodes */
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006697#ifdef CONFIG_SCHED_SMT
6698 for_each_cpu_mask(i, *cpu_map) {
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006699 sd = &per_cpu(cpu_domains, i);
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006700 init_sched_groups_power(i, sd);
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006701 }
6702#endif
6703#ifdef CONFIG_SCHED_MC
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006704 for_each_cpu_mask(i, *cpu_map) {
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006705 sd = &per_cpu(core_domains, i);
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006706 init_sched_groups_power(i, sd);
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006707 }
6708#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07006709
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006710 for_each_cpu_mask(i, *cpu_map) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07006711 sd = &per_cpu(phys_domains, i);
Siddha, Suresh B89c47102006-10-03 01:14:09 -07006712 init_sched_groups_power(i, sd);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006713 }
6714
John Hawkes9c1cfda2005-09-06 15:18:14 -07006715#ifdef CONFIG_NUMA
Siddha, Suresh B08069032006-03-27 01:15:23 -08006716 for (i = 0; i < MAX_NUMNODES; i++)
6717 init_numa_sched_groups_power(sched_group_nodes[i]);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006718
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006719 if (sd_allnodes) {
6720 struct sched_group *sg;
Siddha, Suresh Bf712c0c2006-07-30 03:02:59 -07006721
Siddha, Suresh B6711cab2006-12-10 02:20:07 -08006722 cpu_to_allnodes_group(first_cpu(*cpu_map), cpu_map, &sg);
Siddha, Suresh Bf712c0c2006-07-30 03:02:59 -07006723 init_numa_sched_groups_power(sg);
6724 }
John Hawkes9c1cfda2005-09-06 15:18:14 -07006725#endif
6726
Linus Torvalds1da177e2005-04-16 15:20:36 -07006727 /* Attach the domains */
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006728 for_each_cpu_mask(i, *cpu_map) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07006729 struct sched_domain *sd;
6730#ifdef CONFIG_SCHED_SMT
6731 sd = &per_cpu(cpu_domains, i);
Siddha, Suresh B1e9f28f2006-03-27 01:15:22 -08006732#elif defined(CONFIG_SCHED_MC)
6733 sd = &per_cpu(core_domains, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006734#else
6735 sd = &per_cpu(phys_domains, i);
6736#endif
6737 cpu_attach_domain(sd, i);
6738 }
akpm@osdl.org198e2f12006-01-12 01:05:30 -08006739 /*
6740 * Tune cache-hot values:
6741 */
6742 calibrate_migration_costs(cpu_map);
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006743
6744 return 0;
6745
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006746#ifdef CONFIG_NUMA
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006747error:
6748 free_sched_groups(cpu_map);
6749 return -ENOMEM;
Siddha, Suresh Ba6160582006-10-03 01:14:06 -07006750#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07006751}
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006752/*
6753 * Set up scheduler domains and groups. Callers must hold the hotplug lock.
6754 */
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006755static int arch_init_sched_domains(const cpumask_t *cpu_map)
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006756{
6757 cpumask_t cpu_default_map;
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006758 int err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006759
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006760 /*
6761 * Setup mask for cpus without special case scheduling requirements.
6762 * For now this just excludes isolated cpus, but could be used to
6763 * exclude other special cases in the future.
6764 */
6765 cpus_andnot(cpu_default_map, *cpu_map, cpu_isolated_map);
6766
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006767 err = build_sched_domains(&cpu_default_map);
6768
6769 return err;
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006770}
6771
6772static void arch_destroy_sched_domains(const cpumask_t *cpu_map)
Linus Torvalds1da177e2005-04-16 15:20:36 -07006773{
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006774 free_sched_groups(cpu_map);
John Hawkes9c1cfda2005-09-06 15:18:14 -07006775}
Linus Torvalds1da177e2005-04-16 15:20:36 -07006776
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006777/*
6778 * Detach sched domains from a group of cpus specified in cpu_map
6779 * These cpus will now be attached to the NULL domain
6780 */
Arjan van de Ven858119e2006-01-14 13:20:43 -08006781static void detach_destroy_domains(const cpumask_t *cpu_map)
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006782{
6783 int i;
6784
6785 for_each_cpu_mask(i, *cpu_map)
6786 cpu_attach_domain(NULL, i);
6787 synchronize_sched();
6788 arch_destroy_sched_domains(cpu_map);
6789}
6790
6791/*
6792 * Partition sched domains as specified by the cpumasks below.
6793 * This attaches all cpus from the cpumasks to the NULL domain,
6794 * waits for a RCU quiescent period, recalculates sched
6795 * domain information and then attaches them back to the
6796 * correct sched domains
6797 * Call with hotplug lock held
6798 */
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006799int partition_sched_domains(cpumask_t *partition1, cpumask_t *partition2)
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006800{
6801 cpumask_t change_map;
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006802 int err = 0;
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006803
6804 cpus_and(*partition1, *partition1, cpu_online_map);
6805 cpus_and(*partition2, *partition2, cpu_online_map);
6806 cpus_or(change_map, *partition1, *partition2);
6807
6808 /* Detach sched domains from all of the affected cpus */
6809 detach_destroy_domains(&change_map);
6810 if (!cpus_empty(*partition1))
Srivatsa Vaddagiri51888ca2006-06-27 02:54:38 -07006811 err = build_sched_domains(partition1);
6812 if (!err && !cpus_empty(*partition2))
6813 err = build_sched_domains(partition2);
6814
6815 return err;
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006816}
6817
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006818#if defined(CONFIG_SCHED_MC) || defined(CONFIG_SCHED_SMT)
6819int arch_reinit_sched_domains(void)
6820{
6821 int err;
6822
6823 lock_cpu_hotplug();
6824 detach_destroy_domains(&cpu_online_map);
6825 err = arch_init_sched_domains(&cpu_online_map);
6826 unlock_cpu_hotplug();
6827
6828 return err;
6829}
6830
6831static ssize_t sched_power_savings_store(const char *buf, size_t count, int smt)
6832{
6833 int ret;
6834
6835 if (buf[0] != '0' && buf[0] != '1')
6836 return -EINVAL;
6837
6838 if (smt)
6839 sched_smt_power_savings = (buf[0] == '1');
6840 else
6841 sched_mc_power_savings = (buf[0] == '1');
6842
6843 ret = arch_reinit_sched_domains();
6844
6845 return ret ? ret : count;
6846}
6847
6848int sched_create_sysfs_power_savings_entries(struct sysdev_class *cls)
6849{
6850 int err = 0;
Ingo Molnar48f24c42006-07-03 00:25:40 -07006851
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006852#ifdef CONFIG_SCHED_SMT
6853 if (smt_capable())
6854 err = sysfs_create_file(&cls->kset.kobj,
6855 &attr_sched_smt_power_savings.attr);
6856#endif
6857#ifdef CONFIG_SCHED_MC
6858 if (!err && mc_capable())
6859 err = sysfs_create_file(&cls->kset.kobj,
6860 &attr_sched_mc_power_savings.attr);
6861#endif
6862 return err;
6863}
6864#endif
6865
6866#ifdef CONFIG_SCHED_MC
6867static ssize_t sched_mc_power_savings_show(struct sys_device *dev, char *page)
6868{
6869 return sprintf(page, "%u\n", sched_mc_power_savings);
6870}
Ingo Molnar48f24c42006-07-03 00:25:40 -07006871static ssize_t sched_mc_power_savings_store(struct sys_device *dev,
6872 const char *buf, size_t count)
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006873{
6874 return sched_power_savings_store(buf, count, 0);
6875}
6876SYSDEV_ATTR(sched_mc_power_savings, 0644, sched_mc_power_savings_show,
6877 sched_mc_power_savings_store);
6878#endif
6879
6880#ifdef CONFIG_SCHED_SMT
6881static ssize_t sched_smt_power_savings_show(struct sys_device *dev, char *page)
6882{
6883 return sprintf(page, "%u\n", sched_smt_power_savings);
6884}
Ingo Molnar48f24c42006-07-03 00:25:40 -07006885static ssize_t sched_smt_power_savings_store(struct sys_device *dev,
6886 const char *buf, size_t count)
Siddha, Suresh B5c45bf22006-06-27 02:54:42 -07006887{
6888 return sched_power_savings_store(buf, count, 1);
6889}
6890SYSDEV_ATTR(sched_smt_power_savings, 0644, sched_smt_power_savings_show,
6891 sched_smt_power_savings_store);
6892#endif
6893
Linus Torvalds1da177e2005-04-16 15:20:36 -07006894/*
6895 * Force a reinitialization of the sched domains hierarchy. The domains
6896 * and groups cannot be updated in place without racing with the balancing
Nick Piggin41c7ce92005-06-25 14:57:24 -07006897 * code, so we temporarily attach all running cpus to the NULL domain
Linus Torvalds1da177e2005-04-16 15:20:36 -07006898 * which will prevent rebalancing while the sched domains are recalculated.
6899 */
6900static int update_sched_domains(struct notifier_block *nfb,
6901 unsigned long action, void *hcpu)
6902{
Linus Torvalds1da177e2005-04-16 15:20:36 -07006903 switch (action) {
6904 case CPU_UP_PREPARE:
6905 case CPU_DOWN_PREPARE:
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006906 detach_destroy_domains(&cpu_online_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006907 return NOTIFY_OK;
6908
6909 case CPU_UP_CANCELED:
6910 case CPU_DOWN_FAILED:
6911 case CPU_ONLINE:
6912 case CPU_DEAD:
6913 /*
6914 * Fall through and re-initialise the domains.
6915 */
6916 break;
6917 default:
6918 return NOTIFY_DONE;
6919 }
6920
6921 /* The hotplug lock is already held by cpu_up/cpu_down */
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006922 arch_init_sched_domains(&cpu_online_map);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006923
6924 return NOTIFY_OK;
6925}
Linus Torvalds1da177e2005-04-16 15:20:36 -07006926
6927void __init sched_init_smp(void)
6928{
Nick Piggin5c1e1762006-10-03 01:14:04 -07006929 cpumask_t non_isolated_cpus;
6930
Linus Torvalds1da177e2005-04-16 15:20:36 -07006931 lock_cpu_hotplug();
Dinakar Guniguntala1a20ff22005-06-25 14:57:33 -07006932 arch_init_sched_domains(&cpu_online_map);
Nathan Lynche5e56732007-01-10 23:15:28 -08006933 cpus_andnot(non_isolated_cpus, cpu_possible_map, cpu_isolated_map);
Nick Piggin5c1e1762006-10-03 01:14:04 -07006934 if (cpus_empty(non_isolated_cpus))
6935 cpu_set(smp_processor_id(), non_isolated_cpus);
Linus Torvalds1da177e2005-04-16 15:20:36 -07006936 unlock_cpu_hotplug();
6937 /* XXX: Theoretical race here - CPU may be hotplugged now */
6938 hotcpu_notifier(update_sched_domains, 0);
Nick Piggin5c1e1762006-10-03 01:14:04 -07006939
6940 /* Move init over to a non-isolated CPU */
6941 if (set_cpus_allowed(current, non_isolated_cpus) < 0)
6942 BUG();
Linus Torvalds1da177e2005-04-16 15:20:36 -07006943}
6944#else
6945void __init sched_init_smp(void)
6946{
6947}
6948#endif /* CONFIG_SMP */
6949
6950int in_sched_functions(unsigned long addr)
6951{
6952 /* Linker adds these: start and end of __sched functions */
6953 extern char __sched_text_start[], __sched_text_end[];
Ingo Molnar48f24c42006-07-03 00:25:40 -07006954
Linus Torvalds1da177e2005-04-16 15:20:36 -07006955 return in_lock_functions(addr) ||
6956 (addr >= (unsigned long)__sched_text_start
6957 && addr < (unsigned long)__sched_text_end);
6958}
6959
6960void __init sched_init(void)
6961{
Linus Torvalds1da177e2005-04-16 15:20:36 -07006962 int i, j, k;
Christoph Lameter476f3532007-05-06 14:48:58 -07006963 int highest_cpu = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006964
KAMEZAWA Hiroyuki0a945022006-03-28 01:56:37 -08006965 for_each_possible_cpu(i) {
Ingo Molnar70b97a72006-07-03 00:25:42 -07006966 struct prio_array *array;
6967 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006968
6969 rq = cpu_rq(i);
6970 spin_lock_init(&rq->lock);
Ingo Molnarfcb99372006-07-03 00:25:10 -07006971 lockdep_set_class(&rq->lock, &rq->rq_lock_key);
Nick Piggin78979862005-06-25 14:57:13 -07006972 rq->nr_running = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006973 rq->active = rq->arrays;
6974 rq->expired = rq->arrays + 1;
6975 rq->best_expired_prio = MAX_PRIO;
6976
6977#ifdef CONFIG_SMP
Nick Piggin41c7ce92005-06-25 14:57:24 -07006978 rq->sd = NULL;
Nick Piggin78979862005-06-25 14:57:13 -07006979 for (j = 1; j < 3; j++)
6980 rq->cpu_load[j] = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006981 rq->active_balance = 0;
6982 rq->push_cpu = 0;
Christoph Lameter0a2966b2006-09-25 23:30:51 -07006983 rq->cpu = i;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006984 rq->migration_thread = NULL;
6985 INIT_LIST_HEAD(&rq->migration_queue);
6986#endif
6987 atomic_set(&rq->nr_iowait, 0);
6988
6989 for (j = 0; j < 2; j++) {
6990 array = rq->arrays + j;
6991 for (k = 0; k < MAX_PRIO; k++) {
6992 INIT_LIST_HEAD(array->queue + k);
6993 __clear_bit(k, array->bitmap);
6994 }
6995 // delimiter for bitsearch
6996 __set_bit(MAX_PRIO, array->bitmap);
6997 }
Christoph Lameter476f3532007-05-06 14:48:58 -07006998 highest_cpu = i;
Linus Torvalds1da177e2005-04-16 15:20:36 -07006999 }
7000
Peter Williams2dd73a42006-06-27 02:54:34 -07007001 set_load_weight(&init_task);
Heiko Carstensb50f60c2006-07-30 03:03:52 -07007002
Christoph Lameterc9819f42006-12-10 02:20:25 -08007003#ifdef CONFIG_SMP
Christoph Lameter476f3532007-05-06 14:48:58 -07007004 nr_cpu_ids = highest_cpu + 1;
Christoph Lameterc9819f42006-12-10 02:20:25 -08007005 open_softirq(SCHED_SOFTIRQ, run_rebalance_domains, NULL);
7006#endif
7007
Heiko Carstensb50f60c2006-07-30 03:03:52 -07007008#ifdef CONFIG_RT_MUTEXES
7009 plist_head_init(&init_task.pi_waiters, &init_task.pi_lock);
7010#endif
7011
Linus Torvalds1da177e2005-04-16 15:20:36 -07007012 /*
7013 * The boot idle thread does lazy MMU switching as well:
7014 */
7015 atomic_inc(&init_mm.mm_count);
7016 enter_lazy_tlb(&init_mm, current);
7017
7018 /*
7019 * Make us the idle thread. Technically, schedule() should not be
7020 * called from this thread, however somewhere below it might be,
7021 * but because we are the idle thread, we just pick up running again
7022 * when this runqueue becomes "idle".
7023 */
7024 init_idle(current, smp_processor_id());
7025}
7026
7027#ifdef CONFIG_DEBUG_SPINLOCK_SLEEP
7028void __might_sleep(char *file, int line)
7029{
Ingo Molnar48f24c42006-07-03 00:25:40 -07007030#ifdef in_atomic
Linus Torvalds1da177e2005-04-16 15:20:36 -07007031 static unsigned long prev_jiffy; /* ratelimiting */
7032
7033 if ((in_atomic() || irqs_disabled()) &&
7034 system_state == SYSTEM_RUNNING && !oops_in_progress) {
7035 if (time_before(jiffies, prev_jiffy + HZ) && prev_jiffy)
7036 return;
7037 prev_jiffy = jiffies;
Ingo Molnar91368d72006-03-23 03:00:54 -08007038 printk(KERN_ERR "BUG: sleeping function called from invalid"
Linus Torvalds1da177e2005-04-16 15:20:36 -07007039 " context at %s:%d\n", file, line);
7040 printk("in_atomic():%d, irqs_disabled():%d\n",
7041 in_atomic(), irqs_disabled());
Peter Zijlstraa4c410f2006-12-06 20:37:21 -08007042 debug_show_held_locks(current);
Ingo Molnar3117df02006-12-13 00:34:43 -08007043 if (irqs_disabled())
7044 print_irqtrace_events(current);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007045 dump_stack();
7046 }
7047#endif
7048}
7049EXPORT_SYMBOL(__might_sleep);
7050#endif
7051
7052#ifdef CONFIG_MAGIC_SYSRQ
7053void normalize_rt_tasks(void)
7054{
Ingo Molnar70b97a72006-07-03 00:25:42 -07007055 struct prio_array *array;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007056 struct task_struct *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007057 unsigned long flags;
Ingo Molnar70b97a72006-07-03 00:25:42 -07007058 struct rq *rq;
Linus Torvalds1da177e2005-04-16 15:20:36 -07007059
7060 read_lock_irq(&tasklist_lock);
Chen, Kenneth Wc96d1452006-06-27 02:54:28 -07007061 for_each_process(p) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07007062 if (!rt_task(p))
7063 continue;
7064
Ingo Molnarb29739f2006-06-27 02:54:51 -07007065 spin_lock_irqsave(&p->pi_lock, flags);
7066 rq = __task_rq_lock(p);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007067
7068 array = p->array;
7069 if (array)
7070 deactivate_task(p, task_rq(p));
7071 __setscheduler(p, SCHED_NORMAL, 0);
7072 if (array) {
7073 __activate_task(p, task_rq(p));
7074 resched_task(rq->curr);
7075 }
7076
Ingo Molnarb29739f2006-06-27 02:54:51 -07007077 __task_rq_unlock(rq);
7078 spin_unlock_irqrestore(&p->pi_lock, flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -07007079 }
7080 read_unlock_irq(&tasklist_lock);
7081}
7082
7083#endif /* CONFIG_MAGIC_SYSRQ */
Linus Torvalds1df5c102005-09-12 07:59:21 -07007084
7085#ifdef CONFIG_IA64
7086/*
7087 * These functions are only useful for the IA64 MCA handling.
7088 *
7089 * They can only be called when the whole system has been
7090 * stopped - every CPU needs to be quiescent, and no scheduling
7091 * activity can take place. Using them for anything else would
7092 * be a serious bug, and as a result, they aren't even visible
7093 * under any other configuration.
7094 */
7095
7096/**
7097 * curr_task - return the current task for a given cpu.
7098 * @cpu: the processor in question.
7099 *
7100 * ONLY VALID WHEN THE WHOLE SYSTEM IS STOPPED!
7101 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07007102struct task_struct *curr_task(int cpu)
Linus Torvalds1df5c102005-09-12 07:59:21 -07007103{
7104 return cpu_curr(cpu);
7105}
7106
7107/**
7108 * set_curr_task - set the current task for a given cpu.
7109 * @cpu: the processor in question.
7110 * @p: the task pointer to set.
7111 *
7112 * Description: This function must only be used when non-maskable interrupts
7113 * are serviced on a separate stack. It allows the architecture to switch the
7114 * notion of the current task on a cpu in a non-blocking manner. This function
7115 * must be called with all CPU's synchronized, and interrupts disabled, the
7116 * and caller must save the original value of the current task (see
7117 * curr_task() above) and restore that value before reenabling interrupts and
7118 * re-starting the system.
7119 *
7120 * ONLY VALID WHEN THE WHOLE SYSTEM IS STOPPED!
7121 */
Ingo Molnar36c8b582006-07-03 00:25:41 -07007122void set_curr_task(int cpu, struct task_struct *p)
Linus Torvalds1df5c102005-09-12 07:59:21 -07007123{
7124 cpu_curr(cpu) = p;
7125}
7126
7127#endif