}
 }
 
-/* Returns nid in the range [0..MAX_NUMNODES-1], or -1 if no useful numa
+/*
+ * Returns nid in the range [0..nr_node_ids], or -1 if no useful NUMA
  * info is found.
  */
 static int associativity_to_nid(const __be32 *associativity)
                nid = of_read_number(&associativity[min_common_depth], 1);
 
        /* POWER4 LPAR uses 0xffff as invalid node */
-       if (nid == 0xffff || nid >= MAX_NUMNODES)
+       if (nid == 0xffff || nid >= nr_node_ids)
                nid = NUMA_NO_NODE;
 
        if (nid > 0 &&
                index = lmb->aa_index * aa.array_sz + min_common_depth - 1;
                nid = of_read_number(&aa.arrays[index], 1);
 
-               if (nid == 0xffff || nid >= MAX_NUMNODES)
+               if (nid == 0xffff || nid >= nr_node_ids)
                        nid = default_nid;
 
                if (nid > 0) {