]>
Commit | Line | Data |
---|---|---|
748086b7 | 1 | /* Copyright (C) 2007, 2009 Free Software Foundation, Inc. |
1d0bd356 DS |
2 | Contributed by Danny Smith <dannysmith@users.sourceforge.net> |
3 | ||
4 | This file is part of the GNU OpenMP Library (libgomp). | |
5 | ||
6 | Libgomp is free software; you can redistribute it and/or modify it | |
748086b7 JJ |
7 | under the terms of the GNU General Public License as published by |
8 | the Free Software Foundation; either version 3, or (at your option) | |
9 | any later version. | |
1d0bd356 DS |
10 | |
11 | Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY | |
12 | WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS | |
748086b7 | 13 | FOR A PARTICULAR PURPOSE. See the GNU General Public License for |
1d0bd356 DS |
14 | more details. |
15 | ||
748086b7 JJ |
16 | Under Section 7 of GPL version 3, you are granted additional |
17 | permissions described in the GCC Runtime Library Exception, version | |
18 | 3.1, as published by the Free Software Foundation. | |
19 | ||
20 | You should have received a copy of the GNU General Public License and | |
21 | a copy of the GCC Runtime Library Exception along with this program; | |
22 | see the files COPYING3 and COPYING.RUNTIME respectively. If not, see | |
23 | <http://www.gnu.org/licenses/>. */ | |
1d0bd356 DS |
24 | |
25 | /* This file contains system specific routines related to counting | |
26 | online processors and dynamic load balancing. It is expected that | |
27 | a system may well want to write special versions of each of these. | |
28 | ||
29 | The following implementation uses win32 API routines. */ | |
30 | ||
31 | #include "libgomp.h" | |
32 | #include <windows.h> | |
33 | ||
34 | /* Count the CPU's currently available to this process. */ | |
a68ab351 | 35 | static unsigned int |
1d0bd356 DS |
36 | count_avail_process_cpus () |
37 | { | |
38 | DWORD_PTR process_cpus; | |
39 | DWORD_PTR system_cpus; | |
40 | ||
41 | if (GetProcessAffinityMask (GetCurrentProcess (), | |
42 | &process_cpus, &system_cpus)) | |
43 | { | |
44 | unsigned int count; | |
45 | for (count = 0; process_cpus != 0; process_cpus >>= 1) | |
46 | if (process_cpus & 1) | |
47 | count++; | |
48 | return count; | |
49 | } | |
50 | return 1; | |
51 | } | |
52 | ||
53 | /* At startup, determine the default number of threads. It would seem | |
54 | this should be related to the number of cpus available to the process. */ | |
55 | ||
56 | void | |
57 | gomp_init_num_threads (void) | |
58 | { | |
a68ab351 | 59 | gomp_global_icv.nthreads_var = count_avail_process_cpus (); |
1d0bd356 DS |
60 | } |
61 | ||
62 | /* When OMP_DYNAMIC is set, at thread launch determine the number of | |
63 | threads we should spawn for this team. FIXME: How do we adjust for | |
64 | load average on MS Windows? */ | |
65 | ||
66 | unsigned | |
67 | gomp_dynamic_max_threads (void) | |
68 | { | |
a68ab351 JJ |
69 | unsigned int n_onln = count_avail_process_cpus (); |
70 | unsigned int nthreads_var = gomp_icv (false)->nthreads_var; | |
71 | return n_onln > nthreads_var ? nthreads_var : n_onln; | |
1d0bd356 DS |
72 | } |
73 | ||
74 | int | |
75 | omp_get_num_procs (void) | |
76 | { | |
77 | return count_avail_process_cpus (); | |
78 | } | |
79 | ||
80 | ialias (omp_get_num_procs) |