Re: [RFC 3/4] acpi, hmat: calculate abstract distance with HMAT

From: Aneesh Kumar K.V
Date: Mon Jun 19 2023 - 12:24:36 EST


Huang Ying <ying.huang@xxxxxxxxx> writes:

> A memory tiering abstract distance calculation algorithm based on ACPI
> HMAT is implemented. The basic idea is as follows.
>
> The performance attributes of system default DRAM nodes are recorded
> as the base line. Whose abstract distance is MEMTIER_ADISTANCE_DRAM.
> Then, the ratio of the abstract distance of a memory node (target) to
> MEMTIER_ADISTANCE_DRAM is scaled based on the ratio of the performance
> attributes of the node to that of the default DRAM nodes.
>
> Signed-off-by: "Huang, Ying" <ying.huang@xxxxxxxxx>
> Cc: Aneesh Kumar K.V <aneesh.kumar@xxxxxxxxxxxxx>
> Cc: Wei Xu <weixugc@xxxxxxxxxx>
> Cc: Alistair Popple <apopple@xxxxxxxxxx>
> Cc: Dan Williams <dan.j.williams@xxxxxxxxx>
> Cc: Dave Hansen <dave.hansen@xxxxxxxxx>
> Cc: Davidlohr Bueso <dave@xxxxxxxxxxxx>
> Cc: Johannes Weiner <hannes@xxxxxxxxxxx>
> Cc: Jonathan Cameron <Jonathan.Cameron@xxxxxxxxxx>
> Cc: Michal Hocko <mhocko@xxxxxxxxxx>
> Cc: Yang Shi <shy828301@xxxxxxxxx>
> Cc: Rafael J Wysocki <rafael.j.wysocki@xxxxxxxxx>
> ---
> drivers/acpi/numa/hmat.c | 124 ++++++++++++++++++++++++++++++++++-
> include/linux/memory-tiers.h | 2 +
> mm/memory-tiers.c | 2 +-
> 3 files changed, 126 insertions(+), 2 deletions(-)
>
> diff --git a/drivers/acpi/numa/hmat.c b/drivers/acpi/numa/hmat.c
> index 2dee0098f1a9..21e4deb581ad 100644
> --- a/drivers/acpi/numa/hmat.c
> +++ b/drivers/acpi/numa/hmat.c
> @@ -24,6 +24,7 @@
> #include <linux/node.h>
> #include <linux/sysfs.h>
> #include <linux/dax.h>
> +#include <linux/memory-tiers.h>
>
> static u8 hmat_revision;
> static int hmat_disable __initdata;
> @@ -759,6 +760,123 @@ static int hmat_callback(struct notifier_block *self,
> return NOTIFY_OK;
> }
>
> +static int hmat_adistance_disabled;
> +static struct node_hmem_attrs default_dram_attrs;
> +
> +static void dump_hmem_attrs(struct node_hmem_attrs *attrs)
> +{
> + pr_cont("read_latency: %u, write_latency: %u, read_bandwidth: %u, write_bandwidth: %u\n",
> + attrs->read_latency, attrs->write_latency,
> + attrs->read_bandwidth, attrs->write_bandwidth);
> +}
> +
> +static void disable_hmat_adistance_algorithm(void)
> +{
> + hmat_adistance_disabled = true;
> +}
> +
> +static int hmat_init_default_dram_attrs(void)
> +{
> + struct memory_target *target;
> + struct node_hmem_attrs *attrs;
> + int nid, pxm;
> + int nid_dram = NUMA_NO_NODE;
> +
> + if (default_dram_attrs.read_latency +
> + default_dram_attrs.write_latency != 0)
> + return 0;
> +
> + if (!default_dram_type)
> + return -EIO;
> +
> + for_each_node_mask(nid, default_dram_type->nodes) {
> + pxm = node_to_pxm(nid);
> + target = find_mem_target(pxm);
> + if (!target)
> + continue;
> + attrs = &target->hmem_attrs[1];
> + if (nid_dram == NUMA_NO_NODE) {
> + if (attrs->read_latency + attrs->write_latency == 0 ||
> + attrs->read_bandwidth + attrs->write_bandwidth == 0) {
> + pr_info("hmat: invalid hmem attrs for default DRAM node: %d,\n",
> + nid);
> + pr_info(" ");
> + dump_hmem_attrs(attrs);
> + pr_info(" disable hmat based abstract distance algorithm.\n");
> + disable_hmat_adistance_algorithm();
> + return -EIO;
> + }
> + nid_dram = nid;
> + default_dram_attrs = *attrs;
> + continue;
> + }
> + if (abs(attrs->read_latency - default_dram_attrs.read_latency) * 10 >
> + default_dram_attrs.read_latency ||
> + abs(attrs->write_latency - default_dram_attrs.write_latency) * 10 >
> + default_dram_attrs.write_latency ||
> + abs(attrs->read_bandwidth - default_dram_attrs.read_bandwidth) * 10 >
> + default_dram_attrs.read_bandwidth) {
> + pr_info("hmat: hmem attrs for DRAM nodes mismatch.\n");
> + pr_info(" node %d:", nid_dram);
> + dump_hmem_attrs(&default_dram_attrs);
> + pr_info(" node %d:", nid);
> + dump_hmem_attrs(attrs);
> + pr_info(" disable hmat based abstract distance algorithm.\n");
> + disable_hmat_adistance_algorithm();
> + return -EIO;
> + }

What is this check about? what is the significance of 10? Can you add
the details as a code comment ?


> + }
> +
> + return 0;
> +}
> +
> +static int hmat_calculate_adistance(struct notifier_block *self,
> + unsigned long nid, void *data)
> +{
> + static DECLARE_BITMAP(p_nodes, MAX_NUMNODES);
> + struct memory_target *target;
> + struct node_hmem_attrs *attrs;
> + int *adist = data;
> + int pxm;
> +
> + if (hmat_adistance_disabled)
> + return NOTIFY_OK;
> +
> + pxm = node_to_pxm(nid);
> + target = find_mem_target(pxm);
> + if (!target)
> + return NOTIFY_OK;
> +
> + if (hmat_init_default_dram_attrs())
> + return NOTIFY_OK;
> +
> + mutex_lock(&target_lock);
> + hmat_update_target_attrs(target, p_nodes, 1);
> + mutex_unlock(&target_lock);
> +
> + attrs = &target->hmem_attrs[1];
> +
> + if (attrs->read_latency + attrs->write_latency == 0 ||
> + attrs->read_bandwidth + attrs->write_bandwidth == 0)
> + return NOTIFY_OK;
> +
> + *adist = MEMTIER_ADISTANCE_DRAM *
> + (attrs->read_latency + attrs->write_latency) /
> + (default_dram_attrs.read_latency +
> + default_dram_attrs.write_latency) *
> + (default_dram_attrs.read_bandwidth +
> + default_dram_attrs.write_bandwidth) /
> + (attrs->read_bandwidth + attrs->write_bandwidth);


Can you write a comment describing how we use all these attributes in
deriving the abstract distance value?


> +
> + return NOTIFY_STOP;
> +}
> +
> +static __meminitdata struct notifier_block hmat_adist_nb =
> +{
> + .notifier_call = hmat_calculate_adistance,
> + .priority = 100,
> +};
> +
> static __init void hmat_free_structures(void)
> {
> struct memory_target *target, *tnext;
> @@ -801,6 +919,7 @@ static __init int hmat_init(void)
> struct acpi_table_header *tbl;
> enum acpi_hmat_type i;
> acpi_status status;
> + int usage;
>
> if (srat_disabled() || hmat_disable)
> return 0;
> @@ -841,8 +960,11 @@ static __init int hmat_init(void)
> hmat_register_targets();
>
> /* Keep the table and structures if the notifier may use them */
> - if (!hotplug_memory_notifier(hmat_callback, HMAT_CALLBACK_PRI))
> + usage = !hotplug_memory_notifier(hmat_callback, HMAT_CALLBACK_PRI);
> + usage += !register_mt_adistance_algorithm(&hmat_adist_nb);
> + if (usage)
> return 0;
> +
> out_put:
> hmat_free_structures();
> acpi_put_table(tbl);
> diff --git a/include/linux/memory-tiers.h b/include/linux/memory-tiers.h
> index c6429e624244..9377239c8d34 100644
> --- a/include/linux/memory-tiers.h
> +++ b/include/linux/memory-tiers.h
> @@ -33,6 +33,7 @@ struct memory_dev_type {
>
> #ifdef CONFIG_NUMA
> extern bool numa_demotion_enabled;
> +extern struct memory_dev_type *default_dram_type;
> struct memory_dev_type *alloc_memory_type(int adistance);
> void destroy_memory_type(struct memory_dev_type *memtype);
> void init_node_memory_type(int node, struct memory_dev_type *default_type);
> @@ -64,6 +65,7 @@ static inline bool node_is_toptier(int node)
> #else
>
> #define numa_demotion_enabled false
> +#define default_dram_type NULL
> /*
> * CONFIG_NUMA implementation returns non NULL error.
> */
> diff --git a/mm/memory-tiers.c b/mm/memory-tiers.c
> index fb5398e710cc..3aabc7240402 100644
> --- a/mm/memory-tiers.c
> +++ b/mm/memory-tiers.c
> @@ -37,7 +37,7 @@ struct node_memory_type_map {
> static DEFINE_MUTEX(memory_tier_lock);
> static LIST_HEAD(memory_tiers);
> static struct node_memory_type_map node_memory_types[MAX_NUMNODES];
> -static struct memory_dev_type *default_dram_type;
> +struct memory_dev_type *default_dram_type;
>
> static struct bus_type memory_tier_subsys = {
> .name = "memory_tiering",
> --
> 2.39.2