acpi/numa: memorize pmem node from SRAT table

euleros inclusion
category: feature
feature: etmem
bugzilla: 48246

-------------------------------------------------

Driver dax_kmem will export pmem as a NUMA node. This patch will
record node consists of persistent memory for futher use.
Signed-off-by: NKemeng Shi <shikemeng@huawei.com>
Reviewed-by: louhongxiang <louhongxiang@huawei.com
Signed-off-by: NZheng Zengkai <zhengzengkai@huawei.com>

drivers/acpi/numa/srat.c

@@ -279,6 +279,9 @@ acpi_numa_memory_affinity_init(struct acpi_srat_mem_affinity *ma)
 
 	node_set(node, numa_nodes_parsed);
 
+	if (ma->flags & ACPI_SRAT_MEM_NON_VOLATILE)
+		set_node_pmem(node);
+
 	pr_info("SRAT: Node %u PXM %u [mem %#010Lx-%#010Lx]%s%s\n",
 		node, pxm,
 		(unsigned long long) start, (unsigned long long) end - 1,

include/linux/numa.h

@@ -58,4 +58,7 @@ static inline int phys_to_target_node(u64 start)
 }
 #endif
 
+void set_node_pmem(int nid);
+int is_node_pmem(int nid);
+
 #endif /* _LINUX_NUMA_H */

mm/page_alloc.c

@@ -179,6 +179,8 @@ DEFINE_STATIC_KEY_FALSE(init_on_free);
 #endif
 EXPORT_SYMBOL(init_on_free);
 
+nodemask_t numa_nodes_pmem;
+
 static int __init early_init_on_alloc(char *buf)
 {
 	int ret;
@@ -8882,3 +8884,13 @@ bool take_page_off_buddy(struct page *page)
 	return ret;
 }
 #endif
+
+void set_node_pmem(int nid)
+{
+	node_set(nid, numa_nodes_pmem);
+}
+
+int is_node_pmem(int nid)
+{
+	return node_isset(nid, numa_nodes_pmem);
+}