blob: 64a8f0aa117b1595d28ea0c90adcf7de03985bad [file] [log] [blame]
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001/*
2 * Real-Time Scheduling Class (mapped to the SCHED_FIFO and SCHED_RR
3 * policies)
4 */
5
Steven Rostedt4fd29172008-01-25 21:08:06 +01006#ifdef CONFIG_SMP
Ingo Molnar84de4272008-01-25 21:08:15 +01007
Gregory Haskins637f5082008-01-25 21:08:18 +01008static inline int rt_overloaded(struct rq *rq)
Steven Rostedt4fd29172008-01-25 21:08:06 +01009{
Gregory Haskins637f5082008-01-25 21:08:18 +010010 return atomic_read(&rq->rd->rto_count);
Steven Rostedt4fd29172008-01-25 21:08:06 +010011}
Ingo Molnar84de4272008-01-25 21:08:15 +010012
Steven Rostedt4fd29172008-01-25 21:08:06 +010013static inline void rt_set_overload(struct rq *rq)
14{
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -040015 if (!rq->online)
16 return;
17
Rusty Russellc6c49272008-11-25 02:35:05 +103018 cpumask_set_cpu(rq->cpu, rq->rd->rto_mask);
Steven Rostedt4fd29172008-01-25 21:08:06 +010019 /*
20 * Make sure the mask is visible before we set
21 * the overload count. That is checked to determine
22 * if we should look at the mask. It would be a shame
23 * if we looked at the mask, but the mask was not
24 * updated yet.
25 */
26 wmb();
Gregory Haskins637f5082008-01-25 21:08:18 +010027 atomic_inc(&rq->rd->rto_count);
Steven Rostedt4fd29172008-01-25 21:08:06 +010028}
Ingo Molnar84de4272008-01-25 21:08:15 +010029
Steven Rostedt4fd29172008-01-25 21:08:06 +010030static inline void rt_clear_overload(struct rq *rq)
31{
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -040032 if (!rq->online)
33 return;
34
Steven Rostedt4fd29172008-01-25 21:08:06 +010035 /* the order here really doesn't matter */
Gregory Haskins637f5082008-01-25 21:08:18 +010036 atomic_dec(&rq->rd->rto_count);
Rusty Russellc6c49272008-11-25 02:35:05 +103037 cpumask_clear_cpu(rq->cpu, rq->rd->rto_mask);
Steven Rostedt4fd29172008-01-25 21:08:06 +010038}
Gregory Haskins73fe6aa2008-01-25 21:08:07 +010039
40static void update_rt_migration(struct rq *rq)
41{
Gregory Haskins637f5082008-01-25 21:08:18 +010042 if (rq->rt.rt_nr_migratory && (rq->rt.rt_nr_running > 1)) {
Gregory Haskinscdc8eb92008-01-25 21:08:23 +010043 if (!rq->rt.overloaded) {
44 rt_set_overload(rq);
45 rq->rt.overloaded = 1;
46 }
47 } else if (rq->rt.overloaded) {
Gregory Haskins73fe6aa2008-01-25 21:08:07 +010048 rt_clear_overload(rq);
Gregory Haskins637f5082008-01-25 21:08:18 +010049 rq->rt.overloaded = 0;
50 }
Gregory Haskins73fe6aa2008-01-25 21:08:07 +010051}
Gregory Haskins917b6272008-12-29 09:39:53 -050052
53static void enqueue_pushable_task(struct rq *rq, struct task_struct *p)
54{
55 plist_del(&p->pushable_tasks, &rq->rt.pushable_tasks);
56 plist_node_init(&p->pushable_tasks, p->prio);
57 plist_add(&p->pushable_tasks, &rq->rt.pushable_tasks);
58}
59
60static void dequeue_pushable_task(struct rq *rq, struct task_struct *p)
61{
62 plist_del(&p->pushable_tasks, &rq->rt.pushable_tasks);
63}
64
65#else
66
67#define enqueue_pushable_task(rq, p) do { } while (0)
68#define dequeue_pushable_task(rq, p) do { } while (0)
69
Steven Rostedt4fd29172008-01-25 21:08:06 +010070#endif /* CONFIG_SMP */
71
Peter Zijlstra6f505b12008-01-25 21:08:30 +010072static inline struct task_struct *rt_task_of(struct sched_rt_entity *rt_se)
Peter Zijlstrafa85ae22008-01-25 21:08:29 +010073{
Peter Zijlstra6f505b12008-01-25 21:08:30 +010074 return container_of(rt_se, struct task_struct, rt);
75}
76
77static inline int on_rt_rq(struct sched_rt_entity *rt_se)
78{
79 return !list_empty(&rt_se->run_list);
80}
81
Peter Zijlstra052f1dc2008-02-13 15:45:40 +010082#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +010083
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +010084static inline u64 sched_rt_runtime(struct rt_rq *rt_rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +010085{
86 if (!rt_rq->tg)
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +010087 return RUNTIME_INF;
Peter Zijlstra6f505b12008-01-25 21:08:30 +010088
Peter Zijlstraac086bc2008-04-19 19:44:58 +020089 return rt_rq->rt_runtime;
90}
91
92static inline u64 sched_rt_period(struct rt_rq *rt_rq)
93{
94 return ktime_to_ns(rt_rq->tg->rt_bandwidth.rt_period);
Peter Zijlstra6f505b12008-01-25 21:08:30 +010095}
96
97#define for_each_leaf_rt_rq(rt_rq, rq) \
98 list_for_each_entry(rt_rq, &rq->leaf_rt_rq_list, leaf_rt_rq_list)
99
100static inline struct rq *rq_of_rt_rq(struct rt_rq *rt_rq)
101{
102 return rt_rq->rq;
103}
104
105static inline struct rt_rq *rt_rq_of_se(struct sched_rt_entity *rt_se)
106{
107 return rt_se->rt_rq;
108}
109
110#define for_each_sched_rt_entity(rt_se) \
111 for (; rt_se; rt_se = rt_se->parent)
112
113static inline struct rt_rq *group_rt_rq(struct sched_rt_entity *rt_se)
114{
115 return rt_se->my_q;
116}
117
118static void enqueue_rt_entity(struct sched_rt_entity *rt_se);
119static void dequeue_rt_entity(struct sched_rt_entity *rt_se);
120
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100121static void sched_rt_rq_enqueue(struct rt_rq *rt_rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100122{
Dario Faggiolif6121f42008-10-03 17:40:46 +0200123 struct task_struct *curr = rq_of_rt_rq(rt_rq)->curr;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100124 struct sched_rt_entity *rt_se = rt_rq->rt_se;
125
Dario Faggiolif6121f42008-10-03 17:40:46 +0200126 if (rt_rq->rt_nr_running) {
127 if (rt_se && !on_rt_rq(rt_se))
128 enqueue_rt_entity(rt_se);
Gregory Haskinse864c492008-12-29 09:39:49 -0500129 if (rt_rq->highest_prio.curr < curr->prio)
Peter Zijlstra10203872008-01-25 21:08:32 +0100130 resched_task(curr);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100131 }
132}
133
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100134static void sched_rt_rq_dequeue(struct rt_rq *rt_rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100135{
136 struct sched_rt_entity *rt_se = rt_rq->rt_se;
137
138 if (rt_se && on_rt_rq(rt_se))
139 dequeue_rt_entity(rt_se);
140}
141
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100142static inline int rt_rq_throttled(struct rt_rq *rt_rq)
143{
144 return rt_rq->rt_throttled && !rt_rq->rt_nr_boosted;
145}
146
147static int rt_se_boosted(struct sched_rt_entity *rt_se)
148{
149 struct rt_rq *rt_rq = group_rt_rq(rt_se);
150 struct task_struct *p;
151
152 if (rt_rq)
153 return !!rt_rq->rt_nr_boosted;
154
155 p = rt_task_of(rt_se);
156 return p->prio != p->normal_prio;
157}
158
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200159#ifdef CONFIG_SMP
Rusty Russellc6c49272008-11-25 02:35:05 +1030160static inline const struct cpumask *sched_rt_period_mask(void)
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200161{
162 return cpu_rq(smp_processor_id())->rd->span;
163}
164#else
Rusty Russellc6c49272008-11-25 02:35:05 +1030165static inline const struct cpumask *sched_rt_period_mask(void)
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200166{
Rusty Russellc6c49272008-11-25 02:35:05 +1030167 return cpu_online_mask;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200168}
169#endif
170
171static inline
172struct rt_rq *sched_rt_period_rt_rq(struct rt_bandwidth *rt_b, int cpu)
173{
174 return container_of(rt_b, struct task_group, rt_bandwidth)->rt_rq[cpu];
175}
176
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200177static inline struct rt_bandwidth *sched_rt_bandwidth(struct rt_rq *rt_rq)
178{
179 return &rt_rq->tg->rt_bandwidth;
180}
181
Dhaval Giani55e12e52008-06-24 23:39:43 +0530182#else /* !CONFIG_RT_GROUP_SCHED */
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100183
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100184static inline u64 sched_rt_runtime(struct rt_rq *rt_rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100185{
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200186 return rt_rq->rt_runtime;
187}
188
189static inline u64 sched_rt_period(struct rt_rq *rt_rq)
190{
191 return ktime_to_ns(def_rt_bandwidth.rt_period);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100192}
193
194#define for_each_leaf_rt_rq(rt_rq, rq) \
195 for (rt_rq = &rq->rt; rt_rq; rt_rq = NULL)
196
197static inline struct rq *rq_of_rt_rq(struct rt_rq *rt_rq)
198{
199 return container_of(rt_rq, struct rq, rt);
200}
201
202static inline struct rt_rq *rt_rq_of_se(struct sched_rt_entity *rt_se)
203{
204 struct task_struct *p = rt_task_of(rt_se);
205 struct rq *rq = task_rq(p);
206
207 return &rq->rt;
208}
209
210#define for_each_sched_rt_entity(rt_se) \
211 for (; rt_se; rt_se = NULL)
212
213static inline struct rt_rq *group_rt_rq(struct sched_rt_entity *rt_se)
214{
215 return NULL;
216}
217
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100218static inline void sched_rt_rq_enqueue(struct rt_rq *rt_rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100219{
John Blackwoodf3ade832008-08-26 15:09:43 -0400220 if (rt_rq->rt_nr_running)
221 resched_task(rq_of_rt_rq(rt_rq)->curr);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100222}
223
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100224static inline void sched_rt_rq_dequeue(struct rt_rq *rt_rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100225{
226}
227
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100228static inline int rt_rq_throttled(struct rt_rq *rt_rq)
229{
230 return rt_rq->rt_throttled;
231}
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200232
Rusty Russellc6c49272008-11-25 02:35:05 +1030233static inline const struct cpumask *sched_rt_period_mask(void)
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200234{
Rusty Russellc6c49272008-11-25 02:35:05 +1030235 return cpu_online_mask;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200236}
237
238static inline
239struct rt_rq *sched_rt_period_rt_rq(struct rt_bandwidth *rt_b, int cpu)
240{
241 return &cpu_rq(cpu)->rt;
242}
243
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200244static inline struct rt_bandwidth *sched_rt_bandwidth(struct rt_rq *rt_rq)
245{
246 return &def_rt_bandwidth;
247}
248
Dhaval Giani55e12e52008-06-24 23:39:43 +0530249#endif /* CONFIG_RT_GROUP_SCHED */
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100250
Peter Zijlstrab79f3832008-06-19 14:22:25 +0200251#ifdef CONFIG_SMP
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200252/*
253 * We ran out of runtime, see if we can borrow some from our neighbours.
254 */
Peter Zijlstrab79f3832008-06-19 14:22:25 +0200255static int do_balance_runtime(struct rt_rq *rt_rq)
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200256{
257 struct rt_bandwidth *rt_b = sched_rt_bandwidth(rt_rq);
258 struct root_domain *rd = cpu_rq(smp_processor_id())->rd;
259 int i, weight, more = 0;
260 u64 rt_period;
261
Rusty Russellc6c49272008-11-25 02:35:05 +1030262 weight = cpumask_weight(rd->span);
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200263
264 spin_lock(&rt_b->rt_runtime_lock);
265 rt_period = ktime_to_ns(rt_b->rt_period);
Rusty Russellc6c49272008-11-25 02:35:05 +1030266 for_each_cpu(i, rd->span) {
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200267 struct rt_rq *iter = sched_rt_period_rt_rq(rt_b, i);
268 s64 diff;
269
270 if (iter == rt_rq)
271 continue;
272
273 spin_lock(&iter->rt_runtime_lock);
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200274 /*
275 * Either all rqs have inf runtime and there's nothing to steal
276 * or __disable_runtime() below sets a specific rq to inf to
277 * indicate its been disabled and disalow stealing.
278 */
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200279 if (iter->rt_runtime == RUNTIME_INF)
280 goto next;
281
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200282 /*
283 * From runqueues with spare time, take 1/n part of their
284 * spare time, but no more than our period.
285 */
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200286 diff = iter->rt_runtime - iter->rt_time;
287 if (diff > 0) {
Peter Zijlstra58838cf2008-07-24 12:43:13 +0200288 diff = div_u64((u64)diff, weight);
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200289 if (rt_rq->rt_runtime + diff > rt_period)
290 diff = rt_period - rt_rq->rt_runtime;
291 iter->rt_runtime -= diff;
292 rt_rq->rt_runtime += diff;
293 more = 1;
294 if (rt_rq->rt_runtime == rt_period) {
295 spin_unlock(&iter->rt_runtime_lock);
296 break;
297 }
298 }
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200299next:
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200300 spin_unlock(&iter->rt_runtime_lock);
301 }
302 spin_unlock(&rt_b->rt_runtime_lock);
303
304 return more;
305}
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200306
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200307/*
308 * Ensure this RQ takes back all the runtime it lend to its neighbours.
309 */
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200310static void __disable_runtime(struct rq *rq)
311{
312 struct root_domain *rd = rq->rd;
313 struct rt_rq *rt_rq;
314
315 if (unlikely(!scheduler_running))
316 return;
317
318 for_each_leaf_rt_rq(rt_rq, rq) {
319 struct rt_bandwidth *rt_b = sched_rt_bandwidth(rt_rq);
320 s64 want;
321 int i;
322
323 spin_lock(&rt_b->rt_runtime_lock);
324 spin_lock(&rt_rq->rt_runtime_lock);
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200325 /*
326 * Either we're all inf and nobody needs to borrow, or we're
327 * already disabled and thus have nothing to do, or we have
328 * exactly the right amount of runtime to take out.
329 */
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200330 if (rt_rq->rt_runtime == RUNTIME_INF ||
331 rt_rq->rt_runtime == rt_b->rt_runtime)
332 goto balanced;
333 spin_unlock(&rt_rq->rt_runtime_lock);
334
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200335 /*
336 * Calculate the difference between what we started out with
337 * and what we current have, that's the amount of runtime
338 * we lend and now have to reclaim.
339 */
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200340 want = rt_b->rt_runtime - rt_rq->rt_runtime;
341
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200342 /*
343 * Greedy reclaim, take back as much as we can.
344 */
Rusty Russellc6c49272008-11-25 02:35:05 +1030345 for_each_cpu(i, rd->span) {
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200346 struct rt_rq *iter = sched_rt_period_rt_rq(rt_b, i);
347 s64 diff;
348
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200349 /*
350 * Can't reclaim from ourselves or disabled runqueues.
351 */
Peter Zijlstraf1679d02008-08-14 15:49:00 +0200352 if (iter == rt_rq || iter->rt_runtime == RUNTIME_INF)
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200353 continue;
354
355 spin_lock(&iter->rt_runtime_lock);
356 if (want > 0) {
357 diff = min_t(s64, iter->rt_runtime, want);
358 iter->rt_runtime -= diff;
359 want -= diff;
360 } else {
361 iter->rt_runtime -= want;
362 want -= want;
363 }
364 spin_unlock(&iter->rt_runtime_lock);
365
366 if (!want)
367 break;
368 }
369
370 spin_lock(&rt_rq->rt_runtime_lock);
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200371 /*
372 * We cannot be left wanting - that would mean some runtime
373 * leaked out of the system.
374 */
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200375 BUG_ON(want);
376balanced:
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200377 /*
378 * Disable all the borrow logic by pretending we have inf
379 * runtime - in which case borrowing doesn't make sense.
380 */
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200381 rt_rq->rt_runtime = RUNTIME_INF;
382 spin_unlock(&rt_rq->rt_runtime_lock);
383 spin_unlock(&rt_b->rt_runtime_lock);
384 }
385}
386
387static void disable_runtime(struct rq *rq)
388{
389 unsigned long flags;
390
391 spin_lock_irqsave(&rq->lock, flags);
392 __disable_runtime(rq);
393 spin_unlock_irqrestore(&rq->lock, flags);
394}
395
396static void __enable_runtime(struct rq *rq)
397{
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200398 struct rt_rq *rt_rq;
399
400 if (unlikely(!scheduler_running))
401 return;
402
Peter Zijlstra78333cd2008-09-23 15:33:43 +0200403 /*
404 * Reset each runqueue's bandwidth settings
405 */
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200406 for_each_leaf_rt_rq(rt_rq, rq) {
407 struct rt_bandwidth *rt_b = sched_rt_bandwidth(rt_rq);
408
409 spin_lock(&rt_b->rt_runtime_lock);
410 spin_lock(&rt_rq->rt_runtime_lock);
411 rt_rq->rt_runtime = rt_b->rt_runtime;
412 rt_rq->rt_time = 0;
Zhang, Yanminbaf25732008-09-09 11:26:33 +0800413 rt_rq->rt_throttled = 0;
Peter Zijlstra7def2be2008-06-05 14:49:58 +0200414 spin_unlock(&rt_rq->rt_runtime_lock);
415 spin_unlock(&rt_b->rt_runtime_lock);
416 }
417}
418
419static void enable_runtime(struct rq *rq)
420{
421 unsigned long flags;
422
423 spin_lock_irqsave(&rq->lock, flags);
424 __enable_runtime(rq);
425 spin_unlock_irqrestore(&rq->lock, flags);
426}
427
Peter Zijlstraeff65492008-06-19 14:22:26 +0200428static int balance_runtime(struct rt_rq *rt_rq)
429{
430 int more = 0;
431
432 if (rt_rq->rt_time > rt_rq->rt_runtime) {
433 spin_unlock(&rt_rq->rt_runtime_lock);
434 more = do_balance_runtime(rt_rq);
435 spin_lock(&rt_rq->rt_runtime_lock);
436 }
437
438 return more;
439}
Dhaval Giani55e12e52008-06-24 23:39:43 +0530440#else /* !CONFIG_SMP */
Peter Zijlstraeff65492008-06-19 14:22:26 +0200441static inline int balance_runtime(struct rt_rq *rt_rq)
442{
443 return 0;
444}
Dhaval Giani55e12e52008-06-24 23:39:43 +0530445#endif /* CONFIG_SMP */
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100446
447static int do_sched_rt_period_timer(struct rt_bandwidth *rt_b, int overrun)
448{
449 int i, idle = 1;
Rusty Russellc6c49272008-11-25 02:35:05 +1030450 const struct cpumask *span;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200451
Peter Zijlstra0b148fa2008-08-19 12:33:04 +0200452 if (!rt_bandwidth_enabled() || rt_b->rt_runtime == RUNTIME_INF)
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200453 return 1;
454
455 span = sched_rt_period_mask();
Rusty Russellc6c49272008-11-25 02:35:05 +1030456 for_each_cpu(i, span) {
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200457 int enqueue = 0;
458 struct rt_rq *rt_rq = sched_rt_period_rt_rq(rt_b, i);
459 struct rq *rq = rq_of_rt_rq(rt_rq);
460
461 spin_lock(&rq->lock);
462 if (rt_rq->rt_time) {
463 u64 runtime;
464
465 spin_lock(&rt_rq->rt_runtime_lock);
Peter Zijlstraeff65492008-06-19 14:22:26 +0200466 if (rt_rq->rt_throttled)
467 balance_runtime(rt_rq);
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200468 runtime = rt_rq->rt_runtime;
469 rt_rq->rt_time -= min(rt_rq->rt_time, overrun*runtime);
470 if (rt_rq->rt_throttled && rt_rq->rt_time < runtime) {
471 rt_rq->rt_throttled = 0;
472 enqueue = 1;
473 }
474 if (rt_rq->rt_time || rt_rq->rt_nr_running)
475 idle = 0;
476 spin_unlock(&rt_rq->rt_runtime_lock);
Peter Zijlstra8a8cde12008-06-19 14:22:28 +0200477 } else if (rt_rq->rt_nr_running)
478 idle = 0;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200479
480 if (enqueue)
481 sched_rt_rq_enqueue(rt_rq);
482 spin_unlock(&rq->lock);
483 }
484
485 return idle;
486}
487
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100488static inline int rt_se_prio(struct sched_rt_entity *rt_se)
489{
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100490#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100491 struct rt_rq *rt_rq = group_rt_rq(rt_se);
492
493 if (rt_rq)
Gregory Haskinse864c492008-12-29 09:39:49 -0500494 return rt_rq->highest_prio.curr;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100495#endif
496
497 return rt_task_of(rt_se)->prio;
498}
499
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100500static int sched_rt_runtime_exceeded(struct rt_rq *rt_rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100501{
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100502 u64 runtime = sched_rt_runtime(rt_rq);
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100503
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100504 if (rt_rq->rt_throttled)
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100505 return rt_rq_throttled(rt_rq);
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100506
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200507 if (sched_rt_runtime(rt_rq) >= sched_rt_period(rt_rq))
508 return 0;
509
Peter Zijlstrab79f3832008-06-19 14:22:25 +0200510 balance_runtime(rt_rq);
511 runtime = sched_rt_runtime(rt_rq);
512 if (runtime == RUNTIME_INF)
513 return 0;
Peter Zijlstraac086bc2008-04-19 19:44:58 +0200514
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100515 if (rt_rq->rt_time > runtime) {
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100516 rt_rq->rt_throttled = 1;
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100517 if (rt_rq_throttled(rt_rq)) {
Peter Zijlstra9f0c1e52008-02-13 15:45:39 +0100518 sched_rt_rq_dequeue(rt_rq);
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100519 return 1;
520 }
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100521 }
522
523 return 0;
524}
525
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200526/*
527 * Update the current task's runtime statistics. Skip current tasks that
528 * are not in our scheduling class.
529 */
Alexey Dobriyana9957442007-10-15 17:00:13 +0200530static void update_curr_rt(struct rq *rq)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200531{
532 struct task_struct *curr = rq->curr;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100533 struct sched_rt_entity *rt_se = &curr->rt;
534 struct rt_rq *rt_rq = rt_rq_of_se(rt_se);
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200535 u64 delta_exec;
536
537 if (!task_has_rt_policy(curr))
538 return;
539
Ingo Molnard2819182007-08-09 11:16:47 +0200540 delta_exec = rq->clock - curr->se.exec_start;
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200541 if (unlikely((s64)delta_exec < 0))
542 delta_exec = 0;
Ingo Molnar6cfb0d52007-08-02 17:41:40 +0200543
544 schedstat_set(curr->se.exec_max, max(curr->se.exec_max, delta_exec));
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200545
546 curr->se.sum_exec_runtime += delta_exec;
Frank Mayharf06febc2008-09-12 09:54:39 -0700547 account_group_exec_runtime(curr, delta_exec);
548
Ingo Molnard2819182007-08-09 11:16:47 +0200549 curr->se.exec_start = rq->clock;
Srivatsa Vaddagirid842de82007-12-02 20:04:49 +0100550 cpuacct_charge(curr, delta_exec);
Peter Zijlstrafa85ae22008-01-25 21:08:29 +0100551
Peter Zijlstra0b148fa2008-08-19 12:33:04 +0200552 if (!rt_bandwidth_enabled())
553 return;
554
Dhaval Giani354d60c2008-04-19 19:44:59 +0200555 for_each_sched_rt_entity(rt_se) {
556 rt_rq = rt_rq_of_se(rt_se);
557
Peter Zijlstracc2991c2008-08-19 12:33:03 +0200558 if (sched_rt_runtime(rt_rq) != RUNTIME_INF) {
Dimitri Sivaniche113a742008-10-31 08:03:41 -0500559 spin_lock(&rt_rq->rt_runtime_lock);
Peter Zijlstracc2991c2008-08-19 12:33:03 +0200560 rt_rq->rt_time += delta_exec;
561 if (sched_rt_runtime_exceeded(rt_rq))
562 resched_task(curr);
Dimitri Sivaniche113a742008-10-31 08:03:41 -0500563 spin_unlock(&rt_rq->rt_runtime_lock);
Peter Zijlstracc2991c2008-08-19 12:33:03 +0200564 }
Dhaval Giani354d60c2008-04-19 19:44:59 +0200565 }
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200566}
567
Gregory Haskinse864c492008-12-29 09:39:49 -0500568#if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
569
570static struct task_struct *pick_next_highest_task_rt(struct rq *rq, int cpu);
571
572static inline int next_prio(struct rq *rq)
573{
574 struct task_struct *next = pick_next_highest_task_rt(rq, rq->cpu);
575
576 if (next && rt_prio(next->prio))
577 return next->prio;
578 else
579 return MAX_RT_PRIO;
580}
581#endif
582
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100583static inline
584void inc_rt_tasks(struct sched_rt_entity *rt_se, struct rt_rq *rt_rq)
Steven Rostedt63489e42008-01-25 21:08:03 +0100585{
Gregory Haskins4d984272008-12-29 09:39:49 -0500586 int prio = rt_se_prio(rt_se);
587#ifdef CONFIG_SMP
588 struct rq *rq = rq_of_rt_rq(rt_rq);
589#endif
590
591 WARN_ON(!rt_prio(prio));
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100592 rt_rq->rt_nr_running++;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100593#if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
Gregory Haskinse864c492008-12-29 09:39:49 -0500594 if (prio < rt_rq->highest_prio.curr) {
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -0400595
Gregory Haskinse864c492008-12-29 09:39:49 -0500596 /*
597 * If the new task is higher in priority than anything on the
598 * run-queue, we have a new high that must be published to
599 * the world. We also know that the previous high becomes
600 * our next-highest.
601 */
602 rt_rq->highest_prio.next = rt_rq->highest_prio.curr;
603 rt_rq->highest_prio.curr = prio;
Ingo Molnar1100ac92008-06-05 12:25:37 +0200604#ifdef CONFIG_SMP
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -0400605 if (rq->online)
Gregory Haskins4d984272008-12-29 09:39:49 -0500606 cpupri_set(&rq->rd->cpupri, rq->cpu, prio);
Ingo Molnar1100ac92008-06-05 12:25:37 +0200607#endif
Gregory Haskinse864c492008-12-29 09:39:49 -0500608 } else if (prio == rt_rq->highest_prio.curr)
609 /*
610 * If the next task is equal in priority to the highest on
611 * the run-queue, then we implicitly know that the next highest
612 * task cannot be any lower than current
613 */
614 rt_rq->highest_prio.next = prio;
615 else if (prio < rt_rq->highest_prio.next)
616 /*
617 * Otherwise, we need to recompute next-highest
618 */
619 rt_rq->highest_prio.next = next_prio(rq);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100620#endif
Steven Rostedt764a9d62008-01-25 21:08:04 +0100621#ifdef CONFIG_SMP
Gregory Haskins4d984272008-12-29 09:39:49 -0500622 if (rt_se->nr_cpus_allowed > 1)
Gregory Haskins73fe6aa2008-01-25 21:08:07 +0100623 rq->rt.rt_nr_migratory++;
624
Gregory Haskins4d984272008-12-29 09:39:49 -0500625 update_rt_migration(rq);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100626#endif
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100627#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100628 if (rt_se_boosted(rt_se))
629 rt_rq->rt_nr_boosted++;
Peter Zijlstrad0b27fa2008-04-19 19:44:57 +0200630
631 if (rt_rq->tg)
632 start_rt_bandwidth(&rt_rq->tg->rt_bandwidth);
633#else
634 start_rt_bandwidth(&def_rt_bandwidth);
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100635#endif
Steven Rostedt63489e42008-01-25 21:08:03 +0100636}
637
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100638static inline
639void dec_rt_tasks(struct sched_rt_entity *rt_se, struct rt_rq *rt_rq)
Steven Rostedt63489e42008-01-25 21:08:03 +0100640{
Gregory Haskins6e0534f2008-05-12 21:21:01 +0200641#ifdef CONFIG_SMP
Gregory Haskins4d984272008-12-29 09:39:49 -0500642 struct rq *rq = rq_of_rt_rq(rt_rq);
Gregory Haskinse864c492008-12-29 09:39:49 -0500643 int highest_prio = rt_rq->highest_prio.curr;
Gregory Haskins6e0534f2008-05-12 21:21:01 +0200644#endif
645
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100646 WARN_ON(!rt_prio(rt_se_prio(rt_se)));
647 WARN_ON(!rt_rq->rt_nr_running);
648 rt_rq->rt_nr_running--;
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100649#if defined CONFIG_SMP || defined CONFIG_RT_GROUP_SCHED
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100650 if (rt_rq->rt_nr_running) {
Gregory Haskinse864c492008-12-29 09:39:49 -0500651 int prio = rt_se_prio(rt_se);
Steven Rostedt764a9d62008-01-25 21:08:04 +0100652
Gregory Haskinse864c492008-12-29 09:39:49 -0500653 WARN_ON(prio < rt_rq->highest_prio.curr);
654
655 /*
656 * This may have been our highest or next-highest priority
657 * task and therefore we may have some recomputation to do
658 */
659 if (prio == rt_rq->highest_prio.curr) {
660 struct rt_prio_array *array = &rt_rq->active;
661
662 rt_rq->highest_prio.curr =
Steven Rostedt764a9d62008-01-25 21:08:04 +0100663 sched_find_first_bit(array->bitmap);
Gregory Haskinse864c492008-12-29 09:39:49 -0500664 }
665
666 if (prio <= rt_rq->highest_prio.next)
667 rt_rq->highest_prio.next = next_prio(rq);
Steven Rostedt764a9d62008-01-25 21:08:04 +0100668 } else
Gregory Haskinse864c492008-12-29 09:39:49 -0500669 rt_rq->highest_prio.curr = MAX_RT_PRIO;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100670#endif
671#ifdef CONFIG_SMP
Gregory Haskins4d984272008-12-29 09:39:49 -0500672 if (rt_se->nr_cpus_allowed > 1)
Gregory Haskins73fe6aa2008-01-25 21:08:07 +0100673 rq->rt.rt_nr_migratory--;
674
Gregory Haskinse864c492008-12-29 09:39:49 -0500675 if (rq->online && rt_rq->highest_prio.curr != highest_prio)
676 cpupri_set(&rq->rd->cpupri, rq->cpu, rt_rq->highest_prio.curr);
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -0400677
Gregory Haskins4d984272008-12-29 09:39:49 -0500678 update_rt_migration(rq);
Steven Rostedt764a9d62008-01-25 21:08:04 +0100679#endif /* CONFIG_SMP */
Peter Zijlstra052f1dc2008-02-13 15:45:40 +0100680#ifdef CONFIG_RT_GROUP_SCHED
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100681 if (rt_se_boosted(rt_se))
682 rt_rq->rt_nr_boosted--;
683
684 WARN_ON(!rt_rq->rt_nr_running && rt_rq->rt_nr_boosted);
685#endif
Steven Rostedt63489e42008-01-25 21:08:03 +0100686}
687
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200688static void __enqueue_rt_entity(struct sched_rt_entity *rt_se)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200689{
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100690 struct rt_rq *rt_rq = rt_rq_of_se(rt_se);
691 struct rt_prio_array *array = &rt_rq->active;
692 struct rt_rq *group_rq = group_rt_rq(rt_se);
Dmitry Adamushko20b63312008-06-11 00:58:30 +0200693 struct list_head *queue = array->queue + rt_se_prio(rt_se);
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200694
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200695 /*
696 * Don't enqueue the group if its throttled, or when empty.
697 * The latter is a consequence of the former when a child group
698 * get throttled and the current group doesn't have any other
699 * active members.
700 */
701 if (group_rq && (rt_rq_throttled(group_rq) || !group_rq->rt_nr_running))
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100702 return;
Steven Rostedt63489e42008-01-25 21:08:03 +0100703
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200704 list_add_tail(&rt_se->run_list, queue);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100705 __set_bit(rt_se_prio(rt_se), array->bitmap);
Peter Zijlstra78f2c7d2008-01-25 21:08:27 +0100706
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100707 inc_rt_tasks(rt_se, rt_rq);
708}
709
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200710static void __dequeue_rt_entity(struct sched_rt_entity *rt_se)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100711{
712 struct rt_rq *rt_rq = rt_rq_of_se(rt_se);
713 struct rt_prio_array *array = &rt_rq->active;
714
715 list_del_init(&rt_se->run_list);
716 if (list_empty(array->queue + rt_se_prio(rt_se)))
717 __clear_bit(rt_se_prio(rt_se), array->bitmap);
718
719 dec_rt_tasks(rt_se, rt_rq);
720}
721
722/*
723 * Because the prio of an upper entry depends on the lower
724 * entries, we must remove entries top - down.
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100725 */
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200726static void dequeue_rt_stack(struct sched_rt_entity *rt_se)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100727{
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200728 struct sched_rt_entity *back = NULL;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100729
Peter Zijlstra58d6c2d2008-04-19 19:45:00 +0200730 for_each_sched_rt_entity(rt_se) {
731 rt_se->back = back;
732 back = rt_se;
733 }
734
735 for (rt_se = back; rt_se; rt_se = rt_se->back) {
736 if (on_rt_rq(rt_se))
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200737 __dequeue_rt_entity(rt_se);
738 }
739}
740
741static void enqueue_rt_entity(struct sched_rt_entity *rt_se)
742{
743 dequeue_rt_stack(rt_se);
744 for_each_sched_rt_entity(rt_se)
745 __enqueue_rt_entity(rt_se);
746}
747
748static void dequeue_rt_entity(struct sched_rt_entity *rt_se)
749{
750 dequeue_rt_stack(rt_se);
751
752 for_each_sched_rt_entity(rt_se) {
753 struct rt_rq *rt_rq = group_rt_rq(rt_se);
754
755 if (rt_rq && rt_rq->rt_nr_running)
756 __enqueue_rt_entity(rt_se);
Peter Zijlstra58d6c2d2008-04-19 19:45:00 +0200757 }
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200758}
759
760/*
761 * Adding/removing a task to/from a priority array:
762 */
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100763static void enqueue_task_rt(struct rq *rq, struct task_struct *p, int wakeup)
764{
765 struct sched_rt_entity *rt_se = &p->rt;
766
767 if (wakeup)
768 rt_se->timeout = 0;
769
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200770 enqueue_rt_entity(rt_se);
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200771
Gregory Haskins917b6272008-12-29 09:39:53 -0500772 if (!task_current(rq, p) && p->rt.nr_cpus_allowed > 1)
773 enqueue_pushable_task(rq, p);
774
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200775 inc_cpu_load(rq, p->se.load.weight);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100776}
777
Ingo Molnarf02231e2007-08-09 11:16:48 +0200778static void dequeue_task_rt(struct rq *rq, struct task_struct *p, int sleep)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200779{
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100780 struct sched_rt_entity *rt_se = &p->rt;
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200781
Ingo Molnarf1e14ef2007-08-09 11:16:48 +0200782 update_curr_rt(rq);
Peter Zijlstraad2a3f12008-06-19 09:06:57 +0200783 dequeue_rt_entity(rt_se);
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200784
Gregory Haskins917b6272008-12-29 09:39:53 -0500785 dequeue_pushable_task(rq, p);
786
Peter Zijlstrac09595f2008-06-27 13:41:14 +0200787 dec_cpu_load(rq, p->se.load.weight);
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200788}
789
790/*
791 * Put task to the end of the run list without the overhead of dequeue
792 * followed by enqueue.
793 */
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200794static void
795requeue_rt_entity(struct rt_rq *rt_rq, struct sched_rt_entity *rt_se, int head)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200796{
Ingo Molnar1cdad712008-06-19 09:09:15 +0200797 if (on_rt_rq(rt_se)) {
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200798 struct rt_prio_array *array = &rt_rq->active;
799 struct list_head *queue = array->queue + rt_se_prio(rt_se);
800
801 if (head)
802 list_move(&rt_se->run_list, queue);
803 else
804 list_move_tail(&rt_se->run_list, queue);
Ingo Molnar1cdad712008-06-19 09:09:15 +0200805 }
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200806}
807
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200808static void requeue_task_rt(struct rq *rq, struct task_struct *p, int head)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100809{
810 struct sched_rt_entity *rt_se = &p->rt;
811 struct rt_rq *rt_rq;
812
813 for_each_sched_rt_entity(rt_se) {
814 rt_rq = rt_rq_of_se(rt_se);
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200815 requeue_rt_entity(rt_rq, rt_se, head);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100816 }
817}
818
819static void yield_task_rt(struct rq *rq)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200820{
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200821 requeue_task_rt(rq, rq->curr, 0);
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200822}
823
Gregory Haskinse7693a32008-01-25 21:08:09 +0100824#ifdef CONFIG_SMP
Gregory Haskins318e0892008-01-25 21:08:10 +0100825static int find_lowest_rq(struct task_struct *task);
826
Gregory Haskinse7693a32008-01-25 21:08:09 +0100827static int select_task_rq_rt(struct task_struct *p, int sync)
828{
Gregory Haskins318e0892008-01-25 21:08:10 +0100829 struct rq *rq = task_rq(p);
830
831 /*
Steven Rostedte1f47d82008-01-25 21:08:12 +0100832 * If the current task is an RT task, then
833 * try to see if we can wake this RT task up on another
834 * runqueue. Otherwise simply start this RT task
835 * on its current runqueue.
836 *
837 * We want to avoid overloading runqueues. Even if
838 * the RT task is of higher priority than the current RT task.
839 * RT tasks behave differently than other tasks. If
840 * one gets preempted, we try to push it off to another queue.
841 * So trying to keep a preempting RT task on the same
842 * cache hot CPU will force the running RT task to
843 * a cold CPU. So we waste all the cache for the lower
844 * RT task in hopes of saving some of a RT task
845 * that is just being woken and probably will have
846 * cold cache anyway.
Gregory Haskins318e0892008-01-25 21:08:10 +0100847 */
Gregory Haskins17b32792008-01-25 21:08:13 +0100848 if (unlikely(rt_task(rq->curr)) &&
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100849 (p->rt.nr_cpus_allowed > 1)) {
Gregory Haskins318e0892008-01-25 21:08:10 +0100850 int cpu = find_lowest_rq(p);
851
852 return (cpu == -1) ? task_cpu(p) : cpu;
853 }
854
855 /*
856 * Otherwise, just let it ride on the affined RQ and the
857 * post-schedule router will push the preempted task away
858 */
Gregory Haskinse7693a32008-01-25 21:08:09 +0100859 return task_cpu(p);
860}
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200861
862static void check_preempt_equal_prio(struct rq *rq, struct task_struct *p)
863{
Rusty Russell24600ce2008-11-25 02:35:13 +1030864 cpumask_var_t mask;
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200865
866 if (rq->curr->rt.nr_cpus_allowed == 1)
867 return;
868
Rusty Russell24600ce2008-11-25 02:35:13 +1030869 if (!alloc_cpumask_var(&mask, GFP_ATOMIC))
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200870 return;
871
Rusty Russell24600ce2008-11-25 02:35:13 +1030872 if (p->rt.nr_cpus_allowed != 1
873 && cpupri_find(&rq->rd->cpupri, p, mask))
874 goto free;
875
876 if (!cpupri_find(&rq->rd->cpupri, rq->curr, mask))
877 goto free;
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200878
879 /*
880 * There appears to be other cpus that can accept
881 * current and none to run 'p', so lets reschedule
882 * to try and push current away:
883 */
884 requeue_task_rt(rq, p, 1);
885 resched_task(rq->curr);
Rusty Russell24600ce2008-11-25 02:35:13 +1030886free:
887 free_cpumask_var(mask);
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200888}
889
Gregory Haskinse7693a32008-01-25 21:08:09 +0100890#endif /* CONFIG_SMP */
891
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200892/*
893 * Preempt the current task with a newly woken task if needed:
894 */
Peter Zijlstra15afe092008-09-20 23:38:02 +0200895static void check_preempt_curr_rt(struct rq *rq, struct task_struct *p, int sync)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200896{
Gregory Haskins45c01e82008-05-12 21:20:41 +0200897 if (p->prio < rq->curr->prio) {
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200898 resched_task(rq->curr);
Gregory Haskins45c01e82008-05-12 21:20:41 +0200899 return;
900 }
901
902#ifdef CONFIG_SMP
903 /*
904 * If:
905 *
906 * - the newly woken task is of equal priority to the current task
907 * - the newly woken task is non-migratable while current is migratable
908 * - current will be preempted on the next reschedule
909 *
910 * we should check to see if current can readily move to a different
911 * cpu. If so, we will reschedule to allow the push logic to try
912 * to move current somewhere else, making room for our non-migratable
913 * task.
914 */
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +0200915 if (p->prio == rq->curr->prio && !need_resched())
916 check_preempt_equal_prio(rq, p);
Gregory Haskins45c01e82008-05-12 21:20:41 +0200917#endif
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200918}
919
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100920static struct sched_rt_entity *pick_next_rt_entity(struct rq *rq,
921 struct rt_rq *rt_rq)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200922{
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100923 struct rt_prio_array *array = &rt_rq->active;
924 struct sched_rt_entity *next = NULL;
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200925 struct list_head *queue;
926 int idx;
927
928 idx = sched_find_first_bit(array->bitmap);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100929 BUG_ON(idx >= MAX_RT_PRIO);
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200930
931 queue = array->queue + idx;
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100932 next = list_entry(queue->next, struct sched_rt_entity, run_list);
Dmitry Adamushko326587b2008-01-25 21:08:34 +0100933
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200934 return next;
935}
936
Gregory Haskins917b6272008-12-29 09:39:53 -0500937static struct task_struct *_pick_next_task_rt(struct rq *rq)
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100938{
939 struct sched_rt_entity *rt_se;
940 struct task_struct *p;
941 struct rt_rq *rt_rq;
942
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100943 rt_rq = &rq->rt;
944
945 if (unlikely(!rt_rq->rt_nr_running))
946 return NULL;
947
Peter Zijlstra23b0fdf2008-02-13 15:45:39 +0100948 if (rt_rq_throttled(rt_rq))
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100949 return NULL;
950
951 do {
952 rt_se = pick_next_rt_entity(rq, rt_rq);
Dmitry Adamushko326587b2008-01-25 21:08:34 +0100953 BUG_ON(!rt_se);
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100954 rt_rq = group_rt_rq(rt_se);
955 } while (rt_rq);
956
957 p = rt_task_of(rt_se);
958 p->se.exec_start = rq->clock;
Gregory Haskins917b6272008-12-29 09:39:53 -0500959
960 return p;
961}
962
963static struct task_struct *pick_next_task_rt(struct rq *rq)
964{
965 struct task_struct *p = _pick_next_task_rt(rq);
966
967 /* The running task is never eligible for pushing */
968 if (p)
969 dequeue_pushable_task(rq, p);
970
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100971 return p;
972}
973
Ingo Molnar31ee5292007-08-09 11:16:49 +0200974static void put_prev_task_rt(struct rq *rq, struct task_struct *p)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200975{
Ingo Molnarf1e14ef2007-08-09 11:16:48 +0200976 update_curr_rt(rq);
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200977 p->se.exec_start = 0;
Gregory Haskins917b6272008-12-29 09:39:53 -0500978
979 /*
980 * The previous task needs to be made eligible for pushing
981 * if it is still active
982 */
983 if (p->se.on_rq && p->rt.nr_cpus_allowed > 1)
984 enqueue_pushable_task(rq, p);
Ingo Molnarbb44e5d2007-07-09 18:51:58 +0200985}
986
Peter Williams681f3e62007-10-24 18:23:51 +0200987#ifdef CONFIG_SMP
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100988
Steven Rostedte8fa1362008-01-25 21:08:05 +0100989/* Only try algorithms three times */
990#define RT_MAX_TRIES 3
991
Steven Rostedte8fa1362008-01-25 21:08:05 +0100992static void deactivate_task(struct rq *rq, struct task_struct *p, int sleep);
993
Steven Rostedtf65eda42008-01-25 21:08:07 +0100994static int pick_rt_task(struct rq *rq, struct task_struct *p, int cpu)
995{
996 if (!task_running(rq, p) &&
Rusty Russell96f874e22008-11-25 02:35:14 +1030997 (cpu < 0 || cpumask_test_cpu(cpu, &p->cpus_allowed)) &&
Peter Zijlstra6f505b12008-01-25 21:08:30 +0100998 (p->rt.nr_cpus_allowed > 1))
Steven Rostedtf65eda42008-01-25 21:08:07 +0100999 return 1;
1000 return 0;
1001}
1002
Steven Rostedte8fa1362008-01-25 21:08:05 +01001003/* Return the second highest RT task, NULL otherwise */
Ingo Molnar79064fb2008-01-25 21:08:14 +01001004static struct task_struct *pick_next_highest_task_rt(struct rq *rq, int cpu)
Steven Rostedte8fa1362008-01-25 21:08:05 +01001005{
Peter Zijlstra6f505b12008-01-25 21:08:30 +01001006 struct task_struct *next = NULL;
1007 struct sched_rt_entity *rt_se;
1008 struct rt_prio_array *array;
1009 struct rt_rq *rt_rq;
Steven Rostedte8fa1362008-01-25 21:08:05 +01001010 int idx;
1011
Peter Zijlstra6f505b12008-01-25 21:08:30 +01001012 for_each_leaf_rt_rq(rt_rq, rq) {
1013 array = &rt_rq->active;
1014 idx = sched_find_first_bit(array->bitmap);
1015 next_idx:
1016 if (idx >= MAX_RT_PRIO)
1017 continue;
1018 if (next && next->prio < idx)
1019 continue;
1020 list_for_each_entry(rt_se, array->queue + idx, run_list) {
1021 struct task_struct *p = rt_task_of(rt_se);
1022 if (pick_rt_task(rq, p, cpu)) {
1023 next = p;
1024 break;
1025 }
1026 }
1027 if (!next) {
1028 idx = find_next_bit(array->bitmap, MAX_RT_PRIO, idx+1);
1029 goto next_idx;
1030 }
Steven Rostedte8fa1362008-01-25 21:08:05 +01001031 }
1032
Steven Rostedte8fa1362008-01-25 21:08:05 +01001033 return next;
1034}
1035
Rusty Russell0e3900e2008-11-25 02:35:13 +10301036static DEFINE_PER_CPU(cpumask_var_t, local_cpu_mask);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001037
Gregory Haskins6e1254d2008-01-25 21:08:11 +01001038static inline int pick_optimal_cpu(int this_cpu, cpumask_t *mask)
1039{
1040 int first;
1041
1042 /* "this_cpu" is cheaper to preempt than a remote processor */
1043 if ((this_cpu != -1) && cpu_isset(this_cpu, *mask))
1044 return this_cpu;
1045
1046 first = first_cpu(*mask);
1047 if (first != NR_CPUS)
1048 return first;
1049
1050 return -1;
1051}
1052
1053static int find_lowest_rq(struct task_struct *task)
1054{
1055 struct sched_domain *sd;
Rusty Russell96f874e22008-11-25 02:35:14 +10301056 struct cpumask *lowest_mask = __get_cpu_var(local_cpu_mask);
Gregory Haskins6e1254d2008-01-25 21:08:11 +01001057 int this_cpu = smp_processor_id();
1058 int cpu = task_cpu(task);
1059
Gregory Haskins6e0534f2008-05-12 21:21:01 +02001060 if (task->rt.nr_cpus_allowed == 1)
1061 return -1; /* No other targets possible */
1062
1063 if (!cpupri_find(&task_rq(task)->rd->cpupri, task, lowest_mask))
Gregory Haskins06f90db2008-01-25 21:08:13 +01001064 return -1; /* No targets found */
1065
1066 /*
Max Krasnyanskye761b772008-07-15 04:43:49 -07001067 * Only consider CPUs that are usable for migration.
1068 * I guess we might want to change cpupri_find() to ignore those
1069 * in the first place.
1070 */
Rusty Russell96f874e22008-11-25 02:35:14 +10301071 cpumask_and(lowest_mask, lowest_mask, cpu_active_mask);
Max Krasnyanskye761b772008-07-15 04:43:49 -07001072
1073 /*
Gregory Haskins6e1254d2008-01-25 21:08:11 +01001074 * At this point we have built a mask of cpus representing the
1075 * lowest priority tasks in the system. Now we want to elect
1076 * the best one based on our affinity and topology.
1077 *
1078 * We prioritize the last cpu that the task executed on since
1079 * it is most likely cache-hot in that location.
1080 */
Rusty Russell96f874e22008-11-25 02:35:14 +10301081 if (cpumask_test_cpu(cpu, lowest_mask))
Gregory Haskins6e1254d2008-01-25 21:08:11 +01001082 return cpu;
1083
1084 /*
1085 * Otherwise, we consult the sched_domains span maps to figure
1086 * out which cpu is logically closest to our hot cache data.
1087 */
1088 if (this_cpu == cpu)
1089 this_cpu = -1; /* Skip this_cpu opt if the same */
1090
1091 for_each_domain(cpu, sd) {
1092 if (sd->flags & SD_WAKE_AFFINE) {
1093 cpumask_t domain_mask;
1094 int best_cpu;
1095
Rusty Russell758b2cd2008-11-25 02:35:04 +10301096 cpumask_and(&domain_mask, sched_domain_span(sd),
1097 lowest_mask);
Gregory Haskins6e1254d2008-01-25 21:08:11 +01001098
1099 best_cpu = pick_optimal_cpu(this_cpu,
1100 &domain_mask);
1101 if (best_cpu != -1)
1102 return best_cpu;
1103 }
1104 }
1105
1106 /*
1107 * And finally, if there were no matches within the domains
1108 * just give the caller *something* to work with from the compatible
1109 * locations.
1110 */
1111 return pick_optimal_cpu(this_cpu, lowest_mask);
Gregory Haskins07b40322008-01-25 21:08:10 +01001112}
1113
Steven Rostedte8fa1362008-01-25 21:08:05 +01001114/* Will lock the rq it finds */
Ingo Molnar4df64c02008-01-25 21:08:15 +01001115static struct rq *find_lock_lowest_rq(struct task_struct *task, struct rq *rq)
Steven Rostedte8fa1362008-01-25 21:08:05 +01001116{
1117 struct rq *lowest_rq = NULL;
Steven Rostedte8fa1362008-01-25 21:08:05 +01001118 int tries;
Ingo Molnar4df64c02008-01-25 21:08:15 +01001119 int cpu;
Steven Rostedte8fa1362008-01-25 21:08:05 +01001120
1121 for (tries = 0; tries < RT_MAX_TRIES; tries++) {
Gregory Haskins07b40322008-01-25 21:08:10 +01001122 cpu = find_lowest_rq(task);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001123
Gregory Haskins2de0b462008-01-25 21:08:10 +01001124 if ((cpu == -1) || (cpu == rq->cpu))
Steven Rostedte8fa1362008-01-25 21:08:05 +01001125 break;
1126
Gregory Haskins07b40322008-01-25 21:08:10 +01001127 lowest_rq = cpu_rq(cpu);
1128
Steven Rostedte8fa1362008-01-25 21:08:05 +01001129 /* if the prio of this runqueue changed, try again */
Gregory Haskins07b40322008-01-25 21:08:10 +01001130 if (double_lock_balance(rq, lowest_rq)) {
Steven Rostedte8fa1362008-01-25 21:08:05 +01001131 /*
1132 * We had to unlock the run queue. In
1133 * the mean time, task could have
1134 * migrated already or had its affinity changed.
1135 * Also make sure that it wasn't scheduled on its rq.
1136 */
Gregory Haskins07b40322008-01-25 21:08:10 +01001137 if (unlikely(task_rq(task) != rq ||
Rusty Russell96f874e22008-11-25 02:35:14 +10301138 !cpumask_test_cpu(lowest_rq->cpu,
1139 &task->cpus_allowed) ||
Gregory Haskins07b40322008-01-25 21:08:10 +01001140 task_running(rq, task) ||
Steven Rostedte8fa1362008-01-25 21:08:05 +01001141 !task->se.on_rq)) {
Ingo Molnar4df64c02008-01-25 21:08:15 +01001142
Steven Rostedte8fa1362008-01-25 21:08:05 +01001143 spin_unlock(&lowest_rq->lock);
1144 lowest_rq = NULL;
1145 break;
1146 }
1147 }
1148
1149 /* If this rq is still suitable use it. */
Gregory Haskinse864c492008-12-29 09:39:49 -05001150 if (lowest_rq->rt.highest_prio.curr > task->prio)
Steven Rostedte8fa1362008-01-25 21:08:05 +01001151 break;
1152
1153 /* try again */
Peter Zijlstra1b12bbc2008-08-11 09:30:22 +02001154 double_unlock_balance(rq, lowest_rq);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001155 lowest_rq = NULL;
1156 }
1157
1158 return lowest_rq;
1159}
1160
Gregory Haskins917b6272008-12-29 09:39:53 -05001161static inline int has_pushable_tasks(struct rq *rq)
1162{
1163 return !plist_head_empty(&rq->rt.pushable_tasks);
1164}
1165
1166static struct task_struct *pick_next_pushable_task(struct rq *rq)
1167{
1168 struct task_struct *p;
1169
1170 if (!has_pushable_tasks(rq))
1171 return NULL;
1172
1173 p = plist_first_entry(&rq->rt.pushable_tasks,
1174 struct task_struct, pushable_tasks);
1175
1176 BUG_ON(rq->cpu != task_cpu(p));
1177 BUG_ON(task_current(rq, p));
1178 BUG_ON(p->rt.nr_cpus_allowed <= 1);
1179
1180 BUG_ON(!p->se.on_rq);
1181 BUG_ON(!rt_task(p));
1182
1183 return p;
1184}
1185
Steven Rostedte8fa1362008-01-25 21:08:05 +01001186/*
1187 * If the current CPU has more than one RT task, see if the non
1188 * running task can migrate over to a CPU that is running a task
1189 * of lesser priority.
1190 */
Gregory Haskins697f0a42008-01-25 21:08:09 +01001191static int push_rt_task(struct rq *rq)
Steven Rostedte8fa1362008-01-25 21:08:05 +01001192{
1193 struct task_struct *next_task;
1194 struct rq *lowest_rq;
Steven Rostedte8fa1362008-01-25 21:08:05 +01001195
Gregory Haskinsa22d7fc2008-01-25 21:08:12 +01001196 if (!rq->rt.overloaded)
1197 return 0;
1198
Gregory Haskins917b6272008-12-29 09:39:53 -05001199 next_task = pick_next_pushable_task(rq);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001200 if (!next_task)
1201 return 0;
1202
1203 retry:
Gregory Haskins697f0a42008-01-25 21:08:09 +01001204 if (unlikely(next_task == rq->curr)) {
Steven Rostedtf65eda42008-01-25 21:08:07 +01001205 WARN_ON(1);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001206 return 0;
Steven Rostedtf65eda42008-01-25 21:08:07 +01001207 }
Steven Rostedte8fa1362008-01-25 21:08:05 +01001208
1209 /*
1210 * It's possible that the next_task slipped in of
1211 * higher priority than current. If that's the case
1212 * just reschedule current.
1213 */
Gregory Haskins697f0a42008-01-25 21:08:09 +01001214 if (unlikely(next_task->prio < rq->curr->prio)) {
1215 resched_task(rq->curr);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001216 return 0;
1217 }
1218
Gregory Haskins697f0a42008-01-25 21:08:09 +01001219 /* We might release rq lock */
Steven Rostedte8fa1362008-01-25 21:08:05 +01001220 get_task_struct(next_task);
1221
1222 /* find_lock_lowest_rq locks the rq if found */
Gregory Haskins697f0a42008-01-25 21:08:09 +01001223 lowest_rq = find_lock_lowest_rq(next_task, rq);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001224 if (!lowest_rq) {
1225 struct task_struct *task;
1226 /*
Gregory Haskins697f0a42008-01-25 21:08:09 +01001227 * find lock_lowest_rq releases rq->lock
Gregory Haskins15635132008-12-29 09:39:53 -05001228 * so it is possible that next_task has migrated.
1229 *
1230 * We need to make sure that the task is still on the same
1231 * run-queue and is also still the next task eligible for
1232 * pushing.
Steven Rostedte8fa1362008-01-25 21:08:05 +01001233 */
Gregory Haskins917b6272008-12-29 09:39:53 -05001234 task = pick_next_pushable_task(rq);
Gregory Haskins15635132008-12-29 09:39:53 -05001235 if (task_cpu(next_task) == rq->cpu && task == next_task) {
1236 /*
1237 * If we get here, the task hasnt moved at all, but
1238 * it has failed to push. We will not try again,
1239 * since the other cpus will pull from us when they
1240 * are ready.
1241 */
1242 dequeue_pushable_task(rq, next_task);
1243 goto out;
Steven Rostedte8fa1362008-01-25 21:08:05 +01001244 }
Gregory Haskins917b6272008-12-29 09:39:53 -05001245
Gregory Haskins15635132008-12-29 09:39:53 -05001246 if (!task)
1247 /* No more tasks, just exit */
1248 goto out;
1249
Gregory Haskins917b6272008-12-29 09:39:53 -05001250 /*
Gregory Haskins15635132008-12-29 09:39:53 -05001251 * Something has shifted, try again.
Gregory Haskins917b6272008-12-29 09:39:53 -05001252 */
Gregory Haskins15635132008-12-29 09:39:53 -05001253 put_task_struct(next_task);
1254 next_task = task;
1255 goto retry;
Steven Rostedte8fa1362008-01-25 21:08:05 +01001256 }
1257
Gregory Haskins697f0a42008-01-25 21:08:09 +01001258 deactivate_task(rq, next_task, 0);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001259 set_task_cpu(next_task, lowest_rq->cpu);
1260 activate_task(lowest_rq, next_task, 0);
1261
1262 resched_task(lowest_rq->curr);
1263
Peter Zijlstra1b12bbc2008-08-11 09:30:22 +02001264 double_unlock_balance(rq, lowest_rq);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001265
Steven Rostedte8fa1362008-01-25 21:08:05 +01001266out:
1267 put_task_struct(next_task);
1268
Gregory Haskins917b6272008-12-29 09:39:53 -05001269 return 1;
Steven Rostedte8fa1362008-01-25 21:08:05 +01001270}
1271
Steven Rostedte8fa1362008-01-25 21:08:05 +01001272static void push_rt_tasks(struct rq *rq)
1273{
1274 /* push_rt_task will return true if it moved an RT */
1275 while (push_rt_task(rq))
1276 ;
1277}
1278
Steven Rostedtf65eda42008-01-25 21:08:07 +01001279static int pull_rt_task(struct rq *this_rq)
1280{
Ingo Molnar80bf3172008-01-25 21:08:17 +01001281 int this_cpu = this_rq->cpu, ret = 0, cpu;
Gregory Haskinsa8728942008-12-29 09:39:49 -05001282 struct task_struct *p;
Steven Rostedtf65eda42008-01-25 21:08:07 +01001283 struct rq *src_rq;
Steven Rostedtf65eda42008-01-25 21:08:07 +01001284
Gregory Haskins637f5082008-01-25 21:08:18 +01001285 if (likely(!rt_overloaded(this_rq)))
Steven Rostedtf65eda42008-01-25 21:08:07 +01001286 return 0;
1287
Rusty Russellc6c49272008-11-25 02:35:05 +10301288 for_each_cpu(cpu, this_rq->rd->rto_mask) {
Steven Rostedtf65eda42008-01-25 21:08:07 +01001289 if (this_cpu == cpu)
1290 continue;
1291
1292 src_rq = cpu_rq(cpu);
Gregory Haskins74ab8e42008-12-29 09:39:50 -05001293
1294 /*
1295 * Don't bother taking the src_rq->lock if the next highest
1296 * task is known to be lower-priority than our current task.
1297 * This may look racy, but if this value is about to go
1298 * logically higher, the src_rq will push this task away.
1299 * And if its going logically lower, we do not care
1300 */
1301 if (src_rq->rt.highest_prio.next >=
1302 this_rq->rt.highest_prio.curr)
1303 continue;
1304
Steven Rostedtf65eda42008-01-25 21:08:07 +01001305 /*
1306 * We can potentially drop this_rq's lock in
1307 * double_lock_balance, and another CPU could
Gregory Haskinsa8728942008-12-29 09:39:49 -05001308 * alter this_rq
Steven Rostedtf65eda42008-01-25 21:08:07 +01001309 */
Gregory Haskinsa8728942008-12-29 09:39:49 -05001310 double_lock_balance(this_rq, src_rq);
Steven Rostedtf65eda42008-01-25 21:08:07 +01001311
1312 /*
1313 * Are there still pullable RT tasks?
1314 */
Mike Galbraith614ee1f2008-01-25 21:08:30 +01001315 if (src_rq->rt.rt_nr_running <= 1)
1316 goto skip;
Steven Rostedtf65eda42008-01-25 21:08:07 +01001317
Steven Rostedtf65eda42008-01-25 21:08:07 +01001318 p = pick_next_highest_task_rt(src_rq, this_cpu);
1319
1320 /*
1321 * Do we have an RT task that preempts
1322 * the to-be-scheduled task?
1323 */
Gregory Haskinsa8728942008-12-29 09:39:49 -05001324 if (p && (p->prio < this_rq->rt.highest_prio.curr)) {
Steven Rostedtf65eda42008-01-25 21:08:07 +01001325 WARN_ON(p == src_rq->curr);
1326 WARN_ON(!p->se.on_rq);
1327
1328 /*
1329 * There's a chance that p is higher in priority
1330 * than what's currently running on its cpu.
1331 * This is just that p is wakeing up and hasn't
1332 * had a chance to schedule. We only pull
1333 * p if it is lower in priority than the
Gregory Haskinsa8728942008-12-29 09:39:49 -05001334 * current task on the run queue
Steven Rostedtf65eda42008-01-25 21:08:07 +01001335 */
Gregory Haskinsa8728942008-12-29 09:39:49 -05001336 if (p->prio < src_rq->curr->prio)
Mike Galbraith614ee1f2008-01-25 21:08:30 +01001337 goto skip;
Steven Rostedtf65eda42008-01-25 21:08:07 +01001338
1339 ret = 1;
1340
1341 deactivate_task(src_rq, p, 0);
1342 set_task_cpu(p, this_cpu);
1343 activate_task(this_rq, p, 0);
1344 /*
1345 * We continue with the search, just in
1346 * case there's an even higher prio task
1347 * in another runqueue. (low likelyhood
1348 * but possible)
Steven Rostedtf65eda42008-01-25 21:08:07 +01001349 */
Steven Rostedtf65eda42008-01-25 21:08:07 +01001350 }
Mike Galbraith614ee1f2008-01-25 21:08:30 +01001351 skip:
Peter Zijlstra1b12bbc2008-08-11 09:30:22 +02001352 double_unlock_balance(this_rq, src_rq);
Steven Rostedtf65eda42008-01-25 21:08:07 +01001353 }
1354
1355 return ret;
1356}
1357
Steven Rostedt9a897c52008-01-25 21:08:22 +01001358static void pre_schedule_rt(struct rq *rq, struct task_struct *prev)
Steven Rostedtf65eda42008-01-25 21:08:07 +01001359{
1360 /* Try to pull RT tasks here if we lower this rq's prio */
Gregory Haskinse864c492008-12-29 09:39:49 -05001361 if (unlikely(rt_task(prev)) && rq->rt.highest_prio.curr > prev->prio)
Steven Rostedtf65eda42008-01-25 21:08:07 +01001362 pull_rt_task(rq);
1363}
1364
Gregory Haskins967fc042008-12-29 09:39:52 -05001365/*
1366 * assumes rq->lock is held
1367 */
1368static int needs_post_schedule_rt(struct rq *rq)
1369{
Gregory Haskins917b6272008-12-29 09:39:53 -05001370 return has_pushable_tasks(rq);
Gregory Haskins967fc042008-12-29 09:39:52 -05001371}
1372
Steven Rostedt9a897c52008-01-25 21:08:22 +01001373static void post_schedule_rt(struct rq *rq)
Steven Rostedte8fa1362008-01-25 21:08:05 +01001374{
1375 /*
Gregory Haskins967fc042008-12-29 09:39:52 -05001376 * This is only called if needs_post_schedule_rt() indicates that
1377 * we need to push tasks away
Steven Rostedte8fa1362008-01-25 21:08:05 +01001378 */
Gregory Haskins967fc042008-12-29 09:39:52 -05001379 spin_lock_irq(&rq->lock);
1380 push_rt_tasks(rq);
1381 spin_unlock_irq(&rq->lock);
Steven Rostedte8fa1362008-01-25 21:08:05 +01001382}
1383
Gregory Haskins8ae121a2008-04-23 07:13:29 -04001384/*
1385 * If we are not running and we are not going to reschedule soon, we should
1386 * try to push tasks away now
1387 */
Steven Rostedt9a897c52008-01-25 21:08:22 +01001388static void task_wake_up_rt(struct rq *rq, struct task_struct *p)
Steven Rostedt4642daf2008-01-25 21:08:07 +01001389{
Steven Rostedt9a897c52008-01-25 21:08:22 +01001390 if (!task_running(rq, p) &&
Gregory Haskins8ae121a2008-04-23 07:13:29 -04001391 !test_tsk_need_resched(rq->curr) &&
Gregory Haskins917b6272008-12-29 09:39:53 -05001392 has_pushable_tasks(rq) &&
Gregory Haskins777c2f32008-12-29 09:39:50 -05001393 p->rt.nr_cpus_allowed > 1)
Steven Rostedt4642daf2008-01-25 21:08:07 +01001394 push_rt_tasks(rq);
1395}
1396
Peter Williams43010652007-08-09 11:16:46 +02001397static unsigned long
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001398load_balance_rt(struct rq *this_rq, int this_cpu, struct rq *busiest,
Peter Williamse1d14842007-10-24 18:23:51 +02001399 unsigned long max_load_move,
1400 struct sched_domain *sd, enum cpu_idle_type idle,
1401 int *all_pinned, int *this_best_prio)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001402{
Steven Rostedtc7a1e462008-01-25 21:08:07 +01001403 /* don't touch RT tasks */
1404 return 0;
Peter Williamse1d14842007-10-24 18:23:51 +02001405}
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001406
Peter Williamse1d14842007-10-24 18:23:51 +02001407static int
1408move_one_task_rt(struct rq *this_rq, int this_cpu, struct rq *busiest,
1409 struct sched_domain *sd, enum cpu_idle_type idle)
1410{
Steven Rostedtc7a1e462008-01-25 21:08:07 +01001411 /* don't touch RT tasks */
1412 return 0;
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001413}
Ingo Molnardeeeccd2008-01-25 21:08:15 +01001414
Mike Traviscd8ba7c2008-03-26 14:23:49 -07001415static void set_cpus_allowed_rt(struct task_struct *p,
Rusty Russell96f874e22008-11-25 02:35:14 +10301416 const struct cpumask *new_mask)
Gregory Haskins73fe6aa2008-01-25 21:08:07 +01001417{
Rusty Russell96f874e22008-11-25 02:35:14 +10301418 int weight = cpumask_weight(new_mask);
Gregory Haskins73fe6aa2008-01-25 21:08:07 +01001419
1420 BUG_ON(!rt_task(p));
1421
1422 /*
1423 * Update the migration status of the RQ if we have an RT task
1424 * which is running AND changing its weight value.
1425 */
Peter Zijlstra6f505b12008-01-25 21:08:30 +01001426 if (p->se.on_rq && (weight != p->rt.nr_cpus_allowed)) {
Gregory Haskins73fe6aa2008-01-25 21:08:07 +01001427 struct rq *rq = task_rq(p);
1428
Gregory Haskins917b6272008-12-29 09:39:53 -05001429 if (!task_current(rq, p)) {
1430 /*
1431 * Make sure we dequeue this task from the pushable list
1432 * before going further. It will either remain off of
1433 * the list because we are no longer pushable, or it
1434 * will be requeued.
1435 */
1436 if (p->rt.nr_cpus_allowed > 1)
1437 dequeue_pushable_task(rq, p);
1438
1439 /*
1440 * Requeue if our weight is changing and still > 1
1441 */
1442 if (weight > 1)
1443 enqueue_pushable_task(rq, p);
1444
1445 }
1446
Peter Zijlstra6f505b12008-01-25 21:08:30 +01001447 if ((p->rt.nr_cpus_allowed <= 1) && (weight > 1)) {
Gregory Haskins73fe6aa2008-01-25 21:08:07 +01001448 rq->rt.rt_nr_migratory++;
Peter Zijlstra6f505b12008-01-25 21:08:30 +01001449 } else if ((p->rt.nr_cpus_allowed > 1) && (weight <= 1)) {
Gregory Haskins73fe6aa2008-01-25 21:08:07 +01001450 BUG_ON(!rq->rt.rt_nr_migratory);
1451 rq->rt.rt_nr_migratory--;
1452 }
1453
1454 update_rt_migration(rq);
1455 }
1456
Rusty Russell96f874e22008-11-25 02:35:14 +10301457 cpumask_copy(&p->cpus_allowed, new_mask);
Peter Zijlstra6f505b12008-01-25 21:08:30 +01001458 p->rt.nr_cpus_allowed = weight;
Gregory Haskins73fe6aa2008-01-25 21:08:07 +01001459}
Ingo Molnardeeeccd2008-01-25 21:08:15 +01001460
Ingo Molnarbdd7c812008-01-25 21:08:18 +01001461/* Assumes rq->lock is held */
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04001462static void rq_online_rt(struct rq *rq)
Ingo Molnarbdd7c812008-01-25 21:08:18 +01001463{
1464 if (rq->rt.overloaded)
1465 rt_set_overload(rq);
Gregory Haskins6e0534f2008-05-12 21:21:01 +02001466
Peter Zijlstra7def2be2008-06-05 14:49:58 +02001467 __enable_runtime(rq);
1468
Gregory Haskinse864c492008-12-29 09:39:49 -05001469 cpupri_set(&rq->rd->cpupri, rq->cpu, rq->rt.highest_prio.curr);
Ingo Molnarbdd7c812008-01-25 21:08:18 +01001470}
1471
1472/* Assumes rq->lock is held */
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04001473static void rq_offline_rt(struct rq *rq)
Ingo Molnarbdd7c812008-01-25 21:08:18 +01001474{
1475 if (rq->rt.overloaded)
1476 rt_clear_overload(rq);
Gregory Haskins6e0534f2008-05-12 21:21:01 +02001477
Peter Zijlstra7def2be2008-06-05 14:49:58 +02001478 __disable_runtime(rq);
1479
Gregory Haskins6e0534f2008-05-12 21:21:01 +02001480 cpupri_set(&rq->rd->cpupri, rq->cpu, CPUPRI_INVALID);
Ingo Molnarbdd7c812008-01-25 21:08:18 +01001481}
Steven Rostedtcb469842008-01-25 21:08:22 +01001482
1483/*
1484 * When switch from the rt queue, we bring ourselves to a position
1485 * that we might want to pull RT tasks from other runqueues.
1486 */
1487static void switched_from_rt(struct rq *rq, struct task_struct *p,
1488 int running)
1489{
1490 /*
1491 * If there are other RT tasks then we will reschedule
1492 * and the scheduling of the other RT tasks will handle
1493 * the balancing. But if we are the last RT task
1494 * we may need to handle the pulling of RT tasks
1495 * now.
1496 */
1497 if (!rq->rt.rt_nr_running)
1498 pull_rt_task(rq);
1499}
Rusty Russell3d8cbdf2008-11-25 09:58:41 +10301500
1501static inline void init_sched_rt_class(void)
1502{
1503 unsigned int i;
1504
1505 for_each_possible_cpu(i)
1506 alloc_cpumask_var(&per_cpu(local_cpu_mask, i), GFP_KERNEL);
1507}
Steven Rostedte8fa1362008-01-25 21:08:05 +01001508#endif /* CONFIG_SMP */
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001509
Steven Rostedtcb469842008-01-25 21:08:22 +01001510/*
1511 * When switching a task to RT, we may overload the runqueue
1512 * with RT tasks. In this case we try to push them off to
1513 * other runqueues.
1514 */
1515static void switched_to_rt(struct rq *rq, struct task_struct *p,
1516 int running)
1517{
1518 int check_resched = 1;
1519
1520 /*
1521 * If we are already running, then there's nothing
1522 * that needs to be done. But if we are not running
1523 * we may need to preempt the current running task.
1524 * If that current running task is also an RT task
1525 * then see if we can move to another run queue.
1526 */
1527 if (!running) {
1528#ifdef CONFIG_SMP
1529 if (rq->rt.overloaded && push_rt_task(rq) &&
1530 /* Don't resched if we changed runqueues */
1531 rq != task_rq(p))
1532 check_resched = 0;
1533#endif /* CONFIG_SMP */
1534 if (check_resched && p->prio < rq->curr->prio)
1535 resched_task(rq->curr);
1536 }
1537}
1538
1539/*
1540 * Priority of the task has changed. This may cause
1541 * us to initiate a push or pull.
1542 */
1543static void prio_changed_rt(struct rq *rq, struct task_struct *p,
1544 int oldprio, int running)
1545{
1546 if (running) {
1547#ifdef CONFIG_SMP
1548 /*
1549 * If our priority decreases while running, we
1550 * may need to pull tasks to this runqueue.
1551 */
1552 if (oldprio < p->prio)
1553 pull_rt_task(rq);
1554 /*
1555 * If there's a higher priority task waiting to run
Steven Rostedt6fa46fa2008-03-05 10:00:12 -05001556 * then reschedule. Note, the above pull_rt_task
1557 * can release the rq lock and p could migrate.
1558 * Only reschedule if p is still on the same runqueue.
Steven Rostedtcb469842008-01-25 21:08:22 +01001559 */
Gregory Haskinse864c492008-12-29 09:39:49 -05001560 if (p->prio > rq->rt.highest_prio.curr && rq->curr == p)
Steven Rostedtcb469842008-01-25 21:08:22 +01001561 resched_task(p);
1562#else
1563 /* For UP simply resched on drop of prio */
1564 if (oldprio < p->prio)
1565 resched_task(p);
1566#endif /* CONFIG_SMP */
1567 } else {
1568 /*
1569 * This task is not running, but if it is
1570 * greater than the current running task
1571 * then reschedule.
1572 */
1573 if (p->prio < rq->curr->prio)
1574 resched_task(rq->curr);
1575 }
1576}
1577
Peter Zijlstra78f2c7d2008-01-25 21:08:27 +01001578static void watchdog(struct rq *rq, struct task_struct *p)
1579{
1580 unsigned long soft, hard;
1581
1582 if (!p->signal)
1583 return;
1584
1585 soft = p->signal->rlim[RLIMIT_RTTIME].rlim_cur;
1586 hard = p->signal->rlim[RLIMIT_RTTIME].rlim_max;
1587
1588 if (soft != RLIM_INFINITY) {
1589 unsigned long next;
1590
1591 p->rt.timeout++;
1592 next = DIV_ROUND_UP(min(soft, hard), USEC_PER_SEC/HZ);
Peter Zijlstra5a52dd52008-01-25 21:08:32 +01001593 if (p->rt.timeout > next)
Frank Mayharf06febc2008-09-12 09:54:39 -07001594 p->cputime_expires.sched_exp = p->se.sum_exec_runtime;
Peter Zijlstra78f2c7d2008-01-25 21:08:27 +01001595 }
1596}
Steven Rostedtcb469842008-01-25 21:08:22 +01001597
Peter Zijlstra8f4d37e2008-01-25 21:08:29 +01001598static void task_tick_rt(struct rq *rq, struct task_struct *p, int queued)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001599{
Peter Zijlstra67e2be02007-12-20 15:01:17 +01001600 update_curr_rt(rq);
1601
Peter Zijlstra78f2c7d2008-01-25 21:08:27 +01001602 watchdog(rq, p);
1603
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001604 /*
1605 * RR tasks need a special form of timeslice management.
1606 * FIFO tasks have no timeslices.
1607 */
1608 if (p->policy != SCHED_RR)
1609 return;
1610
Peter Zijlstrafa717062008-01-25 21:08:27 +01001611 if (--p->rt.time_slice)
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001612 return;
1613
Peter Zijlstrafa717062008-01-25 21:08:27 +01001614 p->rt.time_slice = DEF_TIMESLICE;
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001615
Dmitry Adamushko98fbc792007-08-24 20:39:10 +02001616 /*
1617 * Requeue to the end of queue if we are not the only element
1618 * on the queue:
1619 */
Peter Zijlstrafa717062008-01-25 21:08:27 +01001620 if (p->rt.run_list.prev != p->rt.run_list.next) {
Dmitry Adamushko7ebefa82008-07-01 23:32:15 +02001621 requeue_task_rt(rq, p, 0);
Dmitry Adamushko98fbc792007-08-24 20:39:10 +02001622 set_tsk_need_resched(p);
1623 }
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001624}
1625
Srivatsa Vaddagiri83b699e2007-10-15 17:00:08 +02001626static void set_curr_task_rt(struct rq *rq)
1627{
1628 struct task_struct *p = rq->curr;
1629
1630 p->se.exec_start = rq->clock;
Gregory Haskins917b6272008-12-29 09:39:53 -05001631
1632 /* The running task is never eligible for pushing */
1633 dequeue_pushable_task(rq, p);
Srivatsa Vaddagiri83b699e2007-10-15 17:00:08 +02001634}
1635
Harvey Harrison2abdad02008-04-25 10:53:13 -07001636static const struct sched_class rt_sched_class = {
Ingo Molnar5522d5d2007-10-15 17:00:12 +02001637 .next = &fair_sched_class,
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001638 .enqueue_task = enqueue_task_rt,
1639 .dequeue_task = dequeue_task_rt,
1640 .yield_task = yield_task_rt,
1641
1642 .check_preempt_curr = check_preempt_curr_rt,
1643
1644 .pick_next_task = pick_next_task_rt,
1645 .put_prev_task = put_prev_task_rt,
1646
Peter Williams681f3e62007-10-24 18:23:51 +02001647#ifdef CONFIG_SMP
Li Zefan4ce72a22008-10-22 15:25:26 +08001648 .select_task_rq = select_task_rq_rt,
1649
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001650 .load_balance = load_balance_rt,
Peter Williamse1d14842007-10-24 18:23:51 +02001651 .move_one_task = move_one_task_rt,
Gregory Haskins73fe6aa2008-01-25 21:08:07 +01001652 .set_cpus_allowed = set_cpus_allowed_rt,
Gregory Haskins1f11eb6a2008-06-04 15:04:05 -04001653 .rq_online = rq_online_rt,
1654 .rq_offline = rq_offline_rt,
Steven Rostedt9a897c52008-01-25 21:08:22 +01001655 .pre_schedule = pre_schedule_rt,
Gregory Haskins967fc042008-12-29 09:39:52 -05001656 .needs_post_schedule = needs_post_schedule_rt,
Steven Rostedt9a897c52008-01-25 21:08:22 +01001657 .post_schedule = post_schedule_rt,
1658 .task_wake_up = task_wake_up_rt,
Steven Rostedtcb469842008-01-25 21:08:22 +01001659 .switched_from = switched_from_rt,
Peter Williams681f3e62007-10-24 18:23:51 +02001660#endif
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001661
Srivatsa Vaddagiri83b699e2007-10-15 17:00:08 +02001662 .set_curr_task = set_curr_task_rt,
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001663 .task_tick = task_tick_rt,
Steven Rostedtcb469842008-01-25 21:08:22 +01001664
1665 .prio_changed = prio_changed_rt,
1666 .switched_to = switched_to_rt,
Ingo Molnarbb44e5d2007-07-09 18:51:58 +02001667};
Peter Zijlstraada18de2008-06-19 14:22:24 +02001668
1669#ifdef CONFIG_SCHED_DEBUG
1670extern void print_rt_rq(struct seq_file *m, int cpu, struct rt_rq *rt_rq);
1671
1672static void print_rt_stats(struct seq_file *m, int cpu)
1673{
1674 struct rt_rq *rt_rq;
1675
1676 rcu_read_lock();
1677 for_each_leaf_rt_rq(rt_rq, cpu_rq(cpu))
1678 print_rt_rq(m, cpu, rt_rq);
1679 rcu_read_unlock();
1680}
Dhaval Giani55e12e52008-06-24 23:39:43 +05301681#endif /* CONFIG_SCHED_DEBUG */
Rusty Russell0e3900e2008-11-25 02:35:13 +10301682