[mm] [PATCH 3/4] Memory cgroup hierarchical reclaim

From: Balbir Singh
Date: Sat Nov 01 2008 - 14:49:28 EST


This patch introduces hierarchical reclaim. When an ancestor goes over its
limit, the charging routine points to the parent that is above its limit.
The reclaim process then starts from the last scanned child of the ancestor
and reclaims until the ancestor goes below its limit.

Signed-off-by: Balbir Singh <balbir@xxxxxxxxxxxxxxxxxx>
---

mm/memcontrol.c | 153 +++++++++++++++++++++++++++++++++++++++++++++++---------
1 file changed, 129 insertions(+), 24 deletions(-)

diff -puN mm/memcontrol.c~memcg-hierarchical-reclaim mm/memcontrol.c
--- linux-2.6.28-rc2/mm/memcontrol.c~memcg-hierarchical-reclaim 2008-11-02 00:14:59.000000000 +0530
+++ linux-2.6.28-rc2-balbir/mm/memcontrol.c 2008-11-02 00:14:59.000000000 +0530
@@ -132,6 +132,11 @@ struct mem_cgroup {
* statistics.
*/
struct mem_cgroup_stat stat;
+ /*
+ * While reclaiming in a hiearchy, we cache the last child we
+ * reclaimed from.
+ */
+ struct mem_cgroup *last_scanned_child;
};
static struct mem_cgroup init_mem_cgroup;

@@ -467,6 +472,125 @@ unsigned long mem_cgroup_isolate_pages(u
return nr_taken;
}

+static struct mem_cgroup *
+mem_cgroup_from_res_counter(struct res_counter *counter)
+{
+ return container_of(counter, struct mem_cgroup, res);
+}
+
+/*
+ * Dance down the hierarchy if needed to reclaim memory. We remember the
+ * last child we reclaimed from, so that we don't end up penalizing
+ * one child extensively based on its position in the children list
+ */
+static int
+mem_cgroup_hierarchical_reclaim(struct mem_cgroup *mem, gfp_t gfp_mask)
+{
+ struct cgroup *cg, *cg_current, *cgroup;
+ struct mem_cgroup *mem_child;
+ int ret = 0;
+
+ if (try_to_free_mem_cgroup_pages(mem, gfp_mask))
+ return -ENOMEM;
+
+ /*
+ * try_to_free_mem_cgroup_pages() might not give us a full
+ * picture of reclaim. Some pages are reclaimed and might be
+ * moved to swap cache or just unmapped from the cgroup.
+ * Check the limit again to see if the reclaim reduced the
+ * current usage of the cgroup before giving up
+ */
+ if (res_counter_check_under_limit(&mem->res))
+ return 0;
+
+ /*
+ * Scan all children under the mem_cgroup mem
+ */
+ if (!mem->last_scanned_child)
+ cgroup = list_first_entry(&mem->css.cgroup->children,
+ struct cgroup, sibling);
+ else
+ cgroup = mem->last_scanned_child->css.cgroup;
+
+ cg_current = cgroup;
+
+ /*
+ * We iterate twice, one of it is fundamental list issue, where
+ * the elements are inserted using list_add and hence the list
+ * behaves like a stack and list_for_entry_safe_from() stops
+ * after seeing the first child. The two loops help us work
+ * independently of the insertion and it helps us get a full pass at
+ * scanning all list entries for reclaim
+ */
+ list_for_each_entry_safe_from(cgroup, cg, &cg_current->parent->children,
+ sibling) {
+ mem_child = mem_cgroup_from_cont(cgroup);
+
+ /*
+ * Move beyond last scanned child
+ */
+ if (mem_child == mem->last_scanned_child)
+ continue;
+
+ ret = try_to_free_mem_cgroup_pages(mem_child, gfp_mask);
+ mem->last_scanned_child = mem_child;
+
+ if (res_counter_check_under_limit(&mem->res)) {
+ ret = 0;
+ goto done;
+ }
+ }
+
+ list_for_each_entry_safe(cgroup, cg, &cg_current->parent->children,
+ sibling) {
+ mem_child = mem_cgroup_from_cont(cgroup);
+
+ ret = try_to_free_mem_cgroup_pages(mem_child, gfp_mask);
+ mem->last_scanned_child = mem_child;
+
+ if (res_counter_check_under_limit(&mem->res)) {
+ ret = 0;
+ goto done;
+ }
+ }
+
+done:
+ return ret;
+}
+
+/*
+ * Charge memory cgroup mem and check if it is over its limit. If so, reclaim
+ * from mem.
+ */
+static int mem_cgroup_charge_and_reclaim(struct mem_cgroup *mem, gfp_t gfp_mask)
+{
+ int ret = 0;
+ unsigned long nr_retries = MEM_CGROUP_RECLAIM_RETRIES;
+ struct res_counter *fail_res;
+ struct mem_cgroup *mem_over_limit;
+
+ while (unlikely(res_counter_charge(&mem->res, PAGE_SIZE, &fail_res))) {
+ if (!(gfp_mask & __GFP_WAIT))
+ goto out;
+
+ /*
+ * Is one of our ancestors over their limit?
+ */
+ if (fail_res)
+ mem_over_limit = mem_cgroup_from_res_counter(fail_res);
+ else
+ mem_over_limit = mem;
+
+ ret = mem_cgroup_hierarchical_reclaim(mem_over_limit, gfp_mask);
+
+ if (!nr_retries--) {
+ mem_cgroup_out_of_memory(mem, gfp_mask);
+ goto out;
+ }
+ }
+out:
+ return ret;
+}

/**
* mem_cgroup_try_charge - get charge of PAGE_SIZE.
@@ -484,8 +608,7 @@ int mem_cgroup_try_charge(struct mm_stru
gfp_t gfp_mask, struct mem_cgroup **memcg)
{
struct mem_cgroup *mem;
- int nr_retries = MEM_CGROUP_RECLAIM_RETRIES;
- struct res_counter *fail_res;
+
/*
* We always charge the cgroup the mm_struct belongs to.
* The mm_struct's mem_cgroup changes on task migration if the
@@ -510,29 +633,9 @@ int mem_cgroup_try_charge(struct mm_stru
css_get(&mem->css);
}

+ if (mem_cgroup_charge_and_reclaim(mem, gfp_mask))
+ goto nomem;

- while (unlikely(res_counter_charge(&mem->res, PAGE_SIZE, &fail_res))) {
- if (!(gfp_mask & __GFP_WAIT))
- goto nomem;
-
- if (try_to_free_mem_cgroup_pages(mem, gfp_mask))
- continue;
-
- /*
- * try_to_free_mem_cgroup_pages() might not give us a full
- * picture of reclaim. Some pages are reclaimed and might be
- * moved to swap cache or just unmapped from the cgroup.
- * Check the limit again to see if the reclaim reduced the
- * current usage of the cgroup before giving up
- */
- if (res_counter_check_under_limit(&mem->res))
- continue;
-
- if (!nr_retries--) {
- mem_cgroup_out_of_memory(mem, gfp_mask);
- goto nomem;
- }
- }
return 0;
nomem:
css_put(&mem->css);
@@ -1195,6 +1298,8 @@ mem_cgroup_create(struct cgroup_subsys *
if (alloc_mem_cgroup_per_zone_info(mem, node))
goto free_out;

+ mem->last_scanned_child = NULL;
+
return &mem->css;
free_out:
for_each_node_state(node, N_POSSIBLE)
_

--
Balbir
--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majordomo@xxxxxxxxxxxxxxx
More majordomo info at http://vger.kernel.org/majordomo-info.html
Please read the FAQ at http://www.tux.org/lkml/