mm/damon: add node_eligible_mem_bp goal metric

Background and Motivation
=========================

In heterogeneous memory systems, controlling memory distribution across
NUMA nodes is essential for performance optimization.  This patch enables
system-wide page distribution with target-state goals such as "maintain
60% of scheme-eligible memory on DRAM" using PA-mode DAMON schemes.

Rather than using absolute thresholds, this metric tracks the ratio of
memory that matches each scheme's access pattern filters on a target node,
enabling the quota system to automatically adjust migration aggressiveness
to maintain the desired distribution.

What This Metric Measures
=========================

node_eligible_mem_bp:
    scheme_eligible_bytes_on_node / total_scheme_eligible_bytes * 10000

Two-Scheme Setup for Hot Page Distribution
==========================================

For maintaining 60% of hot memory on DRAM (node 0) and 40% on CXL
(node 1):

    PULL scheme: migrate_hot to node 0
      goal: node_eligible_mem_bp, nid=0, target=6000
      addr filter: node 1 address range (only migrate FROM CXL)
      "Move hot pages to DRAM if less than 60% of hot data is in DRAM"

    PUSH scheme: migrate_hot to node 1
      goal: node_eligible_mem_bp, nid=1, target=4000
      addr filter: node 0 address range (only migrate FROM DRAM)
      "Move hot pages to CXL if less than 40% of hot data is in CXL"

Each scheme independently measures its own eligible memory and adjusts its
quota to achieve its target ratio.  The schemes work in concert through
DAMON's unified monitoring context, with the quota autotuner balancing
their relative aggressiveness.

Implementation Details
======================

The implementation adds a new quota goal metric type
DAMOS_QUOTA_NODE_ELIGIBLE_MEM_BP to the existing DAMOS quota goal
framework.  When this metric is configured for a scheme:

1. During each quota adjustment cycle, damos_get_node_eligible_mem_bp()
   is called to calculate the current memory distribution.

2. The function iterates through all regions that match the scheme's
   access pattern (via __damos_valid_target()) and calculates:
   - Total eligible bytes across all nodes
   - Eligible bytes specifically on the target node (goal->nid)

3. For each eligible region, damos_calc_eligible_bytes() walks through
   the physical address range, using damon_get_folio() to look up
   each folio and determine its NUMA node via folio_nid().

4. Large folios are handled by calculating the exact overlap between
   the region boundaries and folio boundaries, ensuring accurate
   byte counts even when regions partially span folios.

5. The ratio (node_eligible / total_eligible * 10000) is returned
   as basis points, which the quota autotuner uses to adjust the
   scheme's effective quota size (esz).

The implementation requires CONFIG_DAMON_PADDR since damon_get_folio()
is only available for physical address space monitoring.

Testing Results
===============

Functionally tested on a two-node heterogeneous memory system with DRAM
(node 0) and CXL memory (node 1).  A PUSH+PULL scheme configuration using
migrate_hot actions was used to reach a target hot memory ratio between
the two tiers.

With the TEMPORAL tuner, the system converges quickly to the target
distribution.  The tuner drives esz to maximum when under goal and to zero
once the goal is met, forming a simple on/off feedback loop that
stabilizes at the desired ratio.

With the CONSIST tuner, the scheme still converges but more slowly, as it
migrates and then throttles itself based on quota feedback.  The time to
reach the goal varies depending on workload intensity.

Note: This metric works with both TEMPORAL and CONSIST goal tuners.

Link: https://lore.kernel.org/20260428030520.701-1-ravis.opensrc@gmail.com
Signed-off-by: Ravi Jonnalagadda <ravis.opensrc@gmail.com>
Suggested-by: SeongJae Park <sj@kernel.org>
Reviewed-by: SeongJae Park <sj@kernel.org>
Cc: Honggyu Kim <honggyu.kim@sk.com>
Cc: Jonathan Corbet <corbet@lwn.net>
Cc: Yunjeong Mun <yunjeong.mun@sk.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>

diff --git a/include/linux/damon.h b/include/linux/damon.h
index f2cdb7c3f5e6ce688fd9214264e9ee242e9e95f2..986b8c90258557d66ae6735e79de308054c606b2 100644 (file)
--- a/include/linux/damon.h
+++ b/include/linux/damon.h
@@ -159,6 +159,8 @@ enum damos_action {
  * @DAMOS_QUOTA_NODE_MEMCG_FREE_BP:    MemFree ratio of a node for a cgroup.
  * @DAMOS_QUOTA_ACTIVE_MEM_BP:         Active to total LRU memory ratio.
  * @DAMOS_QUOTA_INACTIVE_MEM_BP:       Inactive to total LRU memory ratio.
+ * @DAMOS_QUOTA_NODE_ELIGIBLE_MEM_BP:  Scheme-eligible memory ratio of a
+ *                                     node in basis points (0-10000).
  * @NR_DAMOS_QUOTA_GOAL_METRICS:       Number of DAMOS quota goal metrics.
  *
  * Metrics equal to larger than @NR_DAMOS_QUOTA_GOAL_METRICS are unsupported.
@@ -172,6 +174,7 @@ enum damos_quota_goal_metric {
        DAMOS_QUOTA_NODE_MEMCG_FREE_BP,
        DAMOS_QUOTA_ACTIVE_MEM_BP,
        DAMOS_QUOTA_INACTIVE_MEM_BP,
+       DAMOS_QUOTA_NODE_ELIGIBLE_MEM_BP,
        NR_DAMOS_QUOTA_GOAL_METRICS,
 };
 

diff --git a/mm/damon/core.c b/mm/damon/core.c
index 901ffdaefb7f063f464abbb48cef6dac3bf7a82c..e4229294353ebc10e6ce13afa4c39d57c293865f 100644 (file)
--- a/mm/damon/core.c
+++ b/mm/damon/core.c
@@ -13,10 +13,14 @@
 #include <linux/memcontrol.h>
 #include <linux/mm.h>
 #include <linux/psi.h>
+#include <linux/sched.h>
 #include <linux/slab.h>
 #include <linux/string.h>
 #include <linux/string_choices.h>
 
+/* for damon_get_folio() used by node eligible memory metrics */
+#include "ops-common.h"
+
 #define CREATE_TRACE_POINTS
 #include <trace/events/damon.h>
 
@@ -1326,11 +1330,26 @@ static int damon_commit_targets(
 int damon_commit_ctx(struct damon_ctx *dst, struct damon_ctx *src)
 {
        int err;
+       struct damos *scheme;
+       struct damos_quota_goal *goal;
 
        dst->maybe_corrupted = true;
        if (!is_power_of_2(src->min_region_sz))
                return -EINVAL;
 
+       /* node_eligible_mem_bp metric requires PADDR ops */
+       if (src->ops.id != DAMON_OPS_PADDR) {
+               damon_for_each_scheme(scheme, src) {
+                       struct damos_quota *quota = &scheme->quota;
+
+                       damos_for_each_quota_goal(goal, quota) {
+                               if (goal->metric ==
+                                               DAMOS_QUOTA_NODE_ELIGIBLE_MEM_BP)
+                                       return -EINVAL;
+                       }
+               }
+       }
+
        err = damon_commit_schemes(dst, src);
        if (err)
                return err;
@@ -2287,7 +2306,115 @@ static unsigned long damos_get_node_memcg_used_bp(
                numerator = i.totalram - used_pages;
        return mult_frac(numerator, 10000, i.totalram);
 }
-#else
+
+#ifdef CONFIG_DAMON_PADDR
+/*
+ * damos_calc_eligible_bytes() - Calculate raw eligible bytes per node.
+ * @c:         The DAMON context.
+ * @s:         The scheme.
+ * @nid:       The target NUMA node id.
+ * @total:     Output for total eligible bytes across all nodes.
+ *
+ * Iterates through each folio in eligible regions to accurately determine
+ * which node the memory resides on. Returns eligible bytes on the specified
+ * node and sets *total to the sum across all nodes.
+ *
+ * Note: This function requires damon_get_folio() from ops-common.c, which is
+ * only available when CONFIG_DAMON_PADDR is enabled. It also requires the
+ * context to be using PADDR operations for meaningful results.
+ */
+static phys_addr_t damos_calc_eligible_bytes(struct damon_ctx *c,
+               struct damos *s, int nid, phys_addr_t *total)
+{
+       struct damon_target *t;
+       struct damon_region *r;
+       phys_addr_t total_eligible = 0;
+       phys_addr_t node_eligible = 0;
+
+       damon_for_each_target(t, c) {
+               damon_for_each_region(r, t) {
+                       phys_addr_t addr, end_addr;
+
+                       if (!__damos_valid_target(r, s))
+                               continue;
+
+                       /* Convert from core address units to physical bytes */
+                       addr = (phys_addr_t)r->ar.start * c->addr_unit;
+                       end_addr = (phys_addr_t)r->ar.end * c->addr_unit;
+                       while (addr < end_addr) {
+                               struct folio *folio;
+                               phys_addr_t folio_start, folio_end;
+                               phys_addr_t overlap_start, overlap_end;
+                               phys_addr_t counted;
+
+                               folio = damon_get_folio(PHYS_PFN(addr));
+                               if (!folio) {
+                                       addr = PAGE_ALIGN_DOWN(addr +
+                                                       PAGE_SIZE);
+                                       if (!addr)
+                                               break;
+                                       continue;
+                               }
+
+                               /*
+                                * Calculate exact overlap between the region
+                                * [addr, end_addr) and the folio range.
+                                * The folio may start before addr if addr is
+                                * in the middle of a large folio.
+                                */
+                               folio_start = PFN_PHYS(folio_pfn(folio));
+                               folio_end = folio_start + folio_size(folio);
+
+                               overlap_start = max(addr, folio_start);
+                               overlap_end = min(end_addr, folio_end);
+
+                               if (overlap_end > overlap_start) {
+                                       counted = overlap_end - overlap_start;
+                                       total_eligible += counted;
+                                       if (folio_nid(folio) == nid)
+                                               node_eligible += counted;
+                               }
+
+                               /* Advance past the entire folio */
+                               addr = folio_end;
+                               folio_put(folio);
+                       }
+                       cond_resched();
+               }
+       }
+
+       *total = total_eligible;
+       return node_eligible;
+}
+
+static unsigned long damos_get_node_eligible_mem_bp(struct damon_ctx *c,
+               struct damos *s, int nid)
+{
+       phys_addr_t total_eligible = 0;
+       phys_addr_t node_eligible;
+
+       if (c->ops.id != DAMON_OPS_PADDR)
+               return 0;
+
+       if (nid < 0 || nid >= MAX_NUMNODES || !node_online(nid))
+               return 0;
+
+       node_eligible = damos_calc_eligible_bytes(c, s, nid, &total_eligible);
+
+       if (!(unsigned long)total_eligible)
+               return 0;
+
+       return mult_frac((unsigned long)node_eligible, 10000,
+                       (unsigned long)total_eligible);
+}
+#else /* CONFIG_DAMON_PADDR */
+static unsigned long damos_get_node_eligible_mem_bp(struct damon_ctx *c,
+               struct damos *s, int nid)
+{
+       return 0;
+}
+#endif /* CONFIG_DAMON_PADDR */
+#else /* CONFIG_NUMA */
 static __kernel_ulong_t damos_get_node_mem_bp(
                struct damos_quota_goal *goal)
 {
@@ -2299,7 +2426,13 @@ static unsigned long damos_get_node_memcg_used_bp(
 {
        return 0;
 }
-#endif
+
+static unsigned long damos_get_node_eligible_mem_bp(struct damon_ctx *c,
+               struct damos *s, int nid)
+{
+       return 0;
+}
+#endif /* CONFIG_NUMA */
 
 /*
  * Returns LRU-active or inactive memory to total LRU memory size ratio.
@@ -2319,7 +2452,8 @@ static unsigned int damos_get_in_active_mem_bp(bool active_ratio)
        return mult_frac(inactive, 10000, total);
 }
 
-static void damos_set_quota_goal_current_value(struct damos_quota_goal *goal)
+static void damos_set_quota_goal_current_value(struct damon_ctx *c,
+               struct damos *s, struct damos_quota_goal *goal)
 {
        u64 now_psi_total;
 
@@ -2345,19 +2479,24 @@ static void damos_set_quota_goal_current_value(struct damos_quota_goal *goal)
                goal->current_value = damos_get_in_active_mem_bp(
                                goal->metric == DAMOS_QUOTA_ACTIVE_MEM_BP);
                break;
+       case DAMOS_QUOTA_NODE_ELIGIBLE_MEM_BP:
+               goal->current_value = damos_get_node_eligible_mem_bp(c, s,
+                               goal->nid);
+               break;
        default:
                break;
        }
 }
 
 /* Return the highest score since it makes schemes least aggressive */
-static unsigned long damos_quota_score(struct damos_quota *quota)
+static unsigned long damos_quota_score(struct damon_ctx *c, struct damos *s)
 {
        struct damos_quota_goal *goal;
+       struct damos_quota *quota = &s->quota;
        unsigned long highest_score = 0;
 
        damos_for_each_quota_goal(goal, quota) {
-               damos_set_quota_goal_current_value(goal);
+               damos_set_quota_goal_current_value(c, s, goal);
                highest_score = max(highest_score,
                                mult_frac(goal->current_value, 10000,
                                        goal->target_value));
@@ -2366,17 +2505,20 @@ static unsigned long damos_quota_score(struct damos_quota *quota)
        return highest_score;
 }
 
-static void damos_goal_tune_esz_bp_consist(struct damos_quota *quota)
+static void damos_goal_tune_esz_bp_consist(struct damon_ctx *c, struct damos *s)
 {
-       unsigned long score = damos_quota_score(quota);
+       struct damos_quota *quota = &s->quota;
+       unsigned long score = damos_quota_score(c, s);
 
        quota->esz_bp = damon_feed_loop_next_input(
                        max(quota->esz_bp, 10000UL), score);
 }
 
-static void damos_goal_tune_esz_bp_temporal(struct damos_quota *quota)
+static void damos_goal_tune_esz_bp_temporal(struct damon_ctx *c,
+               struct damos *s)
 {
-       unsigned long score = damos_quota_score(quota);
+       struct damos_quota *quota = &s->quota;
+       unsigned long score = damos_quota_score(c, s);
 
        if (score >= 10000)
                quota->esz_bp = 0;
@@ -2389,9 +2531,9 @@ static void damos_goal_tune_esz_bp_temporal(struct damos_quota *quota)
 /*
  * Called only if quota->ms, or quota->sz are set, or quota->goals is not empty
  */
-static void damos_set_effective_quota(struct damos_quota *quota,
-               struct damon_ctx *ctx)
+static void damos_set_effective_quota(struct damon_ctx *ctx, struct damos *s)
 {
+       struct damos_quota *quota = &s->quota;
        unsigned long throughput;
        unsigned long esz = ULONG_MAX;
 
@@ -2402,9 +2544,9 @@ static void damos_set_effective_quota(struct damos_quota *quota,
 
        if (!list_empty(&quota->goals)) {
                if (quota->goal_tuner == DAMOS_QUOTA_GOAL_TUNER_CONSIST)
-                       damos_goal_tune_esz_bp_consist(quota);
+                       damos_goal_tune_esz_bp_consist(ctx, s);
                else if (quota->goal_tuner == DAMOS_QUOTA_GOAL_TUNER_TEMPORAL)
-                       damos_goal_tune_esz_bp_temporal(quota);
+                       damos_goal_tune_esz_bp_temporal(ctx, s);
                esz = quota->esz_bp / 10000;
        }
 
@@ -2452,7 +2594,7 @@ static void damos_adjust_quota(struct damon_ctx *c, struct damos *s)
        /* First charge window */
        if (!quota->total_charged_sz && !quota->charged_from) {
                quota->charged_from = jiffies;
-               damos_set_effective_quota(quota, c);
+               damos_set_effective_quota(c, s);
        }
 
        /* New charge window starts */
@@ -2467,7 +2609,7 @@ static void damos_adjust_quota(struct damon_ctx *c, struct damos *s)
                quota->charged_sz = 0;
                if (trace_damos_esz_enabled())
                        cached_esz = quota->esz;
-               damos_set_effective_quota(quota, c);
+               damos_set_effective_quota(c, s);
                if (trace_damos_esz_enabled() && quota->esz != cached_esz)
                        damos_trace_esz(c, s, quota);
        }

diff --git a/mm/damon/sysfs-schemes.c b/mm/damon/sysfs-schemes.c
index a8014780edae98e4e94ce46d1f24cce7aff58326..d12e741a47ec8bb0b2f54cd27bc582d6a8e7cea2 100644 (file)
--- a/mm/damon/sysfs-schemes.c
+++ b/mm/damon/sysfs-schemes.c
@@ -1093,6 +1093,10 @@ struct damos_sysfs_qgoal_metric_name damos_sysfs_qgoal_metric_names[] = {
                .metric = DAMOS_QUOTA_INACTIVE_MEM_BP,
                .name = "inactive_mem_bp",
        },
+       {
+               .metric = DAMOS_QUOTA_NODE_ELIGIBLE_MEM_BP,
+               .name = "node_eligible_mem_bp",
+       },
 };
 
 static ssize_t target_metric_show(struct kobject *kobj,
@@ -2685,6 +2689,9 @@ static int damos_sysfs_add_quota_score(
                        }
                        goal->nid = sysfs_goal->nid;
                        break;
+               case DAMOS_QUOTA_NODE_ELIGIBLE_MEM_BP:
+                       goal->nid = sysfs_goal->nid;
+                       break;
                default:
                        break;
                }