Commit | Line | Data |
---|---|---|
9f1528a1 AB |
1 | # =========================================================================== |
2 | # https://www.gnu.org/software/autoconf-archive/ax_count_cpus.html | |
3 | # =========================================================================== | |
4 | # | |
5 | # SYNOPSIS | |
6 | # | |
7 | # AX_COUNT_CPUS([ACTION-IF-DETECTED],[ACTION-IF-NOT-DETECTED]) | |
8 | # | |
9 | # DESCRIPTION | |
10 | # | |
11 | # Attempt to count the number of logical processor cores (including | |
12 | # virtual and HT cores) currently available to use on the machine and | |
13 | # place detected value in CPU_COUNT variable. | |
14 | # | |
15 | # On successful detection, ACTION-IF-DETECTED is executed if present. If | |
16 | # the detection fails, then ACTION-IF-NOT-DETECTED is triggered. The | |
17 | # default ACTION-IF-NOT-DETECTED is to set CPU_COUNT to 1. | |
18 | # | |
19 | # LICENSE | |
20 | # | |
21 | # Copyright (c) 2014,2016 Karlson2k (Evgeny Grin) <k2k@narod.ru> | |
22 | # Copyright (c) 2012 Brian Aker <brian@tangent.org> | |
23 | # Copyright (c) 2008 Michael Paul Bailey <jinxidoru@byu.net> | |
24 | # Copyright (c) 2008 Christophe Tournayre <turn3r@users.sourceforge.net> | |
25 | # | |
26 | # Copying and distribution of this file, with or without modification, are | |
27 | # permitted in any medium without royalty provided the copyright notice | |
28 | # and this notice are preserved. This file is offered as-is, without any | |
29 | # warranty. | |
30 | ||
31 | #serial 22 | |
32 | ||
33 | AC_DEFUN([AX_COUNT_CPUS],[dnl | |
34 | AC_REQUIRE([AC_CANONICAL_HOST])dnl | |
35 | AC_REQUIRE([AC_PROG_EGREP])dnl | |
36 | AC_MSG_CHECKING([the number of available CPUs]) | |
37 | CPU_COUNT="0" | |
38 | ||
39 | # Try generic methods | |
40 | ||
41 | # 'getconf' is POSIX utility, but '_NPROCESSORS_ONLN' and | |
42 | # 'NPROCESSORS_ONLN' are platform-specific | |
43 | command -v getconf >/dev/null 2>&1 && \ | |
44 | CPU_COUNT=`getconf _NPROCESSORS_ONLN 2>/dev/null || getconf NPROCESSORS_ONLN 2>/dev/null` || CPU_COUNT="0" | |
45 | AS_IF([[test "$CPU_COUNT" -gt "0" 2>/dev/null || ! command -v nproc >/dev/null 2>&1]],[[: # empty]],[dnl | |
46 | # 'nproc' is part of GNU Coreutils and is widely available | |
47 | CPU_COUNT=`OMP_NUM_THREADS='' nproc 2>/dev/null` || CPU_COUNT=`nproc 2>/dev/null` || CPU_COUNT="0" | |
48 | ])dnl | |
49 | ||
50 | AS_IF([[test "$CPU_COUNT" -gt "0" 2>/dev/null]],[[: # empty]],[dnl | |
51 | # Try platform-specific preferred methods | |
52 | AS_CASE([[$host_os]],dnl | |
53 | [[*linux*]],[[CPU_COUNT=`lscpu -p 2>/dev/null | $EGREP -e '^@<:@0-9@:>@+,' -c` || CPU_COUNT="0"]],dnl | |
54 | [[*darwin*]],[[CPU_COUNT=`sysctl -n hw.logicalcpu 2>/dev/null` || CPU_COUNT="0"]],dnl | |
55 | [[freebsd*]],[[command -v sysctl >/dev/null 2>&1 && CPU_COUNT=`sysctl -n kern.smp.cpus 2>/dev/null` || CPU_COUNT="0"]],dnl | |
56 | [[netbsd*]], [[command -v sysctl >/dev/null 2>&1 && CPU_COUNT=`sysctl -n hw.ncpuonline 2>/dev/null` || CPU_COUNT="0"]],dnl | |
57 | [[solaris*]],[[command -v psrinfo >/dev/null 2>&1 && CPU_COUNT=`psrinfo 2>/dev/null | $EGREP -e '^@<:@0-9@:>@.*on-line' -c 2>/dev/null` || CPU_COUNT="0"]],dnl | |
58 | [[mingw*]],[[CPU_COUNT=`ls -qpU1 /proc/registry/HKEY_LOCAL_MACHINE/HARDWARE/DESCRIPTION/System/CentralProcessor/ 2>/dev/null | $EGREP -e '^@<:@0-9@:>@+/' -c` || CPU_COUNT="0"]],dnl | |
59 | [[msys*]],[[CPU_COUNT=`ls -qpU1 /proc/registry/HKEY_LOCAL_MACHINE/HARDWARE/DESCRIPTION/System/CentralProcessor/ 2>/dev/null | $EGREP -e '^@<:@0-9@:>@+/' -c` || CPU_COUNT="0"]],dnl | |
60 | [[cygwin*]],[[CPU_COUNT=`ls -qpU1 /proc/registry/HKEY_LOCAL_MACHINE/HARDWARE/DESCRIPTION/System/CentralProcessor/ 2>/dev/null | $EGREP -e '^@<:@0-9@:>@+/' -c` || CPU_COUNT="0"]]dnl | |
61 | )dnl | |
62 | ])dnl | |
63 | ||
64 | AS_IF([[test "$CPU_COUNT" -gt "0" 2>/dev/null || ! command -v sysctl >/dev/null 2>&1]],[[: # empty]],[dnl | |
65 | # Try less preferred generic method | |
66 | # 'hw.ncpu' exist on many platforms, but not on GNU/Linux | |
67 | CPU_COUNT=`sysctl -n hw.ncpu 2>/dev/null` || CPU_COUNT="0" | |
68 | ])dnl | |
69 | ||
70 | AS_IF([[test "$CPU_COUNT" -gt "0" 2>/dev/null]],[[: # empty]],[dnl | |
71 | # Try platform-specific fallback methods | |
72 | # They can be less accurate and slower then preferred methods | |
73 | AS_CASE([[$host_os]],dnl | |
74 | [[*linux*]],[[CPU_COUNT=`$EGREP -e '^processor' -c /proc/cpuinfo 2>/dev/null` || CPU_COUNT="0"]],dnl | |
75 | [[*darwin*]],[[CPU_COUNT=`system_profiler SPHardwareDataType 2>/dev/null | $EGREP -i -e 'number of cores:'|cut -d : -f 2 -s|tr -d ' '` || CPU_COUNT="0"]],dnl | |
76 | [[freebsd*]],[[CPU_COUNT=`dmesg 2>/dev/null| $EGREP -e '^cpu@<:@0-9@:>@+: '|sort -u|$EGREP -e '^' -c` || CPU_COUNT="0"]],dnl | |
77 | [[netbsd*]], [[CPU_COUNT=`command -v cpuctl >/dev/null 2>&1 && cpuctl list 2>/dev/null| $EGREP -e '^@<:@0-9@:>@+ .* online ' -c` || \ | |
78 | CPU_COUNT=`dmesg 2>/dev/null| $EGREP -e '^cpu@<:@0-9@:>@+ at'|sort -u|$EGREP -e '^' -c` || CPU_COUNT="0"]],dnl | |
79 | [[solaris*]],[[command -v kstat >/dev/null 2>&1 && CPU_COUNT=`kstat -m cpu_info -s state -p 2>/dev/null | $EGREP -c -e 'on-line'` || \ | |
80 | CPU_COUNT=`kstat -m cpu_info 2>/dev/null | $EGREP -c -e 'module: cpu_info'` || CPU_COUNT="0"]],dnl | |
81 | [[mingw*]],[AS_IF([[CPU_COUNT=`reg query 'HKLM\\Hardware\\Description\\System\\CentralProcessor' 2>/dev/null | $EGREP -e '\\\\@<:@0-9@:>@+$' -c`]],dnl | |
82 | [[: # empty]],[[test "$NUMBER_OF_PROCESSORS" -gt "0" 2>/dev/null && CPU_COUNT="$NUMBER_OF_PROCESSORS"]])],dnl | |
83 | [[msys*]],[[test "$NUMBER_OF_PROCESSORS" -gt "0" 2>/dev/null && CPU_COUNT="$NUMBER_OF_PROCESSORS"]],dnl | |
84 | [[cygwin*]],[[test "$NUMBER_OF_PROCESSORS" -gt "0" 2>/dev/null && CPU_COUNT="$NUMBER_OF_PROCESSORS"]]dnl | |
85 | )dnl | |
86 | ])dnl | |
87 | ||
88 | AS_IF([[test "x$CPU_COUNT" != "x0" && test "$CPU_COUNT" -gt 0 2>/dev/null]],[dnl | |
89 | AC_MSG_RESULT([[$CPU_COUNT]]) | |
90 | m4_ifvaln([$1],[$1],)dnl | |
91 | ],[dnl | |
92 | m4_ifval([$2],[dnl | |
93 | AS_UNSET([[CPU_COUNT]]) | |
94 | AC_MSG_RESULT([[unable to detect]]) | |
95 | $2 | |
96 | ], [dnl | |
97 | CPU_COUNT="1" | |
98 | AC_MSG_RESULT([[unable to detect (assuming 1)]]) | |
99 | ])dnl | |
100 | ])dnl | |
101 | ])dnl |