@@ -6305,16 +6305,26 @@ static const struct mm_walk_ops charge_walk_ops = {
.pmd_entry = mem_cgroup_move_charge_pte_range,
};
-static void mem_cgroup_move_charge(void)
+static void mem_cgroup_start_move_charge(struct mem_cgroup *memcg)
{
- lru_add_drain_all();
/*
* Signal folio_memcg_lock() to take the memcg's move_lock
* while we're moving its pages to another memcg. Then wait
* for already started RCU-only updates to finish.
*/
- atomic_inc(&mc.from->moving_account);
+ atomic_inc(&memcg->moving_account);
synchronize_rcu();
+}
+
+static void mem_cgroup_end_move_charge(struct mem_cgroup *memcg)
+{
+ atomic_dec(&memcg->moving_account);
+}
+
+static void mem_cgroup_move_charge(void)
+{
+ lru_add_drain_all();
+ mem_cgroup_start_move_charge(mc.from);
retry:
if (unlikely(!mmap_read_trylock(mc.mm))) {
/*
@@ -6334,7 +6344,7 @@ static void mem_cgroup_move_charge(void)
*/
walk_page_range(mc.mm, 0, ULONG_MAX, &charge_walk_ops, NULL);
mmap_read_unlock(mc.mm);
- atomic_dec(&mc.from->moving_account);
+ mem_cgroup_end_move_charge(mc.from);
}
static void mem_cgroup_move_task(void)
memcg->moving_account is used to signal that a memcg move is taking place, so that folio_memcg_lock() would start acquiring the per-memcg move lock instead of just initiating an rcu read section. Refactor incrementing and decrementing memcg->moving_account, together with rcu synchornization and the elaborate comment into helpers, to allow for reuse by incoming patches. Signed-off-by: Yosry Ahmed <yosryahmed@google.com> --- mm/memcontrol.c | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-)