diff options
| author | Manuel M T Chakravarty <chak@cse.unsw.edu.au> | 2009-03-20 07:13:47 +0000 |
|---|---|---|
| committer | Manuel M T Chakravarty <chak@cse.unsw.edu.au> | 2009-03-20 07:13:47 +0000 |
| commit | 77998ef4dc3277c3037460125d8fc8a7dc8ead75 (patch) | |
| tree | d7063a001ff18600af38f35ae4fe6ae152102cd4 /rts/posix | |
| parent | c14119ce2639d4e4c868f3d1c9bb59bd3a2dd281 (diff) | |
| download | haskell-77998ef4dc3277c3037460125d8fc8a7dc8ead75.tar.gz | |
Implement setThreadAffinity for Mac OS X (from 10.5)
Diffstat (limited to 'rts/posix')
| -rw-r--r-- | rts/posix/OSThreads.c | 29 |
1 files changed, 27 insertions, 2 deletions
diff --git a/rts/posix/OSThreads.c b/rts/posix/OSThreads.c index 1b85fd41f8..9bcd56f8ad 100644 --- a/rts/posix/OSThreads.c +++ b/rts/posix/OSThreads.c @@ -35,6 +35,10 @@ #include <unistd.h> #endif +#if defined(darwin_HOST_OS) +#include <mach/mach.h> +#endif + /* * This (allegedly) OS threads independent layer was initially * abstracted away from code that used Pthreads, so the functions @@ -206,13 +210,13 @@ getNumberOfProcessors (void) return nproc; } +#if defined(HAVE_SCHED_H) && defined(HAVE_SCHED_SETAFFINITY) // Schedules the thread to run on CPU n of m. m may be less than the // number of physical CPUs, in which case, the thread will be allowed // to run on CPU n, n+m, n+2m etc. void setThreadAffinity (nat n, nat m) { -#if defined(HAVE_SCHED_H) && defined(HAVE_SCHED_SETAFFINITY) nat nproc; cpu_set_t cs; nat i; @@ -223,9 +227,30 @@ setThreadAffinity (nat n, nat m) CPU_SET(n, &cs); } sched_setaffinity(0, sizeof(cpu_set_t), &cs); -#endif } +#elif defined(darwin_HOST_OS) && defined(THREAD_AFFINITY_POLICY) +// Schedules the current thread in the affinity set identified by tag n. +void +setThreadAffinity (nat n, nat m GNUC3_ATTRIBUTE(__unused__)) +{ + thread_affinity_policy_data_t policy; + + policy.affinity_tag = n; + thread_policy_set(mach_thread_self(), + THREAD_AFFINITY_POLICY, + (thread_policy_t) &policy, + THREAD_AFFINITY_POLICY_COUNT); +} + +#else +void +setThreadAffinity (nat n GNUC3_ATTRIBUTE(__unused__), + nat m GNUC3_ATTRIBUTE(__unused__)) +{ +} +#endif + #else /* !defined(THREADED_RTS) */ int |