blob: 496f494593bf6cfa7f7069c58af5c153f6af1965 [file] [log] [blame]
Rusty Russell71ee73e2009-03-13 14:49:52 +10301/* Common code for 32 and 64-bit NUMA */
Tejun Heoa4106ea2011-05-02 14:18:53 +02002#include <linux/kernel.h>
3#include <linux/mm.h>
4#include <linux/string.h>
5#include <linux/init.h>
Rusty Russell71ee73e2009-03-13 14:49:52 +10306#include <linux/bootmem.h>
Tejun Heoa4106ea2011-05-02 14:18:53 +02007#include <linux/memblock.h>
8#include <linux/mmzone.h>
9#include <linux/ctype.h>
10#include <linux/module.h>
11#include <linux/nodemask.h>
12#include <linux/sched.h>
13#include <linux/topology.h>
14
15#include <asm/e820.h>
16#include <asm/proto.h>
17#include <asm/dma.h>
Jan Beulich90321602011-01-19 08:57:21 +000018#include <asm/acpi.h>
Tejun Heoa4106ea2011-05-02 14:18:53 +020019#include <asm/amd_nb.h>
20
21#include "numa_internal.h"
Jan Beulich90321602011-01-19 08:57:21 +000022
23int __initdata numa_off;
Tejun Heoe6df5952011-05-02 14:18:53 +020024nodemask_t numa_nodes_parsed __initdata;
Jan Beulich90321602011-01-19 08:57:21 +000025
Tejun Heoa4106ea2011-05-02 14:18:53 +020026struct pglist_data *node_data[MAX_NUMNODES] __read_mostly;
27EXPORT_SYMBOL(node_data);
28
29static struct numa_meminfo numa_meminfo
30#ifndef CONFIG_MEMORY_HOTPLUG
31__initdata
32#endif
33;
34
35static int numa_distance_cnt;
36static u8 *numa_distance;
Tejun Heoa4106ea2011-05-02 14:18:53 +020037
Jan Beulich90321602011-01-19 08:57:21 +000038static __init int numa_setup(char *opt)
39{
40 if (!opt)
41 return -EINVAL;
42 if (!strncmp(opt, "off", 3))
43 numa_off = 1;
44#ifdef CONFIG_NUMA_EMU
45 if (!strncmp(opt, "fake=", 5))
46 numa_emu_cmdline(opt + 5);
47#endif
48#ifdef CONFIG_ACPI_NUMA
49 if (!strncmp(opt, "noacpi", 6))
50 acpi_numa = -1;
51#endif
52 return 0;
53}
54early_param("numa", numa_setup);
Rusty Russell71ee73e2009-03-13 14:49:52 +103055
Rusty Russell71ee73e2009-03-13 14:49:52 +103056/*
Tejun Heobbc9e2f2011-01-23 14:37:39 +010057 * apicid, cpu, node mappings
Rusty Russell71ee73e2009-03-13 14:49:52 +103058 */
Tejun Heobbc9e2f2011-01-23 14:37:39 +010059s16 __apicid_to_node[MAX_LOCAL_APIC] __cpuinitdata = {
60 [0 ... MAX_LOCAL_APIC-1] = NUMA_NO_NODE
61};
62
Tejun Heo6bd26272011-05-02 14:18:52 +020063int __cpuinit numa_cpu_node(int cpu)
64{
65 int apicid = early_per_cpu(x86_cpu_to_apicid, cpu);
66
67 if (apicid != BAD_APICID)
68 return __apicid_to_node[apicid];
69 return NUMA_NO_NODE;
70}
71
Rusty Russellc032ef602009-03-13 14:49:53 +103072cpumask_var_t node_to_cpumask_map[MAX_NUMNODES];
Rusty Russell71ee73e2009-03-13 14:49:52 +103073EXPORT_SYMBOL(node_to_cpumask_map);
74
75/*
Tejun Heo645a7912011-01-23 14:37:40 +010076 * Map cpu index to node index
77 */
Tejun Heo645a7912011-01-23 14:37:40 +010078DEFINE_EARLY_PER_CPU(int, x86_cpu_to_node_map, NUMA_NO_NODE);
Tejun Heo645a7912011-01-23 14:37:40 +010079EXPORT_EARLY_PER_CPU_SYMBOL(x86_cpu_to_node_map);
80
81void __cpuinit numa_set_node(int cpu, int node)
82{
83 int *cpu_to_node_map = early_per_cpu_ptr(x86_cpu_to_node_map);
84
85 /* early setting, no percpu area yet */
86 if (cpu_to_node_map) {
87 cpu_to_node_map[cpu] = node;
88 return;
89 }
90
91#ifdef CONFIG_DEBUG_PER_CPU_MAPS
92 if (cpu >= nr_cpu_ids || !cpu_possible(cpu)) {
93 printk(KERN_ERR "numa_set_node: invalid cpu# (%d)\n", cpu);
94 dump_stack();
95 return;
96 }
97#endif
98 per_cpu(x86_cpu_to_node_map, cpu) = node;
99
100 if (node != NUMA_NO_NODE)
101 set_cpu_numa_node(cpu, node);
102}
103
104void __cpuinit numa_clear_node(int cpu)
105{
106 numa_set_node(cpu, NUMA_NO_NODE);
107}
108
109/*
Rusty Russell71ee73e2009-03-13 14:49:52 +1030110 * Allocate node_to_cpumask_map based on number of available nodes
111 * Requires node_possible_map to be valid.
112 *
113 * Note: node_to_cpumask() is not valid until after this is done.
114 * (Use CONFIG_DEBUG_PER_CPU_MAPS to check this.)
115 */
116void __init setup_node_to_cpumask_map(void)
117{
118 unsigned int node, num = 0;
Rusty Russell71ee73e2009-03-13 14:49:52 +1030119
120 /* setup nr_node_ids if not done yet */
121 if (nr_node_ids == MAX_NUMNODES) {
122 for_each_node_mask(node, node_possible_map)
123 num = node;
124 nr_node_ids = num + 1;
125 }
126
127 /* allocate the map */
Rusty Russellc032ef602009-03-13 14:49:53 +1030128 for (node = 0; node < nr_node_ids; node++)
129 alloc_bootmem_cpumask_var(&node_to_cpumask_map[node]);
Rusty Russell71ee73e2009-03-13 14:49:52 +1030130
Rusty Russellc032ef602009-03-13 14:49:53 +1030131 /* cpumask_of_node() will now work */
132 pr_debug("Node to cpumask map for %d nodes\n", nr_node_ids);
Rusty Russell71ee73e2009-03-13 14:49:52 +1030133}
134
Tejun Heoa4106ea2011-05-02 14:18:53 +0200135static int __init numa_add_memblk_to(int nid, u64 start, u64 end,
136 struct numa_meminfo *mi)
137{
138 /* ignore zero length blks */
139 if (start == end)
140 return 0;
141
142 /* whine about and ignore invalid blks */
143 if (start > end || nid < 0 || nid >= MAX_NUMNODES) {
144 pr_warning("NUMA: Warning: invalid memblk node %d (%Lx-%Lx)\n",
145 nid, start, end);
146 return 0;
147 }
148
149 if (mi->nr_blks >= NR_NODE_MEMBLKS) {
150 pr_err("NUMA: too many memblk ranges\n");
151 return -EINVAL;
152 }
153
154 mi->blk[mi->nr_blks].start = start;
155 mi->blk[mi->nr_blks].end = end;
156 mi->blk[mi->nr_blks].nid = nid;
157 mi->nr_blks++;
158 return 0;
159}
160
161/**
162 * numa_remove_memblk_from - Remove one numa_memblk from a numa_meminfo
163 * @idx: Index of memblk to remove
164 * @mi: numa_meminfo to remove memblk from
165 *
166 * Remove @idx'th numa_memblk from @mi by shifting @mi->blk[] and
167 * decrementing @mi->nr_blks.
168 */
169void __init numa_remove_memblk_from(int idx, struct numa_meminfo *mi)
170{
171 mi->nr_blks--;
172 memmove(&mi->blk[idx], &mi->blk[idx + 1],
173 (mi->nr_blks - idx) * sizeof(mi->blk[0]));
174}
175
176/**
177 * numa_add_memblk - Add one numa_memblk to numa_meminfo
178 * @nid: NUMA node ID of the new memblk
179 * @start: Start address of the new memblk
180 * @end: End address of the new memblk
181 *
182 * Add a new memblk to the default numa_meminfo.
183 *
184 * RETURNS:
185 * 0 on success, -errno on failure.
186 */
187int __init numa_add_memblk(int nid, u64 start, u64 end)
188{
189 return numa_add_memblk_to(nid, start, end, &numa_meminfo);
190}
191
Yinghai Lua56bca82011-05-02 17:24:49 +0200192/* Initialize NODE_DATA for a node on the local memory */
193static void __init setup_node_data(int nid, u64 start, u64 end)
Tejun Heoa4106ea2011-05-02 14:18:53 +0200194{
Tejun Heoa4106ea2011-05-02 14:18:53 +0200195 const size_t nd_size = roundup(sizeof(pg_data_t), PAGE_SIZE);
Tejun Heo7888e962011-05-02 14:18:54 +0200196 bool remapped = false;
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200197 u64 nd_pa;
Tejun Heo7888e962011-05-02 14:18:54 +0200198 void *nd;
Tejun Heoa4106ea2011-05-02 14:18:53 +0200199 int tnid;
200
201 /*
202 * Don't confuse VM with a node that doesn't have the
203 * minimum amount of memory:
204 */
205 if (end && (end - start) < NODE_MIN_SIZE)
206 return;
207
Tejun Heo7888e962011-05-02 14:18:54 +0200208 /* initialize remap allocator before aligning to ZONE_ALIGN */
209 init_alloc_remap(nid, start, end);
210
Tejun Heoa4106ea2011-05-02 14:18:53 +0200211 start = roundup(start, ZONE_ALIGN);
212
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200213 printk(KERN_INFO "Initmem setup node %d %016Lx-%016Lx\n",
Tejun Heoa4106ea2011-05-02 14:18:53 +0200214 nid, start, end);
215
216 /*
Tejun Heo7888e962011-05-02 14:18:54 +0200217 * Allocate node data. Try remap allocator first, node-local
218 * memory and then any node. Never allocate in DMA zone.
Tejun Heoa4106ea2011-05-02 14:18:53 +0200219 */
Tejun Heo7888e962011-05-02 14:18:54 +0200220 nd = alloc_remap(nid, nd_size);
221 if (nd) {
222 nd_pa = __pa(nd);
223 remapped = true;
224 } else {
Tejun Heoeb40c4c2011-07-12 10:46:35 +0200225 nd_pa = memblock_alloc_nid(nd_size, SMP_CACHE_BYTES, nid);
Tejun Heo1f5026a2011-07-12 09:58:09 +0200226 if (!nd_pa) {
Tejun Heo7888e962011-05-02 14:18:54 +0200227 pr_err("Cannot find %zu bytes in node %d\n",
228 nd_size, nid);
229 return;
230 }
Tejun Heo7888e962011-05-02 14:18:54 +0200231 nd = __va(nd_pa);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200232 }
Tejun Heoa4106ea2011-05-02 14:18:53 +0200233
234 /* report and initialize */
Tejun Heo7888e962011-05-02 14:18:54 +0200235 printk(KERN_INFO " NODE_DATA [%016Lx - %016Lx]%s\n",
236 nd_pa, nd_pa + nd_size - 1, remapped ? " (remapped)" : "");
Tejun Heoa4106ea2011-05-02 14:18:53 +0200237 tnid = early_pfn_to_nid(nd_pa >> PAGE_SHIFT);
Tejun Heo7888e962011-05-02 14:18:54 +0200238 if (!remapped && tnid != nid)
Tejun Heoa4106ea2011-05-02 14:18:53 +0200239 printk(KERN_INFO " NODE_DATA(%d) on node %d\n", nid, tnid);
240
Tejun Heo7888e962011-05-02 14:18:54 +0200241 node_data[nid] = nd;
Tejun Heoa4106ea2011-05-02 14:18:53 +0200242 memset(NODE_DATA(nid), 0, sizeof(pg_data_t));
243 NODE_DATA(nid)->node_id = nid;
244 NODE_DATA(nid)->node_start_pfn = start >> PAGE_SHIFT;
245 NODE_DATA(nid)->node_spanned_pages = (end - start) >> PAGE_SHIFT;
246
247 node_set_online(nid);
248}
249
250/**
251 * numa_cleanup_meminfo - Cleanup a numa_meminfo
252 * @mi: numa_meminfo to clean up
253 *
254 * Sanitize @mi by merging and removing unncessary memblks. Also check for
255 * conflicts and clear unused memblks.
256 *
257 * RETURNS:
258 * 0 on success, -errno on failure.
259 */
260int __init numa_cleanup_meminfo(struct numa_meminfo *mi)
261{
262 const u64 low = 0;
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200263 const u64 high = PFN_PHYS(max_pfn);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200264 int i, j, k;
265
Yinghai Lue5a10c12011-05-02 17:24:49 +0200266 /* first, trim all entries */
Tejun Heoa4106ea2011-05-02 14:18:53 +0200267 for (i = 0; i < mi->nr_blks; i++) {
268 struct numa_memblk *bi = &mi->blk[i];
269
270 /* make sure all blocks are inside the limits */
271 bi->start = max(bi->start, low);
272 bi->end = min(bi->end, high);
273
274 /* and there's no empty block */
Yinghai Lue5a10c12011-05-02 17:24:49 +0200275 if (bi->start >= bi->end)
Tejun Heoa4106ea2011-05-02 14:18:53 +0200276 numa_remove_memblk_from(i--, mi);
Yinghai Lue5a10c12011-05-02 17:24:49 +0200277 }
278
279 /* merge neighboring / overlapping entries */
280 for (i = 0; i < mi->nr_blks; i++) {
281 struct numa_memblk *bi = &mi->blk[i];
Tejun Heoa4106ea2011-05-02 14:18:53 +0200282
283 for (j = i + 1; j < mi->nr_blks; j++) {
284 struct numa_memblk *bj = &mi->blk[j];
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200285 u64 start, end;
Tejun Heoa4106ea2011-05-02 14:18:53 +0200286
287 /*
288 * See whether there are overlapping blocks. Whine
289 * about but allow overlaps of the same nid. They
290 * will be merged below.
291 */
292 if (bi->end > bj->start && bi->start < bj->end) {
293 if (bi->nid != bj->nid) {
294 pr_err("NUMA: node %d (%Lx-%Lx) overlaps with node %d (%Lx-%Lx)\n",
295 bi->nid, bi->start, bi->end,
296 bj->nid, bj->start, bj->end);
297 return -EINVAL;
298 }
299 pr_warning("NUMA: Warning: node %d (%Lx-%Lx) overlaps with itself (%Lx-%Lx)\n",
300 bi->nid, bi->start, bi->end,
301 bj->start, bj->end);
302 }
303
304 /*
305 * Join together blocks on the same node, holes
306 * between which don't overlap with memory on other
307 * nodes.
308 */
309 if (bi->nid != bj->nid)
310 continue;
Yinghai Lue5a10c12011-05-02 17:24:49 +0200311 start = min(bi->start, bj->start);
312 end = max(bi->end, bj->end);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200313 for (k = 0; k < mi->nr_blks; k++) {
314 struct numa_memblk *bk = &mi->blk[k];
315
316 if (bi->nid == bk->nid)
317 continue;
318 if (start < bk->end && end > bk->start)
319 break;
320 }
321 if (k < mi->nr_blks)
322 continue;
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200323 printk(KERN_INFO "NUMA: Node %d [%Lx,%Lx) + [%Lx,%Lx) -> [%Lx,%Lx)\n",
Tejun Heoa4106ea2011-05-02 14:18:53 +0200324 bi->nid, bi->start, bi->end, bj->start, bj->end,
325 start, end);
326 bi->start = start;
327 bi->end = end;
328 numa_remove_memblk_from(j--, mi);
329 }
330 }
331
Yinghai Lue5a10c12011-05-02 17:24:49 +0200332 /* clear unused ones */
Tejun Heoa4106ea2011-05-02 14:18:53 +0200333 for (i = mi->nr_blks; i < ARRAY_SIZE(mi->blk); i++) {
334 mi->blk[i].start = mi->blk[i].end = 0;
335 mi->blk[i].nid = NUMA_NO_NODE;
336 }
337
338 return 0;
339}
340
341/*
342 * Set nodes, which have memory in @mi, in *@nodemask.
343 */
344static void __init numa_nodemask_from_meminfo(nodemask_t *nodemask,
345 const struct numa_meminfo *mi)
346{
347 int i;
348
349 for (i = 0; i < ARRAY_SIZE(mi->blk); i++)
350 if (mi->blk[i].start != mi->blk[i].end &&
351 mi->blk[i].nid != NUMA_NO_NODE)
352 node_set(mi->blk[i].nid, *nodemask);
353}
354
355/**
356 * numa_reset_distance - Reset NUMA distance table
357 *
358 * The current table is freed. The next numa_set_distance() call will
359 * create a new one.
360 */
361void __init numa_reset_distance(void)
362{
363 size_t size = numa_distance_cnt * numa_distance_cnt * sizeof(numa_distance[0]);
364
365 /* numa_distance could be 1LU marking allocation failure, test cnt */
366 if (numa_distance_cnt)
Tejun Heo24aa0782011-07-12 11:16:06 +0200367 memblock_free(__pa(numa_distance), size);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200368 numa_distance_cnt = 0;
369 numa_distance = NULL; /* enable table creation */
370}
371
372static int __init numa_alloc_distance(void)
373{
374 nodemask_t nodes_parsed;
375 size_t size;
376 int i, j, cnt = 0;
377 u64 phys;
378
379 /* size the new table and allocate it */
380 nodes_parsed = numa_nodes_parsed;
381 numa_nodemask_from_meminfo(&nodes_parsed, &numa_meminfo);
382
383 for_each_node_mask(i, nodes_parsed)
384 cnt = i;
385 cnt++;
386 size = cnt * cnt * sizeof(numa_distance[0]);
387
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200388 phys = memblock_find_in_range(0, PFN_PHYS(max_pfn_mapped),
Tejun Heoa4106ea2011-05-02 14:18:53 +0200389 size, PAGE_SIZE);
Tejun Heo1f5026a2011-07-12 09:58:09 +0200390 if (!phys) {
Tejun Heoa4106ea2011-05-02 14:18:53 +0200391 pr_warning("NUMA: Warning: can't allocate distance table!\n");
392 /* don't retry until explicitly reset */
393 numa_distance = (void *)1LU;
394 return -ENOMEM;
395 }
Tejun Heo24aa0782011-07-12 11:16:06 +0200396 memblock_reserve(phys, size);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200397
398 numa_distance = __va(phys);
399 numa_distance_cnt = cnt;
400
401 /* fill with the default distances */
402 for (i = 0; i < cnt; i++)
403 for (j = 0; j < cnt; j++)
404 numa_distance[i * cnt + j] = i == j ?
405 LOCAL_DISTANCE : REMOTE_DISTANCE;
406 printk(KERN_DEBUG "NUMA: Initialized distance table, cnt=%d\n", cnt);
407
408 return 0;
409}
410
411/**
412 * numa_set_distance - Set NUMA distance from one NUMA to another
413 * @from: the 'from' node to set distance
414 * @to: the 'to' node to set distance
415 * @distance: NUMA distance
416 *
417 * Set the distance from node @from to @to to @distance. If distance table
418 * doesn't exist, one which is large enough to accommodate all the currently
419 * known nodes will be created.
420 *
421 * If such table cannot be allocated, a warning is printed and further
422 * calls are ignored until the distance table is reset with
423 * numa_reset_distance().
424 *
425 * If @from or @to is higher than the highest known node at the time of
426 * table creation or @distance doesn't make sense, the call is ignored.
427 * This is to allow simplification of specific NUMA config implementations.
428 */
429void __init numa_set_distance(int from, int to, int distance)
430{
431 if (!numa_distance && numa_alloc_distance() < 0)
432 return;
433
434 if (from >= numa_distance_cnt || to >= numa_distance_cnt) {
435 printk_once(KERN_DEBUG "NUMA: Debug: distance out of bound, from=%d to=%d distance=%d\n",
436 from, to, distance);
437 return;
438 }
439
440 if ((u8)distance != distance ||
441 (from == to && distance != LOCAL_DISTANCE)) {
442 pr_warn_once("NUMA: Warning: invalid distance parameter, from=%d to=%d distance=%d\n",
443 from, to, distance);
444 return;
445 }
446
447 numa_distance[from * numa_distance_cnt + to] = distance;
448}
449
450int __node_distance(int from, int to)
451{
452 if (from >= numa_distance_cnt || to >= numa_distance_cnt)
453 return from == to ? LOCAL_DISTANCE : REMOTE_DISTANCE;
454 return numa_distance[from * numa_distance_cnt + to];
455}
456EXPORT_SYMBOL(__node_distance);
457
458/*
459 * Sanity check to catch more bad NUMA configurations (they are amazingly
460 * common). Make sure the nodes cover all memory.
461 */
462static bool __init numa_meminfo_cover_memory(const struct numa_meminfo *mi)
463{
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200464 u64 numaram, e820ram;
Tejun Heoa4106ea2011-05-02 14:18:53 +0200465 int i;
466
467 numaram = 0;
468 for (i = 0; i < mi->nr_blks; i++) {
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200469 u64 s = mi->blk[i].start >> PAGE_SHIFT;
470 u64 e = mi->blk[i].end >> PAGE_SHIFT;
Tejun Heoa4106ea2011-05-02 14:18:53 +0200471 numaram += e - s;
472 numaram -= __absent_pages_in_range(mi->blk[i].nid, s, e);
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200473 if ((s64)numaram < 0)
Tejun Heoa4106ea2011-05-02 14:18:53 +0200474 numaram = 0;
475 }
476
Tejun Heo474b8812011-07-12 11:16:04 +0200477 e820ram = max_pfn - absent_pages_in_range(0, max_pfn);
478
Tejun Heoa4106ea2011-05-02 14:18:53 +0200479 /* We seem to lose 3 pages somewhere. Allow 1M of slack. */
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200480 if ((s64)(e820ram - numaram) >= (1 << (20 - PAGE_SHIFT))) {
481 printk(KERN_ERR "NUMA: nodes only cover %LuMB of your %LuMB e820 RAM. Not used.\n",
Tejun Heoa4106ea2011-05-02 14:18:53 +0200482 (numaram << PAGE_SHIFT) >> 20,
483 (e820ram << PAGE_SHIFT) >> 20);
484 return false;
485 }
486 return true;
487}
488
489static int __init numa_register_memblks(struct numa_meminfo *mi)
490{
Tejun Heo1e019792011-07-12 09:45:34 +0200491 unsigned long uninitialized_var(pfn_align);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200492 int i, nid;
493
494 /* Account for nodes with cpus and no memory */
495 node_possible_map = numa_nodes_parsed;
496 numa_nodemask_from_meminfo(&node_possible_map, mi);
497 if (WARN_ON(nodes_empty(node_possible_map)))
498 return -EINVAL;
499
Tejun Heo0608f702011-07-14 11:44:23 +0200500 for (i = 0; i < mi->nr_blks; i++) {
501 struct numa_memblk *mb = &mi->blk[i];
502 memblock_set_node(mb->start, mb->end - mb->start, mb->nid);
503 }
Tejun Heo1e019792011-07-12 09:45:34 +0200504
505 /*
506 * If sections array is gonna be used for pfn -> nid mapping, check
507 * whether its granularity is fine enough.
508 */
509#ifdef NODE_NOT_IN_PAGE_FLAGS
510 pfn_align = node_map_pfn_alignment();
511 if (pfn_align && pfn_align < PAGES_PER_SECTION) {
512 printk(KERN_WARNING "Node alignment %LuMB < min %LuMB, rejecting NUMA config\n",
513 PFN_PHYS(pfn_align) >> 20,
514 PFN_PHYS(PAGES_PER_SECTION) >> 20);
515 return -EINVAL;
516 }
517#endif
Tejun Heoa4106ea2011-05-02 14:18:53 +0200518 if (!numa_meminfo_cover_memory(mi))
519 return -EINVAL;
520
521 /* Finally register nodes. */
522 for_each_node_mask(nid, node_possible_map) {
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200523 u64 start = PFN_PHYS(max_pfn);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200524 u64 end = 0;
525
526 for (i = 0; i < mi->nr_blks; i++) {
527 if (nid != mi->blk[i].nid)
528 continue;
529 start = min(mi->blk[i].start, start);
530 end = max(mi->blk[i].end, end);
531 }
532
533 if (start < end)
Yinghai Lua56bca82011-05-02 17:24:49 +0200534 setup_node_data(nid, start, end);
Tejun Heoa4106ea2011-05-02 14:18:53 +0200535 }
536
Tejun Heo0608f702011-07-14 11:44:23 +0200537 /* Dump memblock with node info and return. */
538 memblock_dump_all();
Tejun Heoa4106ea2011-05-02 14:18:53 +0200539 return 0;
540}
Tejun Heoa4106ea2011-05-02 14:18:53 +0200541
Tejun Heo8db78cc2011-01-23 14:37:42 +0100542/*
543 * There are unfortunately some poorly designed mainboards around that
544 * only connect memory to a single CPU. This breaks the 1:1 cpu->node
545 * mapping. To avoid this fill in the mapping for all possible CPUs,
546 * as the number of CPUs is not known yet. We round robin the existing
547 * nodes.
548 */
Tejun Heo752d4f32011-05-02 17:24:48 +0200549static void __init numa_init_array(void)
Tejun Heo8db78cc2011-01-23 14:37:42 +0100550{
551 int rr, i;
552
553 rr = first_node(node_online_map);
554 for (i = 0; i < nr_cpu_ids; i++) {
555 if (early_cpu_to_node(i) != NUMA_NO_NODE)
556 continue;
557 numa_set_node(i, rr);
558 rr = next_node(rr, node_online_map);
559 if (rr == MAX_NUMNODES)
560 rr = first_node(node_online_map);
561 }
562}
563
Tejun Heoa4106ea2011-05-02 14:18:53 +0200564static int __init numa_init(int (*init_func)(void))
565{
566 int i;
567 int ret;
568
569 for (i = 0; i < MAX_LOCAL_APIC; i++)
570 set_apicid_to_node(i, NUMA_NO_NODE);
571
572 nodes_clear(numa_nodes_parsed);
573 nodes_clear(node_possible_map);
574 nodes_clear(node_online_map);
575 memset(&numa_meminfo, 0, sizeof(numa_meminfo));
Tejun Heo0608f702011-07-14 11:44:23 +0200576 WARN_ON(memblock_set_node(0, ULLONG_MAX, MAX_NUMNODES));
Tejun Heoa4106ea2011-05-02 14:18:53 +0200577 numa_reset_distance();
578
579 ret = init_func();
580 if (ret < 0)
581 return ret;
582 ret = numa_cleanup_meminfo(&numa_meminfo);
583 if (ret < 0)
584 return ret;
585
586 numa_emulation(&numa_meminfo, numa_distance_cnt);
587
588 ret = numa_register_memblks(&numa_meminfo);
589 if (ret < 0)
590 return ret;
591
592 for (i = 0; i < nr_cpu_ids; i++) {
593 int nid = early_cpu_to_node(i);
594
595 if (nid == NUMA_NO_NODE)
596 continue;
597 if (!node_online(nid))
598 numa_clear_node(i);
599 }
600 numa_init_array();
601 return 0;
602}
603
604/**
605 * dummy_numa_init - Fallback dummy NUMA init
606 *
607 * Used if there's no underlying NUMA architecture, NUMA initialization
608 * fails, or NUMA is disabled on the command line.
609 *
610 * Must online at least one node and add memory blocks that cover all
611 * allowed memory. This function must not fail.
612 */
613static int __init dummy_numa_init(void)
614{
615 printk(KERN_INFO "%s\n",
616 numa_off ? "NUMA turned off" : "No NUMA configuration found");
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200617 printk(KERN_INFO "Faking a node at %016Lx-%016Lx\n",
618 0LLU, PFN_PHYS(max_pfn));
Tejun Heoa4106ea2011-05-02 14:18:53 +0200619
620 node_set(0, numa_nodes_parsed);
Tejun Heo38f3e1c2011-05-02 14:18:53 +0200621 numa_add_memblk(0, 0, PFN_PHYS(max_pfn));
Tejun Heoa4106ea2011-05-02 14:18:53 +0200622
623 return 0;
624}
625
626/**
627 * x86_numa_init - Initialize NUMA
628 *
629 * Try each configured NUMA initialization method until one succeeds. The
630 * last fallback is dummy single node config encomapssing whole memory and
631 * never fails.
632 */
633void __init x86_numa_init(void)
634{
635 if (!numa_off) {
Tejun Heobd6709a2011-05-02 17:24:48 +0200636#ifdef CONFIG_X86_NUMAQ
637 if (!numa_init(numaq_numa_init))
638 return;
639#endif
Tejun Heoa4106ea2011-05-02 14:18:53 +0200640#ifdef CONFIG_ACPI_NUMA
641 if (!numa_init(x86_acpi_numa_init))
642 return;
643#endif
644#ifdef CONFIG_AMD_NUMA
645 if (!numa_init(amd_numa_init))
646 return;
647#endif
648 }
649
650 numa_init(dummy_numa_init);
651}
Tejun Heoa4106ea2011-05-02 14:18:53 +0200652
Tejun Heo8db78cc2011-01-23 14:37:42 +0100653static __init int find_near_online_node(int node)
654{
655 int n, val;
656 int min_val = INT_MAX;
657 int best_node = -1;
658
659 for_each_online_node(n) {
660 val = node_distance(node, n);
661
662 if (val < min_val) {
663 min_val = val;
664 best_node = n;
665 }
666 }
667
668 return best_node;
669}
670
671/*
672 * Setup early cpu_to_node.
673 *
674 * Populate cpu_to_node[] only if x86_cpu_to_apicid[],
675 * and apicid_to_node[] tables have valid entries for a CPU.
676 * This means we skip cpu_to_node[] initialisation for NUMA
677 * emulation and faking node case (when running a kernel compiled
678 * for NUMA on a non NUMA box), which is OK as cpu_to_node[]
679 * is already initialized in a round robin manner at numa_init_array,
680 * prior to this call, and this initialization is good enough
681 * for the fake NUMA cases.
682 *
683 * Called before the per_cpu areas are setup.
684 */
685void __init init_cpu_to_node(void)
686{
687 int cpu;
688 u16 *cpu_to_apicid = early_per_cpu_ptr(x86_cpu_to_apicid);
689
690 BUG_ON(cpu_to_apicid == NULL);
691
692 for_each_possible_cpu(cpu) {
693 int node = numa_cpu_node(cpu);
694
695 if (node == NUMA_NO_NODE)
696 continue;
697 if (!node_online(node))
698 node = find_near_online_node(node);
699 numa_set_node(cpu, node);
700 }
701}
702
Tejun Heode2d9442011-01-23 14:37:41 +0100703#ifndef CONFIG_DEBUG_PER_CPU_MAPS
704
705# ifndef CONFIG_NUMA_EMU
706void __cpuinit numa_add_cpu(int cpu)
707{
708 cpumask_set_cpu(cpu, node_to_cpumask_map[early_cpu_to_node(cpu)]);
709}
710
711void __cpuinit numa_remove_cpu(int cpu)
712{
713 cpumask_clear_cpu(cpu, node_to_cpumask_map[early_cpu_to_node(cpu)]);
714}
715# endif /* !CONFIG_NUMA_EMU */
716
717#else /* !CONFIG_DEBUG_PER_CPU_MAPS */
Tejun Heo645a7912011-01-23 14:37:40 +0100718
719int __cpu_to_node(int cpu)
720{
721 if (early_per_cpu_ptr(x86_cpu_to_node_map)) {
722 printk(KERN_WARNING
723 "cpu_to_node(%d): usage too early!\n", cpu);
724 dump_stack();
725 return early_per_cpu_ptr(x86_cpu_to_node_map)[cpu];
726 }
727 return per_cpu(x86_cpu_to_node_map, cpu);
728}
729EXPORT_SYMBOL(__cpu_to_node);
730
731/*
732 * Same function as cpu_to_node() but used if called before the
733 * per_cpu areas are setup.
734 */
735int early_cpu_to_node(int cpu)
736{
737 if (early_per_cpu_ptr(x86_cpu_to_node_map))
738 return early_per_cpu_ptr(x86_cpu_to_node_map)[cpu];
739
740 if (!cpu_possible(cpu)) {
741 printk(KERN_WARNING
742 "early_cpu_to_node(%d): no per_cpu area!\n", cpu);
743 dump_stack();
744 return NUMA_NO_NODE;
745 }
746 return per_cpu(x86_cpu_to_node_map, cpu);
747}
748
David Rientjes7a6c6542011-04-20 19:19:13 -0700749void debug_cpumask_set_cpu(int cpu, int node, bool enable)
Tejun Heode2d9442011-01-23 14:37:41 +0100750{
Tejun Heode2d9442011-01-23 14:37:41 +0100751 struct cpumask *mask;
752 char buf[64];
753
David Rientjes14392fd2011-02-07 14:08:53 -0800754 if (node == NUMA_NO_NODE) {
755 /* early_cpu_to_node() already emits a warning and trace */
David Rientjes7a6c6542011-04-20 19:19:13 -0700756 return;
David Rientjes14392fd2011-02-07 14:08:53 -0800757 }
Tejun Heode2d9442011-01-23 14:37:41 +0100758 mask = node_to_cpumask_map[node];
759 if (!mask) {
760 pr_err("node_to_cpumask_map[%i] NULL\n", node);
761 dump_stack();
Tejun Heode2d9442011-01-23 14:37:41 +0100762 return;
David Rientjes7a6c6542011-04-20 19:19:13 -0700763 }
Tejun Heode2d9442011-01-23 14:37:41 +0100764
765 if (enable)
766 cpumask_set_cpu(cpu, mask);
767 else
768 cpumask_clear_cpu(cpu, mask);
David Rientjes7a6c6542011-04-20 19:19:13 -0700769
770 cpulist_scnprintf(buf, sizeof(buf), mask);
771 printk(KERN_DEBUG "%s cpu %d node %d: mask now %s\n",
772 enable ? "numa_add_cpu" : "numa_remove_cpu",
773 cpu, node, buf);
774 return;
775}
776
777# ifndef CONFIG_NUMA_EMU
778static void __cpuinit numa_set_cpumask(int cpu, bool enable)
779{
780 debug_cpumask_set_cpu(cpu, early_cpu_to_node(cpu), enable);
Tejun Heode2d9442011-01-23 14:37:41 +0100781}
782
783void __cpuinit numa_add_cpu(int cpu)
784{
David Rientjes7a6c6542011-04-20 19:19:13 -0700785 numa_set_cpumask(cpu, true);
Tejun Heode2d9442011-01-23 14:37:41 +0100786}
787
788void __cpuinit numa_remove_cpu(int cpu)
789{
David Rientjes7a6c6542011-04-20 19:19:13 -0700790 numa_set_cpumask(cpu, false);
Tejun Heode2d9442011-01-23 14:37:41 +0100791}
792# endif /* !CONFIG_NUMA_EMU */
793
Rusty Russell71ee73e2009-03-13 14:49:52 +1030794/*
795 * Returns a pointer to the bitmask of CPUs on Node 'node'.
796 */
Rusty Russell73e907d2009-03-13 14:49:57 +1030797const struct cpumask *cpumask_of_node(int node)
Rusty Russell71ee73e2009-03-13 14:49:52 +1030798{
Rusty Russell71ee73e2009-03-13 14:49:52 +1030799 if (node >= nr_node_ids) {
800 printk(KERN_WARNING
801 "cpumask_of_node(%d): node > nr_node_ids(%d)\n",
802 node, nr_node_ids);
803 dump_stack();
804 return cpu_none_mask;
805 }
Rusty Russellc032ef602009-03-13 14:49:53 +1030806 if (node_to_cpumask_map[node] == NULL) {
807 printk(KERN_WARNING
808 "cpumask_of_node(%d): no node_to_cpumask_map!\n",
809 node);
810 dump_stack();
811 return cpu_online_mask;
812 }
Rusty Russell0b966252009-03-13 23:42:42 +1030813 return node_to_cpumask_map[node];
Rusty Russell71ee73e2009-03-13 14:49:52 +1030814}
815EXPORT_SYMBOL(cpumask_of_node);
Tejun Heo645a7912011-01-23 14:37:40 +0100816
Tejun Heode2d9442011-01-23 14:37:41 +0100817#endif /* !CONFIG_DEBUG_PER_CPU_MAPS */
Tejun Heoa4106ea2011-05-02 14:18:53 +0200818
Tejun Heobd6709a2011-05-02 17:24:48 +0200819#ifdef CONFIG_MEMORY_HOTPLUG
Tejun Heoa4106ea2011-05-02 14:18:53 +0200820int memory_add_physaddr_to_nid(u64 start)
821{
822 struct numa_meminfo *mi = &numa_meminfo;
823 int nid = mi->blk[0].nid;
824 int i;
825
826 for (i = 0; i < mi->nr_blks; i++)
827 if (mi->blk[i].start <= start && mi->blk[i].end > start)
828 nid = mi->blk[i].nid;
829 return nid;
830}
831EXPORT_SYMBOL_GPL(memory_add_physaddr_to_nid);
832#endif