1 /* ---------------------------------------------------------------------------
3 * (c) The GHC Team, 2001
7 * The notion of a capability is used when operating in multi-threaded
8 * environments (which the SMP and Threads builds of the RTS do), to
9 * hold all the state an OS thread/task needs to run Haskell code:
10 * its STG registers, a pointer to its TSO, a nursery etc. During
11 * STG execution, a pointer to the capabilitity is kept in a
14 * Only in an SMP build will there be multiple capabilities, the threaded
15 * RTS and other non-threaded builds, there is one global capability,
16 * namely MainRegTable.
19 * --------------------------------------------------------------------------*/
20 #include "PosixSource.h"
23 #include "Capability.h"
28 initCapability( Capability *cap )
30 cap->f.stgChk0 = (F_)__stg_chk_0;
31 cap->f.stgChk1 = (F_)__stg_chk_1;
32 cap->f.stgGCEnter1 = (F_)__stg_gc_enter_1;
33 cap->f.stgUpdatePAP = (F_)__stg_update_PAP;
37 static void initCapabilities_(nat n);
46 initCapabilities_(RtsFlags.ParFlags.nNodes);
48 initCapability(&MainCapability);
54 /* Free capability list.
55 * Locks required: sched_mutex.
58 static Capability *free_capabilities; /* Available capabilities for running threads */
61 void grabCapability(Capability** cap)
64 *cap = &MainCapability;
66 *cap = free_capabilities;
67 free_capabilities = (*cap)->link;
68 rts_n_free_capabilities--;
72 void releaseCapability(Capability** cap)
75 (*cap)->link = free_capabilities;
76 free_capabilities = *cap;
77 rts_n_free_capabilities++;
83 /* Allocate 'n' capabilities */
85 initCapabilities_(nat n)
88 Capability *cap, *prev;
91 for (i = 0; i < n; i++) {
92 cap = stgMallocBytes(sizeof(Capability), "initCapabilities");
97 free_capabilities = cap;
98 rts_n_free_capabilities = n;
99 IF_DEBUG(scheduler,fprintf(stderr,"scheduler: Allocated %d capabilities\n", n_free_capabilities););