| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 1 | /* Common code for 32 and 64-bit NUMA */ | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 2 | #include <linux/kernel.h> | 
 | 3 | #include <linux/mm.h> | 
 | 4 | #include <linux/string.h> | 
 | 5 | #include <linux/init.h> | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 6 | #include <linux/bootmem.h> | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 7 | #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 Beulich | 9032160 | 2011-01-19 08:57:21 +0000 | [diff] [blame] | 18 | #include <asm/acpi.h> | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 19 | #include <asm/amd_nb.h> | 
 | 20 |  | 
 | 21 | #include "numa_internal.h" | 
| Jan Beulich | 9032160 | 2011-01-19 08:57:21 +0000 | [diff] [blame] | 22 |  | 
 | 23 | int __initdata numa_off; | 
| Tejun Heo | e6df595 | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 24 | nodemask_t numa_nodes_parsed __initdata; | 
| Jan Beulich | 9032160 | 2011-01-19 08:57:21 +0000 | [diff] [blame] | 25 |  | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 26 | struct pglist_data *node_data[MAX_NUMNODES] __read_mostly; | 
 | 27 | EXPORT_SYMBOL(node_data); | 
 | 28 |  | 
 | 29 | static struct numa_meminfo numa_meminfo | 
 | 30 | #ifndef CONFIG_MEMORY_HOTPLUG | 
 | 31 | __initdata | 
 | 32 | #endif | 
 | 33 | ; | 
 | 34 |  | 
 | 35 | static int numa_distance_cnt; | 
 | 36 | static u8 *numa_distance; | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 37 |  | 
| Jan Beulich | 9032160 | 2011-01-19 08:57:21 +0000 | [diff] [blame] | 38 | static __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 | } | 
 | 54 | early_param("numa", numa_setup); | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 55 |  | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 56 | /* | 
| Tejun Heo | bbc9e2f | 2011-01-23 14:37:39 +0100 | [diff] [blame] | 57 |  * apicid, cpu, node mappings | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 58 |  */ | 
| Tejun Heo | bbc9e2f | 2011-01-23 14:37:39 +0100 | [diff] [blame] | 59 | s16 __apicid_to_node[MAX_LOCAL_APIC] __cpuinitdata = { | 
 | 60 | 	[0 ... MAX_LOCAL_APIC-1] = NUMA_NO_NODE | 
 | 61 | }; | 
 | 62 |  | 
| Tejun Heo | 6bd2627 | 2011-05-02 14:18:52 +0200 | [diff] [blame] | 63 | int __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 Russell | c032ef60 | 2009-03-13 14:49:53 +1030 | [diff] [blame] | 72 | cpumask_var_t node_to_cpumask_map[MAX_NUMNODES]; | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 73 | EXPORT_SYMBOL(node_to_cpumask_map); | 
 | 74 |  | 
 | 75 | /* | 
| Tejun Heo | 645a791 | 2011-01-23 14:37:40 +0100 | [diff] [blame] | 76 |  * Map cpu index to node index | 
 | 77 |  */ | 
| Tejun Heo | 645a791 | 2011-01-23 14:37:40 +0100 | [diff] [blame] | 78 | DEFINE_EARLY_PER_CPU(int, x86_cpu_to_node_map, NUMA_NO_NODE); | 
| Tejun Heo | 645a791 | 2011-01-23 14:37:40 +0100 | [diff] [blame] | 79 | EXPORT_EARLY_PER_CPU_SYMBOL(x86_cpu_to_node_map); | 
 | 80 |  | 
 | 81 | void __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 |  | 
 | 104 | void __cpuinit numa_clear_node(int cpu) | 
 | 105 | { | 
 | 106 | 	numa_set_node(cpu, NUMA_NO_NODE); | 
 | 107 | } | 
 | 108 |  | 
 | 109 | /* | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 110 |  * Allocate node_to_cpumask_map based on number of available nodes | 
 | 111 |  * Requires node_possible_map to be valid. | 
 | 112 |  * | 
| Wanlong Gao | 9512938 | 2012-01-12 17:20:09 -0800 | [diff] [blame] | 113 |  * Note: cpumask_of_node() is not valid until after this is done. | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 114 |  * (Use CONFIG_DEBUG_PER_CPU_MAPS to check this.) | 
 | 115 |  */ | 
 | 116 | void __init setup_node_to_cpumask_map(void) | 
 | 117 | { | 
 | 118 | 	unsigned int node, num = 0; | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 119 |  | 
 | 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 Russell | c032ef60 | 2009-03-13 14:49:53 +1030 | [diff] [blame] | 128 | 	for (node = 0; node < nr_node_ids; node++) | 
 | 129 | 		alloc_bootmem_cpumask_var(&node_to_cpumask_map[node]); | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 130 |  | 
| Rusty Russell | c032ef60 | 2009-03-13 14:49:53 +1030 | [diff] [blame] | 131 | 	/* cpumask_of_node() will now work */ | 
 | 132 | 	pr_debug("Node to cpumask map for %d nodes\n", nr_node_ids); | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 133 | } | 
 | 134 |  | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 135 | static 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 |  */ | 
 | 169 | void __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 |  */ | 
 | 187 | int __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 Lu | a56bca8 | 2011-05-02 17:24:49 +0200 | [diff] [blame] | 192 | /* Initialize NODE_DATA for a node on the local memory */ | 
 | 193 | static void __init setup_node_data(int nid, u64 start, u64 end) | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 194 | { | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 195 | 	const size_t nd_size = roundup(sizeof(pg_data_t), PAGE_SIZE); | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 196 | 	bool remapped = false; | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 197 | 	u64 nd_pa; | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 198 | 	void *nd; | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 199 | 	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 Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 208 | 	/* initialize remap allocator before aligning to ZONE_ALIGN */ | 
 | 209 | 	init_alloc_remap(nid, start, end); | 
 | 210 |  | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 211 | 	start = roundup(start, ZONE_ALIGN); | 
 | 212 |  | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 213 | 	printk(KERN_INFO "Initmem setup node %d %016Lx-%016Lx\n", | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 214 | 	       nid, start, end); | 
 | 215 |  | 
 | 216 | 	/* | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 217 | 	 * Allocate node data.  Try remap allocator first, node-local | 
 | 218 | 	 * memory and then any node.  Never allocate in DMA zone. | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 219 | 	 */ | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 220 | 	nd = alloc_remap(nid, nd_size); | 
 | 221 | 	if (nd) { | 
 | 222 | 		nd_pa = __pa(nd); | 
 | 223 | 		remapped = true; | 
 | 224 | 	} else { | 
| Tejun Heo | eb40c4c | 2011-07-12 10:46:35 +0200 | [diff] [blame] | 225 | 		nd_pa = memblock_alloc_nid(nd_size, SMP_CACHE_BYTES, nid); | 
| Tejun Heo | 1f5026a | 2011-07-12 09:58:09 +0200 | [diff] [blame] | 226 | 		if (!nd_pa) { | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 227 | 			pr_err("Cannot find %zu bytes in node %d\n", | 
 | 228 | 			       nd_size, nid); | 
 | 229 | 			return; | 
 | 230 | 		} | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 231 | 		nd = __va(nd_pa); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 232 | 	} | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 233 |  | 
 | 234 | 	/* report and initialize */ | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 235 | 	printk(KERN_INFO "  NODE_DATA [%016Lx - %016Lx]%s\n", | 
 | 236 | 	       nd_pa, nd_pa + nd_size - 1, remapped ? " (remapped)" : ""); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 237 | 	tnid = early_pfn_to_nid(nd_pa >> PAGE_SHIFT); | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 238 | 	if (!remapped && tnid != nid) | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 239 | 		printk(KERN_INFO "    NODE_DATA(%d) on node %d\n", nid, tnid); | 
 | 240 |  | 
| Tejun Heo | 7888e96 | 2011-05-02 14:18:54 +0200 | [diff] [blame] | 241 | 	node_data[nid] = nd; | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 242 | 	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 |  */ | 
 | 260 | int __init numa_cleanup_meminfo(struct numa_meminfo *mi) | 
 | 261 | { | 
 | 262 | 	const u64 low = 0; | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 263 | 	const u64 high = PFN_PHYS(max_pfn); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 264 | 	int i, j, k; | 
 | 265 |  | 
| Yinghai Lu | e5a10c1 | 2011-05-02 17:24:49 +0200 | [diff] [blame] | 266 | 	/* first, trim all entries */ | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 267 | 	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 Lu | e5a10c1 | 2011-05-02 17:24:49 +0200 | [diff] [blame] | 275 | 		if (bi->start >= bi->end) | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 276 | 			numa_remove_memblk_from(i--, mi); | 
| Yinghai Lu | e5a10c1 | 2011-05-02 17:24:49 +0200 | [diff] [blame] | 277 | 	} | 
 | 278 |  | 
 | 279 | 	/* merge neighboring / overlapping entries */ | 
 | 280 | 	for (i = 0; i < mi->nr_blks; i++) { | 
 | 281 | 		struct numa_memblk *bi = &mi->blk[i]; | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 282 |  | 
 | 283 | 		for (j = i + 1; j < mi->nr_blks; j++) { | 
 | 284 | 			struct numa_memblk *bj = &mi->blk[j]; | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 285 | 			u64 start, end; | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 286 |  | 
 | 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 Lu | e5a10c1 | 2011-05-02 17:24:49 +0200 | [diff] [blame] | 311 | 			start = min(bi->start, bj->start); | 
 | 312 | 			end = max(bi->end, bj->end); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 313 | 			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 Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 323 | 			printk(KERN_INFO "NUMA: Node %d [%Lx,%Lx) + [%Lx,%Lx) -> [%Lx,%Lx)\n", | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 324 | 			       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 Lu | e5a10c1 | 2011-05-02 17:24:49 +0200 | [diff] [blame] | 332 | 	/* clear unused ones */ | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 333 | 	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 |  */ | 
 | 344 | static 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 |  */ | 
 | 361 | void __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 Heo | 24aa078 | 2011-07-12 11:16:06 +0200 | [diff] [blame] | 367 | 		memblock_free(__pa(numa_distance), size); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 368 | 	numa_distance_cnt = 0; | 
 | 369 | 	numa_distance = NULL;	/* enable table creation */ | 
 | 370 | } | 
 | 371 |  | 
 | 372 | static 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 Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 388 | 	phys = memblock_find_in_range(0, PFN_PHYS(max_pfn_mapped), | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 389 | 				      size, PAGE_SIZE); | 
| Tejun Heo | 1f5026a | 2011-07-12 09:58:09 +0200 | [diff] [blame] | 390 | 	if (!phys) { | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 391 | 		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 Heo | 24aa078 | 2011-07-12 11:16:06 +0200 | [diff] [blame] | 396 | 	memblock_reserve(phys, size); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 397 |  | 
 | 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 |  * | 
| Petr Holasek | 54eed6c | 2011-12-08 13:16:41 +0100 | [diff] [blame] | 425 |  * If @from or @to is higher than the highest known node or lower than zero | 
 | 426 |  * at the time of table creation or @distance doesn't make sense, the call | 
 | 427 |  * is ignored. | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 428 |  * This is to allow simplification of specific NUMA config implementations. | 
 | 429 |  */ | 
 | 430 | void __init numa_set_distance(int from, int to, int distance) | 
 | 431 | { | 
 | 432 | 	if (!numa_distance && numa_alloc_distance() < 0) | 
 | 433 | 		return; | 
 | 434 |  | 
| Petr Holasek | 54eed6c | 2011-12-08 13:16:41 +0100 | [diff] [blame] | 435 | 	if (from >= numa_distance_cnt || to >= numa_distance_cnt || | 
 | 436 | 			from < 0 || to < 0) { | 
 | 437 | 		pr_warn_once("NUMA: Warning: node ids are out of bound, from=%d to=%d distance=%d\n", | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 438 | 			    from, to, distance); | 
 | 439 | 		return; | 
 | 440 | 	} | 
 | 441 |  | 
 | 442 | 	if ((u8)distance != distance || | 
 | 443 | 	    (from == to && distance != LOCAL_DISTANCE)) { | 
 | 444 | 		pr_warn_once("NUMA: Warning: invalid distance parameter, from=%d to=%d distance=%d\n", | 
 | 445 | 			     from, to, distance); | 
 | 446 | 		return; | 
 | 447 | 	} | 
 | 448 |  | 
 | 449 | 	numa_distance[from * numa_distance_cnt + to] = distance; | 
 | 450 | } | 
 | 451 |  | 
 | 452 | int __node_distance(int from, int to) | 
 | 453 | { | 
 | 454 | 	if (from >= numa_distance_cnt || to >= numa_distance_cnt) | 
 | 455 | 		return from == to ? LOCAL_DISTANCE : REMOTE_DISTANCE; | 
 | 456 | 	return numa_distance[from * numa_distance_cnt + to]; | 
 | 457 | } | 
 | 458 | EXPORT_SYMBOL(__node_distance); | 
 | 459 |  | 
 | 460 | /* | 
 | 461 |  * Sanity check to catch more bad NUMA configurations (they are amazingly | 
 | 462 |  * common).  Make sure the nodes cover all memory. | 
 | 463 |  */ | 
 | 464 | static bool __init numa_meminfo_cover_memory(const struct numa_meminfo *mi) | 
 | 465 | { | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 466 | 	u64 numaram, e820ram; | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 467 | 	int i; | 
 | 468 |  | 
 | 469 | 	numaram = 0; | 
 | 470 | 	for (i = 0; i < mi->nr_blks; i++) { | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 471 | 		u64 s = mi->blk[i].start >> PAGE_SHIFT; | 
 | 472 | 		u64 e = mi->blk[i].end >> PAGE_SHIFT; | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 473 | 		numaram += e - s; | 
 | 474 | 		numaram -= __absent_pages_in_range(mi->blk[i].nid, s, e); | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 475 | 		if ((s64)numaram < 0) | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 476 | 			numaram = 0; | 
 | 477 | 	} | 
 | 478 |  | 
| Tejun Heo | 474b881 | 2011-07-12 11:16:04 +0200 | [diff] [blame] | 479 | 	e820ram = max_pfn - absent_pages_in_range(0, max_pfn); | 
 | 480 |  | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 481 | 	/* We seem to lose 3 pages somewhere. Allow 1M of slack. */ | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 482 | 	if ((s64)(e820ram - numaram) >= (1 << (20 - PAGE_SHIFT))) { | 
 | 483 | 		printk(KERN_ERR "NUMA: nodes only cover %LuMB of your %LuMB e820 RAM. Not used.\n", | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 484 | 		       (numaram << PAGE_SHIFT) >> 20, | 
 | 485 | 		       (e820ram << PAGE_SHIFT) >> 20); | 
 | 486 | 		return false; | 
 | 487 | 	} | 
 | 488 | 	return true; | 
 | 489 | } | 
 | 490 |  | 
 | 491 | static int __init numa_register_memblks(struct numa_meminfo *mi) | 
 | 492 | { | 
| Tejun Heo | 1e01979 | 2011-07-12 09:45:34 +0200 | [diff] [blame] | 493 | 	unsigned long uninitialized_var(pfn_align); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 494 | 	int i, nid; | 
 | 495 |  | 
 | 496 | 	/* Account for nodes with cpus and no memory */ | 
 | 497 | 	node_possible_map = numa_nodes_parsed; | 
 | 498 | 	numa_nodemask_from_meminfo(&node_possible_map, mi); | 
 | 499 | 	if (WARN_ON(nodes_empty(node_possible_map))) | 
 | 500 | 		return -EINVAL; | 
 | 501 |  | 
| Tejun Heo | 0608f70 | 2011-07-14 11:44:23 +0200 | [diff] [blame] | 502 | 	for (i = 0; i < mi->nr_blks; i++) { | 
 | 503 | 		struct numa_memblk *mb = &mi->blk[i]; | 
 | 504 | 		memblock_set_node(mb->start, mb->end - mb->start, mb->nid); | 
 | 505 | 	} | 
| Tejun Heo | 1e01979 | 2011-07-12 09:45:34 +0200 | [diff] [blame] | 506 |  | 
 | 507 | 	/* | 
 | 508 | 	 * If sections array is gonna be used for pfn -> nid mapping, check | 
 | 509 | 	 * whether its granularity is fine enough. | 
 | 510 | 	 */ | 
 | 511 | #ifdef NODE_NOT_IN_PAGE_FLAGS | 
 | 512 | 	pfn_align = node_map_pfn_alignment(); | 
 | 513 | 	if (pfn_align && pfn_align < PAGES_PER_SECTION) { | 
 | 514 | 		printk(KERN_WARNING "Node alignment %LuMB < min %LuMB, rejecting NUMA config\n", | 
 | 515 | 		       PFN_PHYS(pfn_align) >> 20, | 
 | 516 | 		       PFN_PHYS(PAGES_PER_SECTION) >> 20); | 
 | 517 | 		return -EINVAL; | 
 | 518 | 	} | 
 | 519 | #endif | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 520 | 	if (!numa_meminfo_cover_memory(mi)) | 
 | 521 | 		return -EINVAL; | 
 | 522 |  | 
 | 523 | 	/* Finally register nodes. */ | 
 | 524 | 	for_each_node_mask(nid, node_possible_map) { | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 525 | 		u64 start = PFN_PHYS(max_pfn); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 526 | 		u64 end = 0; | 
 | 527 |  | 
 | 528 | 		for (i = 0; i < mi->nr_blks; i++) { | 
 | 529 | 			if (nid != mi->blk[i].nid) | 
 | 530 | 				continue; | 
 | 531 | 			start = min(mi->blk[i].start, start); | 
 | 532 | 			end = max(mi->blk[i].end, end); | 
 | 533 | 		} | 
 | 534 |  | 
 | 535 | 		if (start < end) | 
| Yinghai Lu | a56bca8 | 2011-05-02 17:24:49 +0200 | [diff] [blame] | 536 | 			setup_node_data(nid, start, end); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 537 | 	} | 
 | 538 |  | 
| Tejun Heo | 0608f70 | 2011-07-14 11:44:23 +0200 | [diff] [blame] | 539 | 	/* Dump memblock with node info and return. */ | 
 | 540 | 	memblock_dump_all(); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 541 | 	return 0; | 
 | 542 | } | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 543 |  | 
| Tejun Heo | 8db78cc | 2011-01-23 14:37:42 +0100 | [diff] [blame] | 544 | /* | 
 | 545 |  * There are unfortunately some poorly designed mainboards around that | 
 | 546 |  * only connect memory to a single CPU. This breaks the 1:1 cpu->node | 
 | 547 |  * mapping. To avoid this fill in the mapping for all possible CPUs, | 
 | 548 |  * as the number of CPUs is not known yet. We round robin the existing | 
 | 549 |  * nodes. | 
 | 550 |  */ | 
| Tejun Heo | 752d4f3 | 2011-05-02 17:24:48 +0200 | [diff] [blame] | 551 | static void __init numa_init_array(void) | 
| Tejun Heo | 8db78cc | 2011-01-23 14:37:42 +0100 | [diff] [blame] | 552 | { | 
 | 553 | 	int rr, i; | 
 | 554 |  | 
 | 555 | 	rr = first_node(node_online_map); | 
 | 556 | 	for (i = 0; i < nr_cpu_ids; i++) { | 
 | 557 | 		if (early_cpu_to_node(i) != NUMA_NO_NODE) | 
 | 558 | 			continue; | 
 | 559 | 		numa_set_node(i, rr); | 
 | 560 | 		rr = next_node(rr, node_online_map); | 
 | 561 | 		if (rr == MAX_NUMNODES) | 
 | 562 | 			rr = first_node(node_online_map); | 
 | 563 | 	} | 
 | 564 | } | 
 | 565 |  | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 566 | static int __init numa_init(int (*init_func)(void)) | 
 | 567 | { | 
 | 568 | 	int i; | 
 | 569 | 	int ret; | 
 | 570 |  | 
 | 571 | 	for (i = 0; i < MAX_LOCAL_APIC; i++) | 
 | 572 | 		set_apicid_to_node(i, NUMA_NO_NODE); | 
 | 573 |  | 
 | 574 | 	nodes_clear(numa_nodes_parsed); | 
 | 575 | 	nodes_clear(node_possible_map); | 
 | 576 | 	nodes_clear(node_online_map); | 
 | 577 | 	memset(&numa_meminfo, 0, sizeof(numa_meminfo)); | 
| Tejun Heo | 0608f70 | 2011-07-14 11:44:23 +0200 | [diff] [blame] | 578 | 	WARN_ON(memblock_set_node(0, ULLONG_MAX, MAX_NUMNODES)); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 579 | 	numa_reset_distance(); | 
 | 580 |  | 
 | 581 | 	ret = init_func(); | 
 | 582 | 	if (ret < 0) | 
 | 583 | 		return ret; | 
 | 584 | 	ret = numa_cleanup_meminfo(&numa_meminfo); | 
 | 585 | 	if (ret < 0) | 
 | 586 | 		return ret; | 
 | 587 |  | 
 | 588 | 	numa_emulation(&numa_meminfo, numa_distance_cnt); | 
 | 589 |  | 
 | 590 | 	ret = numa_register_memblks(&numa_meminfo); | 
 | 591 | 	if (ret < 0) | 
 | 592 | 		return ret; | 
 | 593 |  | 
 | 594 | 	for (i = 0; i < nr_cpu_ids; i++) { | 
 | 595 | 		int nid = early_cpu_to_node(i); | 
 | 596 |  | 
 | 597 | 		if (nid == NUMA_NO_NODE) | 
 | 598 | 			continue; | 
 | 599 | 		if (!node_online(nid)) | 
 | 600 | 			numa_clear_node(i); | 
 | 601 | 	} | 
 | 602 | 	numa_init_array(); | 
 | 603 | 	return 0; | 
 | 604 | } | 
 | 605 |  | 
 | 606 | /** | 
 | 607 |  * dummy_numa_init - Fallback dummy NUMA init | 
 | 608 |  * | 
 | 609 |  * Used if there's no underlying NUMA architecture, NUMA initialization | 
 | 610 |  * fails, or NUMA is disabled on the command line. | 
 | 611 |  * | 
 | 612 |  * Must online at least one node and add memory blocks that cover all | 
 | 613 |  * allowed memory.  This function must not fail. | 
 | 614 |  */ | 
 | 615 | static int __init dummy_numa_init(void) | 
 | 616 | { | 
 | 617 | 	printk(KERN_INFO "%s\n", | 
 | 618 | 	       numa_off ? "NUMA turned off" : "No NUMA configuration found"); | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 619 | 	printk(KERN_INFO "Faking a node at %016Lx-%016Lx\n", | 
 | 620 | 	       0LLU, PFN_PHYS(max_pfn)); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 621 |  | 
 | 622 | 	node_set(0, numa_nodes_parsed); | 
| Tejun Heo | 38f3e1c | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 623 | 	numa_add_memblk(0, 0, PFN_PHYS(max_pfn)); | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 624 |  | 
 | 625 | 	return 0; | 
 | 626 | } | 
 | 627 |  | 
 | 628 | /** | 
 | 629 |  * x86_numa_init - Initialize NUMA | 
 | 630 |  * | 
 | 631 |  * Try each configured NUMA initialization method until one succeeds.  The | 
 | 632 |  * last fallback is dummy single node config encomapssing whole memory and | 
 | 633 |  * never fails. | 
 | 634 |  */ | 
 | 635 | void __init x86_numa_init(void) | 
 | 636 | { | 
 | 637 | 	if (!numa_off) { | 
| Tejun Heo | bd6709a | 2011-05-02 17:24:48 +0200 | [diff] [blame] | 638 | #ifdef CONFIG_X86_NUMAQ | 
 | 639 | 		if (!numa_init(numaq_numa_init)) | 
 | 640 | 			return; | 
 | 641 | #endif | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 642 | #ifdef CONFIG_ACPI_NUMA | 
 | 643 | 		if (!numa_init(x86_acpi_numa_init)) | 
 | 644 | 			return; | 
 | 645 | #endif | 
 | 646 | #ifdef CONFIG_AMD_NUMA | 
 | 647 | 		if (!numa_init(amd_numa_init)) | 
 | 648 | 			return; | 
 | 649 | #endif | 
 | 650 | 	} | 
 | 651 |  | 
 | 652 | 	numa_init(dummy_numa_init); | 
 | 653 | } | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 654 |  | 
| Tejun Heo | 8db78cc | 2011-01-23 14:37:42 +0100 | [diff] [blame] | 655 | static __init int find_near_online_node(int node) | 
 | 656 | { | 
 | 657 | 	int n, val; | 
 | 658 | 	int min_val = INT_MAX; | 
 | 659 | 	int best_node = -1; | 
 | 660 |  | 
 | 661 | 	for_each_online_node(n) { | 
 | 662 | 		val = node_distance(node, n); | 
 | 663 |  | 
 | 664 | 		if (val < min_val) { | 
 | 665 | 			min_val = val; | 
 | 666 | 			best_node = n; | 
 | 667 | 		} | 
 | 668 | 	} | 
 | 669 |  | 
 | 670 | 	return best_node; | 
 | 671 | } | 
 | 672 |  | 
 | 673 | /* | 
 | 674 |  * Setup early cpu_to_node. | 
 | 675 |  * | 
 | 676 |  * Populate cpu_to_node[] only if x86_cpu_to_apicid[], | 
 | 677 |  * and apicid_to_node[] tables have valid entries for a CPU. | 
 | 678 |  * This means we skip cpu_to_node[] initialisation for NUMA | 
 | 679 |  * emulation and faking node case (when running a kernel compiled | 
 | 680 |  * for NUMA on a non NUMA box), which is OK as cpu_to_node[] | 
 | 681 |  * is already initialized in a round robin manner at numa_init_array, | 
 | 682 |  * prior to this call, and this initialization is good enough | 
 | 683 |  * for the fake NUMA cases. | 
 | 684 |  * | 
 | 685 |  * Called before the per_cpu areas are setup. | 
 | 686 |  */ | 
 | 687 | void __init init_cpu_to_node(void) | 
 | 688 | { | 
 | 689 | 	int cpu; | 
 | 690 | 	u16 *cpu_to_apicid = early_per_cpu_ptr(x86_cpu_to_apicid); | 
 | 691 |  | 
 | 692 | 	BUG_ON(cpu_to_apicid == NULL); | 
 | 693 |  | 
 | 694 | 	for_each_possible_cpu(cpu) { | 
 | 695 | 		int node = numa_cpu_node(cpu); | 
 | 696 |  | 
 | 697 | 		if (node == NUMA_NO_NODE) | 
 | 698 | 			continue; | 
 | 699 | 		if (!node_online(node)) | 
 | 700 | 			node = find_near_online_node(node); | 
 | 701 | 		numa_set_node(cpu, node); | 
 | 702 | 	} | 
 | 703 | } | 
 | 704 |  | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 705 | #ifndef CONFIG_DEBUG_PER_CPU_MAPS | 
 | 706 |  | 
 | 707 | # ifndef CONFIG_NUMA_EMU | 
 | 708 | void __cpuinit numa_add_cpu(int cpu) | 
 | 709 | { | 
 | 710 | 	cpumask_set_cpu(cpu, node_to_cpumask_map[early_cpu_to_node(cpu)]); | 
 | 711 | } | 
 | 712 |  | 
 | 713 | void __cpuinit numa_remove_cpu(int cpu) | 
 | 714 | { | 
 | 715 | 	cpumask_clear_cpu(cpu, node_to_cpumask_map[early_cpu_to_node(cpu)]); | 
 | 716 | } | 
 | 717 | # endif	/* !CONFIG_NUMA_EMU */ | 
 | 718 |  | 
 | 719 | #else	/* !CONFIG_DEBUG_PER_CPU_MAPS */ | 
| Tejun Heo | 645a791 | 2011-01-23 14:37:40 +0100 | [diff] [blame] | 720 |  | 
 | 721 | int __cpu_to_node(int cpu) | 
 | 722 | { | 
 | 723 | 	if (early_per_cpu_ptr(x86_cpu_to_node_map)) { | 
 | 724 | 		printk(KERN_WARNING | 
 | 725 | 			"cpu_to_node(%d): usage too early!\n", cpu); | 
 | 726 | 		dump_stack(); | 
 | 727 | 		return early_per_cpu_ptr(x86_cpu_to_node_map)[cpu]; | 
 | 728 | 	} | 
 | 729 | 	return per_cpu(x86_cpu_to_node_map, cpu); | 
 | 730 | } | 
 | 731 | EXPORT_SYMBOL(__cpu_to_node); | 
 | 732 |  | 
 | 733 | /* | 
 | 734 |  * Same function as cpu_to_node() but used if called before the | 
 | 735 |  * per_cpu areas are setup. | 
 | 736 |  */ | 
 | 737 | int early_cpu_to_node(int cpu) | 
 | 738 | { | 
 | 739 | 	if (early_per_cpu_ptr(x86_cpu_to_node_map)) | 
 | 740 | 		return early_per_cpu_ptr(x86_cpu_to_node_map)[cpu]; | 
 | 741 |  | 
 | 742 | 	if (!cpu_possible(cpu)) { | 
 | 743 | 		printk(KERN_WARNING | 
 | 744 | 			"early_cpu_to_node(%d): no per_cpu area!\n", cpu); | 
 | 745 | 		dump_stack(); | 
 | 746 | 		return NUMA_NO_NODE; | 
 | 747 | 	} | 
 | 748 | 	return per_cpu(x86_cpu_to_node_map, cpu); | 
 | 749 | } | 
 | 750 |  | 
| David Rientjes | 7a6c654 | 2011-04-20 19:19:13 -0700 | [diff] [blame] | 751 | void debug_cpumask_set_cpu(int cpu, int node, bool enable) | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 752 | { | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 753 | 	struct cpumask *mask; | 
 | 754 | 	char buf[64]; | 
 | 755 |  | 
| David Rientjes | 14392fd | 2011-02-07 14:08:53 -0800 | [diff] [blame] | 756 | 	if (node == NUMA_NO_NODE) { | 
 | 757 | 		/* early_cpu_to_node() already emits a warning and trace */ | 
| David Rientjes | 7a6c654 | 2011-04-20 19:19:13 -0700 | [diff] [blame] | 758 | 		return; | 
| David Rientjes | 14392fd | 2011-02-07 14:08:53 -0800 | [diff] [blame] | 759 | 	} | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 760 | 	mask = node_to_cpumask_map[node]; | 
 | 761 | 	if (!mask) { | 
 | 762 | 		pr_err("node_to_cpumask_map[%i] NULL\n", node); | 
 | 763 | 		dump_stack(); | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 764 | 		return; | 
| David Rientjes | 7a6c654 | 2011-04-20 19:19:13 -0700 | [diff] [blame] | 765 | 	} | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 766 |  | 
 | 767 | 	if (enable) | 
 | 768 | 		cpumask_set_cpu(cpu, mask); | 
 | 769 | 	else | 
 | 770 | 		cpumask_clear_cpu(cpu, mask); | 
| David Rientjes | 7a6c654 | 2011-04-20 19:19:13 -0700 | [diff] [blame] | 771 |  | 
 | 772 | 	cpulist_scnprintf(buf, sizeof(buf), mask); | 
 | 773 | 	printk(KERN_DEBUG "%s cpu %d node %d: mask now %s\n", | 
 | 774 | 		enable ? "numa_add_cpu" : "numa_remove_cpu", | 
 | 775 | 		cpu, node, buf); | 
 | 776 | 	return; | 
 | 777 | } | 
 | 778 |  | 
 | 779 | # ifndef CONFIG_NUMA_EMU | 
 | 780 | static void __cpuinit numa_set_cpumask(int cpu, bool enable) | 
 | 781 | { | 
 | 782 | 	debug_cpumask_set_cpu(cpu, early_cpu_to_node(cpu), enable); | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 783 | } | 
 | 784 |  | 
 | 785 | void __cpuinit numa_add_cpu(int cpu) | 
 | 786 | { | 
| David Rientjes | 7a6c654 | 2011-04-20 19:19:13 -0700 | [diff] [blame] | 787 | 	numa_set_cpumask(cpu, true); | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 788 | } | 
 | 789 |  | 
 | 790 | void __cpuinit numa_remove_cpu(int cpu) | 
 | 791 | { | 
| David Rientjes | 7a6c654 | 2011-04-20 19:19:13 -0700 | [diff] [blame] | 792 | 	numa_set_cpumask(cpu, false); | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 793 | } | 
 | 794 | # endif	/* !CONFIG_NUMA_EMU */ | 
 | 795 |  | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 796 | /* | 
 | 797 |  * Returns a pointer to the bitmask of CPUs on Node 'node'. | 
 | 798 |  */ | 
| Rusty Russell | 73e907d | 2009-03-13 14:49:57 +1030 | [diff] [blame] | 799 | const struct cpumask *cpumask_of_node(int node) | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 800 | { | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 801 | 	if (node >= nr_node_ids) { | 
 | 802 | 		printk(KERN_WARNING | 
 | 803 | 			"cpumask_of_node(%d): node > nr_node_ids(%d)\n", | 
 | 804 | 			node, nr_node_ids); | 
 | 805 | 		dump_stack(); | 
 | 806 | 		return cpu_none_mask; | 
 | 807 | 	} | 
| Rusty Russell | c032ef60 | 2009-03-13 14:49:53 +1030 | [diff] [blame] | 808 | 	if (node_to_cpumask_map[node] == NULL) { | 
 | 809 | 		printk(KERN_WARNING | 
 | 810 | 			"cpumask_of_node(%d): no node_to_cpumask_map!\n", | 
 | 811 | 			node); | 
 | 812 | 		dump_stack(); | 
 | 813 | 		return cpu_online_mask; | 
 | 814 | 	} | 
| Rusty Russell | 0b96625 | 2009-03-13 23:42:42 +1030 | [diff] [blame] | 815 | 	return node_to_cpumask_map[node]; | 
| Rusty Russell | 71ee73e | 2009-03-13 14:49:52 +1030 | [diff] [blame] | 816 | } | 
 | 817 | EXPORT_SYMBOL(cpumask_of_node); | 
| Tejun Heo | 645a791 | 2011-01-23 14:37:40 +0100 | [diff] [blame] | 818 |  | 
| Tejun Heo | de2d944 | 2011-01-23 14:37:41 +0100 | [diff] [blame] | 819 | #endif	/* !CONFIG_DEBUG_PER_CPU_MAPS */ | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 820 |  | 
| Tejun Heo | bd6709a | 2011-05-02 17:24:48 +0200 | [diff] [blame] | 821 | #ifdef CONFIG_MEMORY_HOTPLUG | 
| Tejun Heo | a4106ea | 2011-05-02 14:18:53 +0200 | [diff] [blame] | 822 | int memory_add_physaddr_to_nid(u64 start) | 
 | 823 | { | 
 | 824 | 	struct numa_meminfo *mi = &numa_meminfo; | 
 | 825 | 	int nid = mi->blk[0].nid; | 
 | 826 | 	int i; | 
 | 827 |  | 
 | 828 | 	for (i = 0; i < mi->nr_blks; i++) | 
 | 829 | 		if (mi->blk[i].start <= start && mi->blk[i].end > start) | 
 | 830 | 			nid = mi->blk[i].nid; | 
 | 831 | 	return nid; | 
 | 832 | } | 
 | 833 | EXPORT_SYMBOL_GPL(memory_add_physaddr_to_nid); | 
 | 834 | #endif |