xref: /linux/arch/riscv/kernel/acpi_numa.c (revision 53597deca0e38c30e6cd4ba2114fa42d2bcd85bb)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * ACPI 6.6 based NUMA setup for RISCV
4  * Lots of code was borrowed from arch/arm64/kernel/acpi_numa.c
5  *
6  * Copyright 2004 Andi Kleen, SuSE Labs.
7  * Copyright (C) 2013-2016, Linaro Ltd.
8  *		Author: Hanjun Guo <hanjun.guo@linaro.org>
9  * Copyright (C) 2024 Intel Corporation.
10  *
11  * Reads the ACPI SRAT table to figure out what memory belongs to which CPUs.
12  *
13  * Called from acpi_numa_init while reading the SRAT and SLIT tables.
14  * Assumes all memory regions belonging to a single proximity domain
15  * are in one chunk. Holes between them will be included in the node.
16  */
17 
18 #define pr_fmt(fmt) "ACPI: NUMA: " fmt
19 
20 #include <linux/acpi.h>
21 #include <linux/bitmap.h>
22 #include <linux/kernel.h>
23 #include <linux/mm.h>
24 #include <linux/memblock.h>
25 #include <linux/mmzone.h>
26 #include <linux/module.h>
27 #include <linux/topology.h>
28 
29 #include <asm/numa.h>
30 
31 static int acpi_early_node_map[NR_CPUS] __initdata = { [0 ... NR_CPUS - 1] = NUMA_NO_NODE };
32 
33 static int __init acpi_numa_get_nid(unsigned int cpu)
34 {
35 	return acpi_early_node_map[cpu];
36 }
37 
38 static inline int get_cpu_for_acpi_id(u32 uid)
39 {
40 	u32 cpu_uid;
41 	int ret;
42 
43 	for (int cpu = 0; cpu < nr_cpu_ids; cpu++) {
44 		ret = acpi_get_cpu_uid(cpu, &cpu_uid);
45 		if (ret == 0 && uid == cpu_uid)
46 			return cpu;
47 	}
48 
49 	return -EINVAL;
50 }
51 
52 static int __init acpi_parse_rintc_pxm(union acpi_subtable_headers *header,
53 				       const unsigned long end)
54 {
55 	struct acpi_srat_rintc_affinity *pa;
56 	int cpu, pxm, node;
57 
58 	if (srat_disabled())
59 		return -EINVAL;
60 
61 	pa = (struct acpi_srat_rintc_affinity *)header;
62 	if (!pa)
63 		return -EINVAL;
64 
65 	if (!(pa->flags & ACPI_SRAT_RINTC_ENABLED))
66 		return 0;
67 
68 	pxm = pa->proximity_domain;
69 	node = pxm_to_node(pxm);
70 
71 	/*
72 	 * If we can't map the UID to a logical cpu this
73 	 * means that the UID is not part of possible cpus
74 	 * so we do not need a NUMA mapping for it, skip
75 	 * the SRAT entry and keep parsing.
76 	 */
77 	cpu = get_cpu_for_acpi_id(pa->acpi_processor_uid);
78 	if (cpu < 0)
79 		return 0;
80 
81 	acpi_early_node_map[cpu] = node;
82 	pr_info("SRAT: PXM %d -> HARTID 0x%lx -> Node %d\n", pxm,
83 		cpuid_to_hartid_map(cpu), node);
84 
85 	return 0;
86 }
87 
88 void __init acpi_map_cpus_to_nodes(void)
89 {
90 	int i;
91 
92 	/*
93 	 * In ACPI, SMP and CPU NUMA information is provided in separate
94 	 * static tables, namely the MADT and the SRAT.
95 	 *
96 	 * Thus, it is simpler to first create the cpu logical map through
97 	 * an MADT walk and then map the logical cpus to their node ids
98 	 * as separate steps.
99 	 */
100 	acpi_table_parse_entries(ACPI_SIG_SRAT, sizeof(struct acpi_table_srat),
101 				 ACPI_SRAT_TYPE_RINTC_AFFINITY, acpi_parse_rintc_pxm, 0);
102 
103 	for (i = 0; i < nr_cpu_ids; i++)
104 		early_map_cpu_to_node(i, acpi_numa_get_nid(i));
105 }
106 
107 /* Callback for Proximity Domain -> logical node ID mapping */
108 void __init acpi_numa_rintc_affinity_init(struct acpi_srat_rintc_affinity *pa)
109 {
110 	int pxm, node;
111 
112 	if (srat_disabled())
113 		return;
114 
115 	if (pa->header.length < sizeof(struct acpi_srat_rintc_affinity)) {
116 		pr_err("SRAT: Invalid SRAT header length: %d\n", pa->header.length);
117 		bad_srat();
118 		return;
119 	}
120 
121 	if (!(pa->flags & ACPI_SRAT_RINTC_ENABLED))
122 		return;
123 
124 	pxm = pa->proximity_domain;
125 	node = acpi_map_pxm_to_node(pxm);
126 
127 	if (node == NUMA_NO_NODE) {
128 		pr_err("SRAT: Too many proximity domains %d\n", pxm);
129 		bad_srat();
130 		return;
131 	}
132 
133 	node_set(node, numa_nodes_parsed);
134 }
135