From: SeongJae Park <sjp...@amazon.de>

This commit implements callbacks for DAMON.  Using this, DAMON users can
install their callbacks for each step of the access monitoring so that
they can do something interesting with the monitored access patterns
online.  For example, callbacks can report the monitored patterns to
users or do some access pattern based memory management such as
proactive reclamations or access pattern based THP promotions/demotions
decision makings.

Signed-off-by: SeongJae Park <sjp...@amazon.de>
---
 include/linux/damon.h | 13 +++++++++++++
 mm/damon.c            |  4 ++++
 2 files changed, 17 insertions(+)

diff --git a/include/linux/damon.h b/include/linux/damon.h
index 7d59f57028ca..e77518dd7f0d 100644
--- a/include/linux/damon.h
+++ b/include/linux/damon.h
@@ -80,6 +80,15 @@ struct damon_task {
  * @kdamond_lock.  Accesses to other fields must be protected by themselves.
  *
  * @tasks_list:                Head of monitring target tasks (&damon_task) 
list.
+ *
+ * @sample_cb:                 Called for each sampling interval.
+ * @aggregate_cb:              Called for each aggregation interval.
+ *
+ * @sample_cb and @aggregate_cb are called from @kdamond for each of the
+ * sampling intervals and aggregation intervals, respectively.  Therefore,
+ * users can safely access to the monitoring results via @tasks_list without
+ * additional protection of @kdamond_lock.  For the reason, users are
+ * recommended to use these callback for the accesses to the results.
  */
 struct damon_ctx {
        unsigned long sample_interval;
@@ -96,6 +105,10 @@ struct damon_ctx {
        struct mutex kdamond_lock;
 
        struct list_head tasks_list;    /* 'damon_task' objects */
+
+       /* callbacks */
+       void (*sample_cb)(struct damon_ctx *context);
+       void (*aggregate_cb)(struct damon_ctx *context);
 };
 
 int damon_set_pids(struct damon_ctx *ctx, int *pids, ssize_t nr_pids);
diff --git a/mm/damon.c b/mm/damon.c
index a2be399ab0ab..98367c91596a 100644
--- a/mm/damon.c
+++ b/mm/damon.c
@@ -851,6 +851,8 @@ static int kdamond_fn(void *data)
        kdamond_init_regions(ctx);
        while (!kdamond_need_stop(ctx)) {
                kdamond_prepare_access_checks(ctx);
+               if (ctx->sample_cb)
+                       ctx->sample_cb(ctx);
 
                usleep_range(ctx->sample_interval, ctx->sample_interval + 1);
 
@@ -858,6 +860,8 @@ static int kdamond_fn(void *data)
 
                if (kdamond_aggregate_interval_passed(ctx)) {
                        kdamond_merge_regions(ctx, max_nr_accesses / 10);
+                       if (ctx->aggregate_cb)
+                               ctx->aggregate_cb(ctx);
                        kdamond_reset_aggregated(ctx);
                        kdamond_split_regions(ctx);
                }
-- 
2.17.1

Reply via email to