1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
/*
* Copyright (c) 2004-2008 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2005 The University of Tennessee and The University
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* Copyright (c) 2007 Cisco Systems, Inc. All rights reserved.
* Copyright (c) 2008 Sun Microsystems, Inc. All rights reserved.
*
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "opal_config.h"
#include "opal/constants.h"
#include "opal/mca/base/mca_base_param.h"
#include "opal/mca/paffinity/paffinity.h"
#include "opal/mca/paffinity/base/base.h"
#include "paffinity_test.h"
/*
* Local functions
*/
static int init(void);
static int set(opal_paffinity_base_cpu_set_t cpumask);
static int get(opal_paffinity_base_cpu_set_t *cpumask);
static int finalize(void);
static int map_to_processor_id(int socket, int core, int *processor_id);
static int map_to_socket_core(int processor_id, int *socket, int *core);
static int get_processor_info(int *num_processors);
static int get_socket_info(int *num_sockets);
static int get_core_info(int socket, int *num_cores);
static int get_physical_processor_id(int logical_processor_id);
static int get_physical_socket_id(int logical_socket_id);
static int get_physical_core_id(int physical_socket_id, int logical_core_id);
/*
* Test paffinity module
*/
opal_paffinity_base_module_t opal_paffinity_test_module = {
/* Initialization function */
init,
/* Module function pointers */
set,
get,
map_to_processor_id,
map_to_socket_core,
get_processor_info,
get_socket_info,
get_core_info,
get_physical_processor_id,
get_physical_socket_id,
get_physical_core_id,
finalize
};
/* nothing to init here */
static int init(void)
{
return OPAL_SUCCESS;
}
/* this gives us a cpumask which tells which CPU to bind */
static int set(opal_paffinity_base_cpu_set_t cpumask)
{
return OPAL_SUCCESS;
}
/* This get function returns the CPU id that's currently binded,
* and then sets the cpumask. */
static int get(opal_paffinity_base_cpu_set_t *cpumask)
{
int i;
OPAL_PAFFINITY_CPU_ZERO(*cpumask);
if (mca_paffinity_test_component.bound) {
for (i=0; i < mca_paffinity_test_component.num_sockets*mca_paffinity_test_component.num_cores; i+=2) {
OPAL_PAFFINITY_CPU_SET(i, *cpumask);
}
/* assign all cores in the 2nd socket, if it exists */
if (mca_paffinity_test_component.num_sockets >= 2) {
for (i=mca_paffinity_test_component.num_cores; i < 2*mca_paffinity_test_component.num_cores; i++) {
OPAL_PAFFINITY_CPU_SET(i, *cpumask);
}
}
} else {
for (i=0; i < mca_paffinity_test_component.num_sockets*mca_paffinity_test_component.num_cores; i++) {
OPAL_PAFFINITY_CPU_SET(i, *cpumask);
}
}
return OPAL_SUCCESS;
}
static int map_to_processor_id(int socket, int core, int *processor_id)
{
*processor_id = socket*mca_paffinity_test_component.num_cores + core;
return OPAL_SUCCESS;
}
static int map_to_socket_core(int processor_id, int *socket, int *core)
{
*socket = processor_id / mca_paffinity_test_component.num_cores;
*core = processor_id % mca_paffinity_test_component.num_cores;
return OPAL_SUCCESS;
}
static int get_processor_info(int *num_processors)
{
*num_processors = mca_paffinity_test_component.num_sockets * mca_paffinity_test_component.num_cores;
return OPAL_SUCCESS;
}
static int get_socket_info(int *num_sockets)
{
*num_sockets = mca_paffinity_test_component.num_sockets;
return OPAL_SUCCESS;
}
static int get_core_info(int socket, int *num_cores)
{
*num_cores = mca_paffinity_test_component.num_cores;
return OPAL_SUCCESS;
}
static int get_physical_processor_id(int logical_processor_id)
{
return logical_processor_id;
}
static int get_physical_socket_id(int logical_socket_id)
{
return logical_socket_id;
}
static int get_physical_core_id(int physical_socket_id, int logical_core_id)
{
if (mca_paffinity_test_component.num_cores < logical_core_id) {
return OPAL_ERROR;
}
return logical_core_id;
}
static int finalize(void)
{
return OPAL_SUCCESS;
}
|
