cpusets: randomize node rotor used in cpuset_mem_spread_node()
[linux-2.6.git] / arch / x86 / mm / numa.c
1 /* Common code for 32 and 64-bit NUMA */
2 #include <linux/topology.h>
3 #include <linux/module.h>
4 #include <linux/bootmem.h>
5 #include <linux/random.h>
6
7 #ifdef CONFIG_DEBUG_PER_CPU_MAPS
8 # define DBG(x...) printk(KERN_DEBUG x)
9 #else
10 # define DBG(x...)
11 #endif
12
13 /*
14  * Which logical CPUs are on which nodes
15  */
16 cpumask_var_t node_to_cpumask_map[MAX_NUMNODES];
17 EXPORT_SYMBOL(node_to_cpumask_map);
18
19 /*
20  * Allocate node_to_cpumask_map based on number of available nodes
21  * Requires node_possible_map to be valid.
22  *
23  * Note: node_to_cpumask() is not valid until after this is done.
24  * (Use CONFIG_DEBUG_PER_CPU_MAPS to check this.)
25  */
26 void __init setup_node_to_cpumask_map(void)
27 {
28         unsigned int node, num = 0;
29
30         /* setup nr_node_ids if not done yet */
31         if (nr_node_ids == MAX_NUMNODES) {
32                 for_each_node_mask(node, node_possible_map)
33                         num = node;
34                 nr_node_ids = num + 1;
35         }
36
37         /* allocate the map */
38         for (node = 0; node < nr_node_ids; node++)
39                 alloc_bootmem_cpumask_var(&node_to_cpumask_map[node]);
40
41         /* cpumask_of_node() will now work */
42         pr_debug("Node to cpumask map for %d nodes\n", nr_node_ids);
43 }
44
45 #ifdef CONFIG_DEBUG_PER_CPU_MAPS
46 /*
47  * Returns a pointer to the bitmask of CPUs on Node 'node'.
48  */
49 const struct cpumask *cpumask_of_node(int node)
50 {
51         if (node >= nr_node_ids) {
52                 printk(KERN_WARNING
53                         "cpumask_of_node(%d): node > nr_node_ids(%d)\n",
54                         node, nr_node_ids);
55                 dump_stack();
56                 return cpu_none_mask;
57         }
58         if (node_to_cpumask_map[node] == NULL) {
59                 printk(KERN_WARNING
60                         "cpumask_of_node(%d): no node_to_cpumask_map!\n",
61                         node);
62                 dump_stack();
63                 return cpu_online_mask;
64         }
65         return node_to_cpumask_map[node];
66 }
67 EXPORT_SYMBOL(cpumask_of_node);
68 #endif
69
70 /*
71  * Return the bit number of a random bit set in the nodemask.
72  *   (returns -1 if nodemask is empty)
73  */
74 int __node_random(const nodemask_t *maskp)
75 {
76         int w, bit = -1;
77
78         w = nodes_weight(*maskp);
79         if (w)
80                 bit = bitmap_ord_to_pos(maskp->bits,
81                         get_random_int() % w, MAX_NUMNODES);
82         return bit;
83 }
84 EXPORT_SYMBOL(__node_random);