1 files changed, 12 insertions, 13 deletions

diff --git a/mm/memcontrol.c b/mm/memcontrol.c
index 597d58101872..f3c15bb07cce 100644
--- a/mm/memcontrol.c
+++ b/mm/memcontrol.c
@@ -25,7 +25,7 @@
 #include <linux/page_counter.h>
 #include <linux/memcontrol.h>
 #include <linux/cgroup.h>
-#include <linux/mm.h>
+#include <linux/pagewalk.h>
 #include <linux/sched/mm.h>
 #include <linux/shmem_fs.h>
 #include <linux/hugetlb.h>
@@ -5499,17 +5499,16 @@ static int mem_cgroup_count_precharge_pte_range(pmd_t *pmd,
         return 0;
 }
 
+static const struct mm_walk_ops precharge_walk_ops = {
+        .pmd_entry      = mem_cgroup_count_precharge_pte_range,
+};
+
 static unsigned long mem_cgroup_count_precharge(struct mm_struct *mm)
 {
         unsigned long precharge;
 
-        struct mm_walk mem_cgroup_count_precharge_walk = {
-                .pmd_entry = mem_cgroup_count_precharge_pte_range,
-                .mm = mm,
-        };
         down_read(&mm->mmap_sem);
-        walk_page_range(0, mm->highest_vm_end,
-                        &mem_cgroup_count_precharge_walk);
+        walk_page_range(mm, 0, mm->highest_vm_end, &precharge_walk_ops, NULL);
         up_read(&mm->mmap_sem);
 
         precharge = mc.precharge;
@@ -5778,13 +5777,12 @@ put:			/* get_mctgt_type() gets the page */
         return ret;
 }
 
+static const struct mm_walk_ops charge_walk_ops = {
+        .pmd_entry      = mem_cgroup_move_charge_pte_range,
+};
+
 static void mem_cgroup_move_charge(void)
 {
-        struct mm_walk mem_cgroup_move_charge_walk = {
-                .pmd_entry = mem_cgroup_move_charge_pte_range,
-                .mm = mc.mm,
-        };
-
         lru_add_drain_all();
         /*
          * Signal lock_page_memcg() to take the memcg's move_lock
@@ -5810,7 +5808,8 @@ retry:
          * When we have consumed all precharges and failed in doing
          * additional charge, the page walk just aborts.
          */
-        walk_page_range(0, mc.mm->highest_vm_end, &mem_cgroup_move_charge_walk);
+        walk_page_range(mc.mm, 0, mc.mm->highest_vm_end, &charge_walk_ops,
+                        NULL);
 
         up_read(&mc.mm->mmap_sem);
         atomic_dec(&mc.from->moving_account);