1 #include <linux/kernel.h>
2 #include <linux/module.h>
3 #include <linux/init.h>
4 #include <linux/bootmem.h>
5 #include <linux/percpu.h>
7 #include <asm/percpu.h>
8 #include <asm/sections.h>
9 #include <asm/processor.h>
10 #include <asm/setup.h>
11 #include <asm/topology.h>
12 #include <asm/mpspec.h>
13 #include <asm/apicdef.h>
15 #ifdef CONFIG_X86_LOCAL_APIC
16 unsigned int num_processors;
17 unsigned disabled_cpus __cpuinitdata;
18 /* Processor that is doing the boot up */
19 unsigned int boot_cpu_physical_apicid = -1U;
20 EXPORT_SYMBOL(boot_cpu_physical_apicid);
22 /* Bitmask of physically existing CPUs */
23 physid_mask_t phys_cpu_present_map;
26 /* map cpu index to physical APIC ID */
27 DEFINE_EARLY_PER_CPU(u16, x86_cpu_to_apicid, BAD_APICID);
28 DEFINE_EARLY_PER_CPU(u16, x86_bios_cpu_apicid, BAD_APICID);
29 EXPORT_EARLY_PER_CPU_SYMBOL(x86_cpu_to_apicid);
30 EXPORT_EARLY_PER_CPU_SYMBOL(x86_bios_cpu_apicid);
32 #if defined(CONFIG_NUMA) && defined(CONFIG_X86_64)
35 /* map cpu index to node index */
36 DEFINE_EARLY_PER_CPU(int, x86_cpu_to_node_map, NUMA_NO_NODE);
37 EXPORT_EARLY_PER_CPU_SYMBOL(x86_cpu_to_node_map);
39 /* which logical CPUs are on which nodes */
40 cpumask_t *node_to_cpumask_map;
41 EXPORT_SYMBOL(node_to_cpumask_map);
43 /* setup node_to_cpumask_map */
44 static void __init setup_node_to_cpumask_map(void);
47 static inline void setup_node_to_cpumask_map(void) { }
50 #if defined(CONFIG_HAVE_SETUP_PER_CPU_AREA) && defined(CONFIG_X86_SMP)
52 * Copy data used in early init routines from the initial arrays to the
53 * per cpu data areas. These arrays then become expendable and the
54 * *_early_ptr's are zeroed indicating that the static arrays are gone.
56 static void __init setup_per_cpu_maps(void)
60 for_each_possible_cpu(cpu) {
61 per_cpu(x86_cpu_to_apicid, cpu) =
62 early_per_cpu_map(x86_cpu_to_apicid, cpu);
63 per_cpu(x86_bios_cpu_apicid, cpu) =
64 early_per_cpu_map(x86_bios_cpu_apicid, cpu);
66 per_cpu(x86_cpu_to_node_map, cpu) =
67 early_per_cpu_map(x86_cpu_to_node_map, cpu);
71 /* indicate the early static arrays will soon be gone */
72 early_per_cpu_ptr(x86_cpu_to_apicid) = NULL;
73 early_per_cpu_ptr(x86_bios_cpu_apicid) = NULL;
75 early_per_cpu_ptr(x86_cpu_to_node_map) = NULL;
79 #ifdef CONFIG_HAVE_CPUMASK_OF_CPU_MAP
80 cpumask_t *cpumask_of_cpu_map __read_mostly;
81 EXPORT_SYMBOL(cpumask_of_cpu_map);
83 /* requires nr_cpu_ids to be initialized */
84 static void __init setup_cpumask_of_cpu(void)
88 /* alloc_bootmem zeroes memory */
89 cpumask_of_cpu_map = alloc_bootmem_low(sizeof(cpumask_t) * nr_cpu_ids);
90 for (i = 0; i < nr_cpu_ids; i++)
91 cpu_set(i, cpumask_of_cpu_map[i]);
94 static inline void setup_cpumask_of_cpu(void) { }
99 * Great future not-so-futuristic plan: make i386 and x86_64 do it
102 unsigned long __per_cpu_offset[NR_CPUS] __read_mostly;
103 EXPORT_SYMBOL(__per_cpu_offset);
108 * Declare PDA itself and support (irqstack,tss,pgd) as per cpu data.
109 * Always point %gs to its beginning
111 void __init setup_per_cpu_areas(void)
113 int i, highest_cpu = 0;
116 #ifdef CONFIG_HOTPLUG_CPU
117 prefill_possible_map();
120 /* Copy section for each CPU (we discard the original) */
121 size = PERCPU_ENOUGH_ROOM;
122 printk(KERN_INFO "PERCPU: Allocating %lu bytes of per cpu data\n",
125 for_each_possible_cpu(i) {
127 #ifndef CONFIG_NEED_MULTIPLE_NODES
128 ptr = alloc_bootmem_pages(size);
130 int node = early_cpu_to_node(i);
131 if (!node_online(node) || !NODE_DATA(node)) {
132 ptr = alloc_bootmem_pages(size);
134 "cpu %d has no node %d or node-local memory\n",
138 ptr = alloc_bootmem_pages_node(NODE_DATA(node), size);
141 panic("Cannot allocate cpu data for CPU %d\n", i);
143 cpu_pda(i)->data_offset = ptr - __per_cpu_start;
145 __per_cpu_offset[i] = ptr - __per_cpu_start;
147 memcpy(ptr, __per_cpu_start, __per_cpu_end - __per_cpu_start);
152 nr_cpu_ids = highest_cpu + 1;
153 printk(KERN_DEBUG "NR_CPUS: %d, nr_cpu_ids: %d, nr_node_ids %d\n",
154 NR_CPUS, nr_cpu_ids, nr_node_ids);
156 /* Setup percpu data maps */
157 setup_per_cpu_maps();
159 /* Setup node to cpumask map */
160 setup_node_to_cpumask_map();
162 /* Setup cpumask_of_cpu map */
163 setup_cpumask_of_cpu();
171 * Allocate node_to_cpumask_map based on number of available nodes
172 * Requires node_possible_map to be valid.
174 * Note: node_to_cpumask() is not valid until after this is done.
176 static void __init setup_node_to_cpumask_map(void)
178 unsigned int node, num = 0;
181 /* setup nr_node_ids if not done yet */
182 if (nr_node_ids == MAX_NUMNODES) {
183 for_each_node_mask(node, node_possible_map)
185 nr_node_ids = num + 1;
188 /* allocate the map */
189 map = alloc_bootmem_low(nr_node_ids * sizeof(cpumask_t));
191 Dprintk(KERN_DEBUG "Node to cpumask map at %p for %d nodes\n",
194 /* node_to_cpumask() will now work */
195 node_to_cpumask_map = map;
198 void __cpuinit numa_set_node(int cpu, int node)
200 int *cpu_to_node_map = early_per_cpu_ptr(x86_cpu_to_node_map);
202 if (node != NUMA_NO_NODE)
203 cpu_pda(cpu)->nodenumber = node;
206 cpu_to_node_map[cpu] = node;
208 else if (per_cpu_offset(cpu))
209 per_cpu(x86_cpu_to_node_map, cpu) = node;
212 Dprintk(KERN_INFO "Setting node for non-present cpu %d\n", cpu);
215 void __cpuinit numa_clear_node(int cpu)
217 numa_set_node(cpu, NUMA_NO_NODE);
220 #ifndef CONFIG_DEBUG_PER_CPU_MAPS
222 void __cpuinit numa_add_cpu(int cpu)
224 cpu_set(cpu, node_to_cpumask_map[early_cpu_to_node(cpu)]);
227 void __cpuinit numa_remove_cpu(int cpu)
229 cpu_clear(cpu, node_to_cpumask_map[cpu_to_node(cpu)]);
232 #else /* CONFIG_DEBUG_PER_CPU_MAPS */
235 * --------- debug versions of the numa functions ---------
237 static void __cpuinit numa_set_cpumask(int cpu, int enable)
239 int node = cpu_to_node(cpu);
243 if (node_to_cpumask_map == NULL) {
244 printk(KERN_ERR "node_to_cpumask_map NULL\n");
249 mask = &node_to_cpumask_map[node];
253 cpu_clear(cpu, *mask);
255 cpulist_scnprintf(buf, sizeof(buf), *mask);
256 printk(KERN_DEBUG "%s cpu %d node %d: mask now %s\n",
257 enable? "numa_add_cpu":"numa_remove_cpu", cpu, node, buf);
260 void __cpuinit numa_add_cpu(int cpu)
262 numa_set_cpumask(cpu, 1);
265 void __cpuinit numa_remove_cpu(int cpu)
267 numa_set_cpumask(cpu, 0);
270 int cpu_to_node(int cpu)
272 if (early_per_cpu_ptr(x86_cpu_to_node_map)) {
274 "cpu_to_node(%d): usage too early!\n", cpu);
276 return early_per_cpu_ptr(x86_cpu_to_node_map)[cpu];
278 return per_cpu(x86_cpu_to_node_map, cpu);
280 EXPORT_SYMBOL(cpu_to_node);
283 * Same function as cpu_to_node() but used if called before the
284 * per_cpu areas are setup.
286 int early_cpu_to_node(int cpu)
288 if (early_per_cpu_ptr(x86_cpu_to_node_map))
289 return early_per_cpu_ptr(x86_cpu_to_node_map)[cpu];
291 if (!per_cpu_offset(cpu)) {
293 "early_cpu_to_node(%d): no per_cpu area!\n", cpu);
297 return per_cpu(x86_cpu_to_node_map, cpu);
301 * Returns a pointer to the bitmask of CPUs on Node 'node'.
303 cpumask_t *_node_to_cpumask_ptr(int node)
305 if (node_to_cpumask_map == NULL) {
307 "_node_to_cpumask_ptr(%d): no node_to_cpumask_map!\n",
310 return &cpu_online_map;
312 return &node_to_cpumask_map[node];
314 EXPORT_SYMBOL(_node_to_cpumask_ptr);
317 * Returns a bitmask of CPUs on Node 'node'.
319 cpumask_t node_to_cpumask(int node)
321 if (node_to_cpumask_map == NULL) {
323 "node_to_cpumask(%d): no node_to_cpumask_map!\n", node);
325 return cpu_online_map;
327 return node_to_cpumask_map[node];
329 EXPORT_SYMBOL(node_to_cpumask);
332 * --------- end of debug versions of the numa functions ---------
335 #endif /* CONFIG_DEBUG_PER_CPU_MAPS */
337 #endif /* X86_64_NUMA */