Home My Page Projects Code Snippets Project Openings diderot
Summary Activity Tracker Tasks SCM

SCM Repository

[diderot] Annotation of /branches/lamont/src/lib/parallel-target/cpu.c
ViewVC logotype

Annotation of /branches/lamont/src/lib/parallel-target/cpu.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2081 - (view) (download) (as text)

1 : jhr 1232 /*! \file cpu.c
2 :     *
3 :     * \author John Reppy
4 :     */
5 :    
6 :     /*
7 :     * COPYRIGHT (c) 2011 The Diderot Project (http://diderot-language.cs.uchicago.edu)
8 :     * All rights reserved.
9 :     */
10 :    
11 :     #include "Diderot/diderot.h"
12 :     #if defined (__APPLE__)
13 :     # include <sys/sysctl.h>
14 :     #endif
15 :     #ifdef HAVE_LIBNUMA
16 :     # include <numa.h>
17 :     #endif
18 :     #include <errno.h>
19 :    
20 : jhr 1831 /*! \brief determine the number of CPUs
21 :     * \return true if there is an error, false otherwise.
22 :     */
23 :     bool Diderot_GetNumCPUs (CPUInfo_t *info)
24 : jhr 1232 {
25 :     #if defined(HAVE__PROC_CPUINFO)
26 :     /* Get information from /proc/cpuinfo. The interesting
27 :     * fields are:
28 :     *
29 : jhr 1301 * processor : <id> # logical processor id
30 :     * physical id : <id> # node id
31 :     * core id : <id> # core id (per node)
32 :     * cpu cores : <n> # number of cores per node
33 : jhr 1232 */
34 :     FILE *cpuinfo = fopen("/proc/cpuinfo", "r");
35 :     char buf[1024];
36 :     if (cpuinfo != NULL) {
37 : jhr 1301 int maxProcId = 0, maxNodeId = 0, maxCoreId = 0, nCores = 0;
38 :     while (fgets(buf, sizeof(buf), cpuinfo) != 0) {
39 :     int tmp;
40 :     if (sscanf(buf, "processor : %d", &tmp) == 1) {
41 :     maxProcId = (tmp > maxProcId) ? tmp : maxProcId;
42 :     }
43 :     else if (sscanf(buf, "physical id : %d", &tmp) == 1) {
44 :     maxNodeId = (tmp > maxNodeId) ? tmp : maxNodeId;
45 :     }
46 :     else if (sscanf(buf, "core id : %d", &tmp) == 1) {
47 :     maxCoreId = (tmp > maxCoreId) ? tmp : maxCoreId;
48 :     }
49 :     else if (sscanf(buf, "cpu cores : %d", &tmp) == 1) {
50 :     nCores = (tmp > nCores) ? tmp : nCores;
51 :     }
52 :     }
53 :     fclose (cpuinfo);
54 : jhr 1232
55 :     #ifdef HAVE_LIBNUMA
56 : jhr 1301 info->numHWThreads = info->numHWCores = maxProcId + 1;
57 :     info->numHWNodes = numa_max_node()+1;
58 :     info->numThdsPerCore = 1;
59 :     info->numCoresPerNode = info->numHWThreads / info->numHWNodes;
60 : jhr 1232 #else
61 : jhr 1301 info->numHWNodes = maxNodeId + 1;
62 :     info->numHWCores = info->numHWNodes * nCores;
63 :     info->numHWThreads = maxProcId + 1;
64 :     info->numCoresPerNode = nCores;
65 :     info->numThdsPerCore = info->numHWThreads / info->numHWCores;
66 : jhr 1232 #endif
67 :    
68 : jhr 1831 return false;
69 : jhr 1232 }
70 :     else {
71 : jhr 1301 fprintf(stderr, "unable to determine the number of processors\n");
72 : jhr 1831 return true;
73 : jhr 1232 }
74 :     #elif defined(__APPLE__)
75 : jhr 1301 size_t len = sizeof(int);
76 : jhr 1232
77 :     /* the number of nodes */
78 :     if (sysctlbyname("hw.packages", &(info->numHWNodes), &len, 0, 0) < 0) {
79 : jhr 1301 if (errno == ENOENT) {
80 :     // "hw.packages" is not known
81 :     info->numHWNodes = 1;
82 :     }
83 :     else {
84 :     fprintf(stderr, "unable to determine the number of nodes\n");
85 : jhr 1831 return true;
86 : jhr 1301 }
87 : jhr 1232 }
88 :    
89 :     /* the number of cores */
90 :     if (sysctlbyname("hw.physicalcpu", &(info->numHWCores), &len, 0, 0) < 0) {
91 : jhr 1301 fprintf(stderr, "unable to determine the number of physical CPUs\n");
92 : jhr 1831 return true;
93 : jhr 1232 }
94 :    
95 :     /* the number of hardware threads */
96 :     if (sysctlbyname("hw.logicalcpu", &(info->numHWThreads), &len, 0, 0) < 0) {
97 : jhr 1301 if (errno == ENOENT) {
98 :     // "hw.packages" is not known
99 :     info->numHWThreads = info->numHWCores;
100 :     }
101 :     else {
102 :     fprintf(stderr, "unable to determine the number of logical CPUs\n");
103 : jhr 1831 return true;
104 : jhr 1301 }
105 : jhr 1232 }
106 :    
107 :     info->numCoresPerNode = info->numHWCores / info->numHWNodes;
108 :     info->numThdsPerCore = info->numHWThreads / info->numHWCores;
109 :    
110 : jhr 1831 return false;
111 :     #else
112 : jhr 1232 return true;
113 :     #endif
114 :    
115 : jhr 1834 } /* end of Diderot_GetNumCPUs */

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0