sched.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797
  1. /* sched.c - SPU scheduler.
  2. *
  3. * Copyright (C) IBM 2005
  4. * Author: Mark Nutter <mnutter@us.ibm.com>
  5. *
  6. * 2006-03-31 NUMA domains added.
  7. *
  8. * This program is free software; you can redistribute it and/or modify
  9. * it under the terms of the GNU General Public License as published by
  10. * the Free Software Foundation; either version 2, or (at your option)
  11. * any later version.
  12. *
  13. * This program is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. * GNU General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU General Public License
  19. * along with this program; if not, write to the Free Software
  20. * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21. */
  22. #undef DEBUG
  23. #include <linux/module.h>
  24. #include <linux/errno.h>
  25. #include <linux/sched.h>
  26. #include <linux/kernel.h>
  27. #include <linux/mm.h>
  28. #include <linux/completion.h>
  29. #include <linux/vmalloc.h>
  30. #include <linux/smp.h>
  31. #include <linux/stddef.h>
  32. #include <linux/unistd.h>
  33. #include <linux/numa.h>
  34. #include <linux/mutex.h>
  35. #include <linux/notifier.h>
  36. #include <linux/kthread.h>
  37. #include <linux/pid_namespace.h>
  38. #include <linux/proc_fs.h>
  39. #include <linux/seq_file.h>
  40. #include <asm/io.h>
  41. #include <asm/mmu_context.h>
  42. #include <asm/spu.h>
  43. #include <asm/spu_csa.h>
  44. #include <asm/spu_priv1.h>
  45. #include "spufs.h"
  46. struct spu_prio_array {
  47. DECLARE_BITMAP(bitmap, MAX_PRIO);
  48. struct list_head runq[MAX_PRIO];
  49. spinlock_t runq_lock;
  50. struct list_head active_list[MAX_NUMNODES];
  51. struct mutex active_mutex[MAX_NUMNODES];
  52. int nr_active[MAX_NUMNODES];
  53. int nr_waiting;
  54. };
  55. static unsigned long spu_avenrun[3];
  56. static struct spu_prio_array *spu_prio;
  57. static struct task_struct *spusched_task;
  58. static struct timer_list spusched_timer;
  59. /*
  60. * Priority of a normal, non-rt, non-niced'd process (aka nice level 0).
  61. */
  62. #define NORMAL_PRIO 120
  63. /*
  64. * Frequency of the spu scheduler tick. By default we do one SPU scheduler
  65. * tick for every 10 CPU scheduler ticks.
  66. */
  67. #define SPUSCHED_TICK (10)
  68. /*
  69. * These are the 'tuning knobs' of the scheduler:
  70. *
  71. * Minimum timeslice is 5 msecs (or 1 spu scheduler tick, whichever is
  72. * larger), default timeslice is 100 msecs, maximum timeslice is 800 msecs.
  73. */
  74. #define MIN_SPU_TIMESLICE max(5 * HZ / (1000 * SPUSCHED_TICK), 1)
  75. #define DEF_SPU_TIMESLICE (100 * HZ / (1000 * SPUSCHED_TICK))
  76. #define MAX_USER_PRIO (MAX_PRIO - MAX_RT_PRIO)
  77. #define SCALE_PRIO(x, prio) \
  78. max(x * (MAX_PRIO - prio) / (MAX_USER_PRIO / 2), MIN_SPU_TIMESLICE)
  79. /*
  80. * scale user-nice values [ -20 ... 0 ... 19 ] to time slice values:
  81. * [800ms ... 100ms ... 5ms]
  82. *
  83. * The higher a thread's priority, the bigger timeslices
  84. * it gets during one round of execution. But even the lowest
  85. * priority thread gets MIN_TIMESLICE worth of execution time.
  86. */
  87. void spu_set_timeslice(struct spu_context *ctx)
  88. {
  89. if (ctx->prio < NORMAL_PRIO)
  90. ctx->time_slice = SCALE_PRIO(DEF_SPU_TIMESLICE * 4, ctx->prio);
  91. else
  92. ctx->time_slice = SCALE_PRIO(DEF_SPU_TIMESLICE, ctx->prio);
  93. }
  94. /*
  95. * Update scheduling information from the owning thread.
  96. */
  97. void __spu_update_sched_info(struct spu_context *ctx)
  98. {
  99. /*
  100. * 32-Bit assignment are atomic on powerpc, and we don't care about
  101. * memory ordering here because retriving the controlling thread is
  102. * per defintion racy.
  103. */
  104. ctx->tid = current->pid;
  105. /*
  106. * We do our own priority calculations, so we normally want
  107. * ->static_prio to start with. Unfortunately thies field
  108. * contains junk for threads with a realtime scheduling
  109. * policy so we have to look at ->prio in this case.
  110. */
  111. if (rt_prio(current->prio))
  112. ctx->prio = current->prio;
  113. else
  114. ctx->prio = current->static_prio;
  115. ctx->policy = current->policy;
  116. /*
  117. * A lot of places that don't hold active_mutex poke into
  118. * cpus_allowed, including grab_runnable_context which
  119. * already holds the runq_lock. So abuse runq_lock
  120. * to protect this field aswell.
  121. */
  122. spin_lock(&spu_prio->runq_lock);
  123. ctx->cpus_allowed = current->cpus_allowed;
  124. spin_unlock(&spu_prio->runq_lock);
  125. }
  126. void spu_update_sched_info(struct spu_context *ctx)
  127. {
  128. int node = ctx->spu->node;
  129. mutex_lock(&spu_prio->active_mutex[node]);
  130. __spu_update_sched_info(ctx);
  131. mutex_unlock(&spu_prio->active_mutex[node]);
  132. }
  133. static int __node_allowed(struct spu_context *ctx, int node)
  134. {
  135. if (nr_cpus_node(node)) {
  136. cpumask_t mask = node_to_cpumask(node);
  137. if (cpus_intersects(mask, ctx->cpus_allowed))
  138. return 1;
  139. }
  140. return 0;
  141. }
  142. static int node_allowed(struct spu_context *ctx, int node)
  143. {
  144. int rval;
  145. spin_lock(&spu_prio->runq_lock);
  146. rval = __node_allowed(ctx, node);
  147. spin_unlock(&spu_prio->runq_lock);
  148. return rval;
  149. }
  150. /**
  151. * spu_add_to_active_list - add spu to active list
  152. * @spu: spu to add to the active list
  153. */
  154. static void spu_add_to_active_list(struct spu *spu)
  155. {
  156. int node = spu->node;
  157. mutex_lock(&spu_prio->active_mutex[node]);
  158. spu_prio->nr_active[node]++;
  159. list_add_tail(&spu->list, &spu_prio->active_list[node]);
  160. mutex_unlock(&spu_prio->active_mutex[node]);
  161. }
  162. static void __spu_remove_from_active_list(struct spu *spu)
  163. {
  164. list_del_init(&spu->list);
  165. spu_prio->nr_active[spu->node]--;
  166. }
  167. /**
  168. * spu_remove_from_active_list - remove spu from active list
  169. * @spu: spu to remove from the active list
  170. */
  171. static void spu_remove_from_active_list(struct spu *spu)
  172. {
  173. int node = spu->node;
  174. mutex_lock(&spu_prio->active_mutex[node]);
  175. __spu_remove_from_active_list(spu);
  176. mutex_unlock(&spu_prio->active_mutex[node]);
  177. }
  178. static BLOCKING_NOTIFIER_HEAD(spu_switch_notifier);
  179. static void spu_switch_notify(struct spu *spu, struct spu_context *ctx)
  180. {
  181. blocking_notifier_call_chain(&spu_switch_notifier,
  182. ctx ? ctx->object_id : 0, spu);
  183. }
  184. int spu_switch_event_register(struct notifier_block * n)
  185. {
  186. return blocking_notifier_chain_register(&spu_switch_notifier, n);
  187. }
  188. int spu_switch_event_unregister(struct notifier_block * n)
  189. {
  190. return blocking_notifier_chain_unregister(&spu_switch_notifier, n);
  191. }
  192. /**
  193. * spu_bind_context - bind spu context to physical spu
  194. * @spu: physical spu to bind to
  195. * @ctx: context to bind
  196. */
  197. static void spu_bind_context(struct spu *spu, struct spu_context *ctx)
  198. {
  199. pr_debug("%s: pid=%d SPU=%d NODE=%d\n", __FUNCTION__, current->pid,
  200. spu->number, spu->node);
  201. spuctx_switch_state(ctx, SPU_UTIL_SYSTEM);
  202. if (ctx->flags & SPU_CREATE_NOSCHED)
  203. atomic_inc(&cbe_spu_info[spu->node].reserved_spus);
  204. ctx->stats.slb_flt_base = spu->stats.slb_flt;
  205. ctx->stats.class2_intr_base = spu->stats.class2_intr;
  206. spu->ctx = ctx;
  207. spu->flags = 0;
  208. ctx->spu = spu;
  209. ctx->ops = &spu_hw_ops;
  210. spu->pid = current->pid;
  211. spu_associate_mm(spu, ctx->owner);
  212. spu->ibox_callback = spufs_ibox_callback;
  213. spu->wbox_callback = spufs_wbox_callback;
  214. spu->stop_callback = spufs_stop_callback;
  215. spu->mfc_callback = spufs_mfc_callback;
  216. spu->dma_callback = spufs_dma_callback;
  217. mb();
  218. spu_unmap_mappings(ctx);
  219. spu_restore(&ctx->csa, spu);
  220. spu->timestamp = jiffies;
  221. spu_cpu_affinity_set(spu, raw_smp_processor_id());
  222. spu_switch_notify(spu, ctx);
  223. ctx->state = SPU_STATE_RUNNABLE;
  224. spuctx_switch_state(ctx, SPU_UTIL_IDLE_LOADED);
  225. }
  226. /**
  227. * spu_unbind_context - unbind spu context from physical spu
  228. * @spu: physical spu to unbind from
  229. * @ctx: context to unbind
  230. */
  231. static void spu_unbind_context(struct spu *spu, struct spu_context *ctx)
  232. {
  233. pr_debug("%s: unbind pid=%d SPU=%d NODE=%d\n", __FUNCTION__,
  234. spu->pid, spu->number, spu->node);
  235. spuctx_switch_state(ctx, SPU_UTIL_SYSTEM);
  236. if (spu->ctx->flags & SPU_CREATE_NOSCHED)
  237. atomic_dec(&cbe_spu_info[spu->node].reserved_spus);
  238. spu_switch_notify(spu, NULL);
  239. spu_unmap_mappings(ctx);
  240. spu_save(&ctx->csa, spu);
  241. spu->timestamp = jiffies;
  242. ctx->state = SPU_STATE_SAVED;
  243. spu->ibox_callback = NULL;
  244. spu->wbox_callback = NULL;
  245. spu->stop_callback = NULL;
  246. spu->mfc_callback = NULL;
  247. spu->dma_callback = NULL;
  248. spu_associate_mm(spu, NULL);
  249. spu->pid = 0;
  250. ctx->ops = &spu_backing_ops;
  251. spu->flags = 0;
  252. spu->ctx = NULL;
  253. ctx->stats.slb_flt +=
  254. (spu->stats.slb_flt - ctx->stats.slb_flt_base);
  255. ctx->stats.class2_intr +=
  256. (spu->stats.class2_intr - ctx->stats.class2_intr_base);
  257. /* This maps the underlying spu state to idle */
  258. spuctx_switch_state(ctx, SPU_UTIL_IDLE_LOADED);
  259. ctx->spu = NULL;
  260. }
  261. /**
  262. * spu_add_to_rq - add a context to the runqueue
  263. * @ctx: context to add
  264. */
  265. static void __spu_add_to_rq(struct spu_context *ctx)
  266. {
  267. /*
  268. * Unfortunately this code path can be called from multiple threads
  269. * on behalf of a single context due to the way the problem state
  270. * mmap support works.
  271. *
  272. * Fortunately we need to wake up all these threads at the same time
  273. * and can simply skip the runqueue addition for every but the first
  274. * thread getting into this codepath.
  275. *
  276. * It's still quite hacky, and long-term we should proxy all other
  277. * threads through the owner thread so that spu_run is in control
  278. * of all the scheduling activity for a given context.
  279. */
  280. if (list_empty(&ctx->rq)) {
  281. list_add_tail(&ctx->rq, &spu_prio->runq[ctx->prio]);
  282. set_bit(ctx->prio, spu_prio->bitmap);
  283. if (!spu_prio->nr_waiting++)
  284. __mod_timer(&spusched_timer, jiffies + SPUSCHED_TICK);
  285. }
  286. }
  287. static void __spu_del_from_rq(struct spu_context *ctx)
  288. {
  289. int prio = ctx->prio;
  290. if (!list_empty(&ctx->rq)) {
  291. if (!--spu_prio->nr_waiting)
  292. del_timer(&spusched_timer);
  293. list_del_init(&ctx->rq);
  294. if (list_empty(&spu_prio->runq[prio]))
  295. clear_bit(prio, spu_prio->bitmap);
  296. }
  297. }
  298. static void spu_prio_wait(struct spu_context *ctx)
  299. {
  300. DEFINE_WAIT(wait);
  301. spin_lock(&spu_prio->runq_lock);
  302. prepare_to_wait_exclusive(&ctx->stop_wq, &wait, TASK_INTERRUPTIBLE);
  303. if (!signal_pending(current)) {
  304. __spu_add_to_rq(ctx);
  305. spin_unlock(&spu_prio->runq_lock);
  306. mutex_unlock(&ctx->state_mutex);
  307. schedule();
  308. mutex_lock(&ctx->state_mutex);
  309. spin_lock(&spu_prio->runq_lock);
  310. __spu_del_from_rq(ctx);
  311. }
  312. spin_unlock(&spu_prio->runq_lock);
  313. __set_current_state(TASK_RUNNING);
  314. remove_wait_queue(&ctx->stop_wq, &wait);
  315. }
  316. static struct spu *spu_get_idle(struct spu_context *ctx)
  317. {
  318. struct spu *spu = NULL;
  319. int node = cpu_to_node(raw_smp_processor_id());
  320. int n;
  321. for (n = 0; n < MAX_NUMNODES; n++, node++) {
  322. node = (node < MAX_NUMNODES) ? node : 0;
  323. if (!node_allowed(ctx, node))
  324. continue;
  325. spu = spu_alloc_node(node);
  326. if (spu)
  327. break;
  328. }
  329. return spu;
  330. }
  331. /**
  332. * find_victim - find a lower priority context to preempt
  333. * @ctx: canidate context for running
  334. *
  335. * Returns the freed physical spu to run the new context on.
  336. */
  337. static struct spu *find_victim(struct spu_context *ctx)
  338. {
  339. struct spu_context *victim = NULL;
  340. struct spu *spu;
  341. int node, n;
  342. /*
  343. * Look for a possible preemption candidate on the local node first.
  344. * If there is no candidate look at the other nodes. This isn't
  345. * exactly fair, but so far the whole spu schedule tries to keep
  346. * a strong node affinity. We might want to fine-tune this in
  347. * the future.
  348. */
  349. restart:
  350. node = cpu_to_node(raw_smp_processor_id());
  351. for (n = 0; n < MAX_NUMNODES; n++, node++) {
  352. node = (node < MAX_NUMNODES) ? node : 0;
  353. if (!node_allowed(ctx, node))
  354. continue;
  355. mutex_lock(&spu_prio->active_mutex[node]);
  356. list_for_each_entry(spu, &spu_prio->active_list[node], list) {
  357. struct spu_context *tmp = spu->ctx;
  358. if (tmp->prio > ctx->prio &&
  359. (!victim || tmp->prio > victim->prio))
  360. victim = spu->ctx;
  361. }
  362. mutex_unlock(&spu_prio->active_mutex[node]);
  363. if (victim) {
  364. /*
  365. * This nests ctx->state_mutex, but we always lock
  366. * higher priority contexts before lower priority
  367. * ones, so this is safe until we introduce
  368. * priority inheritance schemes.
  369. */
  370. if (!mutex_trylock(&victim->state_mutex)) {
  371. victim = NULL;
  372. goto restart;
  373. }
  374. spu = victim->spu;
  375. if (!spu) {
  376. /*
  377. * This race can happen because we've dropped
  378. * the active list mutex. No a problem, just
  379. * restart the search.
  380. */
  381. mutex_unlock(&victim->state_mutex);
  382. victim = NULL;
  383. goto restart;
  384. }
  385. spu_remove_from_active_list(spu);
  386. spu_unbind_context(spu, victim);
  387. victim->stats.invol_ctx_switch++;
  388. spu->stats.invol_ctx_switch++;
  389. mutex_unlock(&victim->state_mutex);
  390. /*
  391. * We need to break out of the wait loop in spu_run
  392. * manually to ensure this context gets put on the
  393. * runqueue again ASAP.
  394. */
  395. wake_up(&victim->stop_wq);
  396. return spu;
  397. }
  398. }
  399. return NULL;
  400. }
  401. /**
  402. * spu_activate - find a free spu for a context and execute it
  403. * @ctx: spu context to schedule
  404. * @flags: flags (currently ignored)
  405. *
  406. * Tries to find a free spu to run @ctx. If no free spu is available
  407. * add the context to the runqueue so it gets woken up once an spu
  408. * is available.
  409. */
  410. int spu_activate(struct spu_context *ctx, unsigned long flags)
  411. {
  412. do {
  413. struct spu *spu;
  414. /*
  415. * If there are multiple threads waiting for a single context
  416. * only one actually binds the context while the others will
  417. * only be able to acquire the state_mutex once the context
  418. * already is in runnable state.
  419. */
  420. if (ctx->spu)
  421. return 0;
  422. spu = spu_get_idle(ctx);
  423. /*
  424. * If this is a realtime thread we try to get it running by
  425. * preempting a lower priority thread.
  426. */
  427. if (!spu && rt_prio(ctx->prio))
  428. spu = find_victim(ctx);
  429. if (spu) {
  430. spu_bind_context(spu, ctx);
  431. spu_add_to_active_list(spu);
  432. return 0;
  433. }
  434. spu_prio_wait(ctx);
  435. } while (!signal_pending(current));
  436. return -ERESTARTSYS;
  437. }
  438. /**
  439. * grab_runnable_context - try to find a runnable context
  440. *
  441. * Remove the highest priority context on the runqueue and return it
  442. * to the caller. Returns %NULL if no runnable context was found.
  443. */
  444. static struct spu_context *grab_runnable_context(int prio, int node)
  445. {
  446. struct spu_context *ctx;
  447. int best;
  448. spin_lock(&spu_prio->runq_lock);
  449. best = find_first_bit(spu_prio->bitmap, prio);
  450. while (best < prio) {
  451. struct list_head *rq = &spu_prio->runq[best];
  452. list_for_each_entry(ctx, rq, rq) {
  453. /* XXX(hch): check for affinity here aswell */
  454. if (__node_allowed(ctx, node)) {
  455. __spu_del_from_rq(ctx);
  456. goto found;
  457. }
  458. }
  459. best++;
  460. }
  461. ctx = NULL;
  462. found:
  463. spin_unlock(&spu_prio->runq_lock);
  464. return ctx;
  465. }
  466. static int __spu_deactivate(struct spu_context *ctx, int force, int max_prio)
  467. {
  468. struct spu *spu = ctx->spu;
  469. struct spu_context *new = NULL;
  470. if (spu) {
  471. new = grab_runnable_context(max_prio, spu->node);
  472. if (new || force) {
  473. spu_remove_from_active_list(spu);
  474. spu_unbind_context(spu, ctx);
  475. ctx->stats.vol_ctx_switch++;
  476. spu->stats.vol_ctx_switch++;
  477. spu_free(spu);
  478. if (new)
  479. wake_up(&new->stop_wq);
  480. }
  481. }
  482. return new != NULL;
  483. }
  484. /**
  485. * spu_deactivate - unbind a context from it's physical spu
  486. * @ctx: spu context to unbind
  487. *
  488. * Unbind @ctx from the physical spu it is running on and schedule
  489. * the highest priority context to run on the freed physical spu.
  490. */
  491. void spu_deactivate(struct spu_context *ctx)
  492. {
  493. __spu_deactivate(ctx, 1, MAX_PRIO);
  494. }
  495. /**
  496. * spu_yield - yield a physical spu if others are waiting
  497. * @ctx: spu context to yield
  498. *
  499. * Check if there is a higher priority context waiting and if yes
  500. * unbind @ctx from the physical spu and schedule the highest
  501. * priority context to run on the freed physical spu instead.
  502. */
  503. void spu_yield(struct spu_context *ctx)
  504. {
  505. if (!(ctx->flags & SPU_CREATE_NOSCHED)) {
  506. mutex_lock(&ctx->state_mutex);
  507. __spu_deactivate(ctx, 0, MAX_PRIO);
  508. mutex_unlock(&ctx->state_mutex);
  509. }
  510. }
  511. static void spusched_tick(struct spu_context *ctx)
  512. {
  513. if (ctx->flags & SPU_CREATE_NOSCHED)
  514. return;
  515. if (ctx->policy == SCHED_FIFO)
  516. return;
  517. if (--ctx->time_slice)
  518. return;
  519. /*
  520. * Unfortunately active_mutex ranks outside of state_mutex, so
  521. * we have to trylock here. If we fail give the context another
  522. * tick and try again.
  523. */
  524. if (mutex_trylock(&ctx->state_mutex)) {
  525. struct spu *spu = ctx->spu;
  526. struct spu_context *new;
  527. new = grab_runnable_context(ctx->prio + 1, spu->node);
  528. if (new) {
  529. __spu_remove_from_active_list(spu);
  530. spu_unbind_context(spu, ctx);
  531. ctx->stats.invol_ctx_switch++;
  532. spu->stats.invol_ctx_switch++;
  533. spu_free(spu);
  534. wake_up(&new->stop_wq);
  535. /*
  536. * We need to break out of the wait loop in
  537. * spu_run manually to ensure this context
  538. * gets put on the runqueue again ASAP.
  539. */
  540. wake_up(&ctx->stop_wq);
  541. }
  542. spu_set_timeslice(ctx);
  543. mutex_unlock(&ctx->state_mutex);
  544. } else {
  545. ctx->time_slice++;
  546. }
  547. }
  548. /**
  549. * count_active_contexts - count nr of active tasks
  550. *
  551. * Return the number of tasks currently running or waiting to run.
  552. *
  553. * Note that we don't take runq_lock / active_mutex here. Reading
  554. * a single 32bit value is atomic on powerpc, and we don't care
  555. * about memory ordering issues here.
  556. */
  557. static unsigned long count_active_contexts(void)
  558. {
  559. int nr_active = 0, node;
  560. for (node = 0; node < MAX_NUMNODES; node++)
  561. nr_active += spu_prio->nr_active[node];
  562. nr_active += spu_prio->nr_waiting;
  563. return nr_active;
  564. }
  565. /**
  566. * spu_calc_load - given tick count, update the avenrun load estimates.
  567. * @tick: tick count
  568. *
  569. * No locking against reading these values from userspace, as for
  570. * the CPU loadavg code.
  571. */
  572. static void spu_calc_load(unsigned long ticks)
  573. {
  574. unsigned long active_tasks; /* fixed-point */
  575. static int count = LOAD_FREQ;
  576. count -= ticks;
  577. if (unlikely(count < 0)) {
  578. active_tasks = count_active_contexts() * FIXED_1;
  579. do {
  580. CALC_LOAD(spu_avenrun[0], EXP_1, active_tasks);
  581. CALC_LOAD(spu_avenrun[1], EXP_5, active_tasks);
  582. CALC_LOAD(spu_avenrun[2], EXP_15, active_tasks);
  583. count += LOAD_FREQ;
  584. } while (count < 0);
  585. }
  586. }
  587. static void spusched_wake(unsigned long data)
  588. {
  589. mod_timer(&spusched_timer, jiffies + SPUSCHED_TICK);
  590. wake_up_process(spusched_task);
  591. spu_calc_load(SPUSCHED_TICK);
  592. }
  593. static int spusched_thread(void *unused)
  594. {
  595. struct spu *spu, *next;
  596. int node;
  597. while (!kthread_should_stop()) {
  598. set_current_state(TASK_INTERRUPTIBLE);
  599. schedule();
  600. for (node = 0; node < MAX_NUMNODES; node++) {
  601. mutex_lock(&spu_prio->active_mutex[node]);
  602. list_for_each_entry_safe(spu, next,
  603. &spu_prio->active_list[node],
  604. list)
  605. spusched_tick(spu->ctx);
  606. mutex_unlock(&spu_prio->active_mutex[node]);
  607. }
  608. }
  609. return 0;
  610. }
  611. #define LOAD_INT(x) ((x) >> FSHIFT)
  612. #define LOAD_FRAC(x) LOAD_INT(((x) & (FIXED_1-1)) * 100)
  613. static int show_spu_loadavg(struct seq_file *s, void *private)
  614. {
  615. int a, b, c;
  616. a = spu_avenrun[0] + (FIXED_1/200);
  617. b = spu_avenrun[1] + (FIXED_1/200);
  618. c = spu_avenrun[2] + (FIXED_1/200);
  619. /*
  620. * Note that last_pid doesn't really make much sense for the
  621. * SPU loadavg (it even seems very odd on the CPU side..),
  622. * but we include it here to have a 100% compatible interface.
  623. */
  624. seq_printf(s, "%d.%02d %d.%02d %d.%02d %ld/%d %d\n",
  625. LOAD_INT(a), LOAD_FRAC(a),
  626. LOAD_INT(b), LOAD_FRAC(b),
  627. LOAD_INT(c), LOAD_FRAC(c),
  628. count_active_contexts(),
  629. atomic_read(&nr_spu_contexts),
  630. current->nsproxy->pid_ns->last_pid);
  631. return 0;
  632. }
  633. static int spu_loadavg_open(struct inode *inode, struct file *file)
  634. {
  635. return single_open(file, show_spu_loadavg, NULL);
  636. }
  637. static const struct file_operations spu_loadavg_fops = {
  638. .open = spu_loadavg_open,
  639. .read = seq_read,
  640. .llseek = seq_lseek,
  641. .release = single_release,
  642. };
  643. int __init spu_sched_init(void)
  644. {
  645. struct proc_dir_entry *entry;
  646. int err = -ENOMEM, i;
  647. spu_prio = kzalloc(sizeof(struct spu_prio_array), GFP_KERNEL);
  648. if (!spu_prio)
  649. goto out;
  650. for (i = 0; i < MAX_PRIO; i++) {
  651. INIT_LIST_HEAD(&spu_prio->runq[i]);
  652. __clear_bit(i, spu_prio->bitmap);
  653. }
  654. for (i = 0; i < MAX_NUMNODES; i++) {
  655. mutex_init(&spu_prio->active_mutex[i]);
  656. INIT_LIST_HEAD(&spu_prio->active_list[i]);
  657. }
  658. spin_lock_init(&spu_prio->runq_lock);
  659. setup_timer(&spusched_timer, spusched_wake, 0);
  660. spusched_task = kthread_run(spusched_thread, NULL, "spusched");
  661. if (IS_ERR(spusched_task)) {
  662. err = PTR_ERR(spusched_task);
  663. goto out_free_spu_prio;
  664. }
  665. entry = create_proc_entry("spu_loadavg", 0, NULL);
  666. if (!entry)
  667. goto out_stop_kthread;
  668. entry->proc_fops = &spu_loadavg_fops;
  669. pr_debug("spusched: tick: %d, min ticks: %d, default ticks: %d\n",
  670. SPUSCHED_TICK, MIN_SPU_TIMESLICE, DEF_SPU_TIMESLICE);
  671. return 0;
  672. out_stop_kthread:
  673. kthread_stop(spusched_task);
  674. out_free_spu_prio:
  675. kfree(spu_prio);
  676. out:
  677. return err;
  678. }
  679. void spu_sched_exit(void)
  680. {
  681. struct spu *spu, *tmp;
  682. int node;
  683. remove_proc_entry("spu_loadavg", NULL);
  684. del_timer_sync(&spusched_timer);
  685. kthread_stop(spusched_task);
  686. for (node = 0; node < MAX_NUMNODES; node++) {
  687. mutex_lock(&spu_prio->active_mutex[node]);
  688. list_for_each_entry_safe(spu, tmp, &spu_prio->active_list[node],
  689. list) {
  690. list_del_init(&spu->list);
  691. spu_free(spu);
  692. }
  693. mutex_unlock(&spu_prio->active_mutex[node]);
  694. }
  695. kfree(spu_prio);
  696. }