mm/vmscan: unify writeback reclaim statistic and throttling

Currently MGLRU and non-MGLRU handle the reclaim statistic and writeback
handling very differently, especially throttling.  Basically MGLRU just
ignored the throttling part.

Let's just unify this part, use a helper to deduplicate the code so both
setups will share the same behavior.

Test using following reproducer using bash:

  echo "Setup a slow device using dm delay"
  dd if=/dev/zero of=/var/tmp/backing bs=1M count=2048
  LOOP=$(losetup --show -f /var/tmp/backing)
  mkfs.ext4 -q $LOOP
  echo "0 $(blockdev --getsz $LOOP) delay $LOOP 0 0 $LOOP 0 1000" | \
      dmsetup create slow_dev
  mkdir -p /mnt/slow && mount /dev/mapper/slow_dev /mnt/slow

  echo "Start writeback pressure"
  sync && echo 3 > /proc/sys/vm/drop_caches
  mkdir /sys/fs/cgroup/test_wb
  echo 128M > /sys/fs/cgroup/test_wb/memory.max
  (echo $BASHPID > /sys/fs/cgroup/test_wb/cgroup.procs && \
      dd if=/dev/zero of=/mnt/slow/testfile bs=1M count=192)

  echo "Clean up"
  echo "0 $(blockdev --getsz $LOOP) error" | dmsetup load slow_dev
  dmsetup resume slow_dev
  umount -l /mnt/slow && sync
  dmsetup remove slow_dev

Before this commit, `dd` will get OOM killed immediately if MGLRU is
enabled.  Classic LRU is fine.

After this commit, throttling is now effective and no more spin on LRU or
premature OOM.  Stress test on other workloads also looks good.

Global throttling is not here yet, we will fix that separately later.

Link: https://lore.kernel.org/20260428-mglru-reclaim-v7-15-02fabb92dc43@tencent.com
Signed-off-by: Kairui Song <kasong@tencent.com>
Suggested-by: Chen Ridong <chenridong@huaweicloud.com>
Tested-by: Leno Hou <lenohou@gmail.com>
Reviewed-by: Axel Rasmussen <axelrasmussen@google.com>
Reviewed-by: Baolin Wang <baolin.wang@linux.alibaba.com>
Cc: Barry Song <baohua@kernel.org>
Cc: Chris Li <chrisl@kernel.org>
Cc: David Hildenbrand <david@kernel.org>
Cc: David Stevens <stevensd@google.com>
Cc: Johannes Weiner <hannes@cmpxchg.org>
Cc: Kalesh Singh <kaleshsingh@google.com>
Cc: Lorenzo Stoakes <ljs@kernel.org>
Cc: Michal Hocko <mhocko@kernel.org>
Cc: Shakeel Butt <shakeel.butt@linux.dev>
Cc: Suren Baghdasaryan <surenb@google.com>
Cc: Vernon Yang <vernon2gm@gmail.com>
Cc: Wei Xu <weixugc@google.com>
Cc: Yafang <laoar.shao@gmail.com>
Cc: Yuanchu Xie <yuanchu@google.com>
Cc: Yu Zhao <yuzhao@google.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>

diff --git a/mm/vmscan.c b/mm/vmscan.c
index 7494ac73e3f19344ec00753d3d0cae009d0eac12..e8a90911bf88fba6279daee1d2b59f4db2346316 100644 (file)
--- a/mm/vmscan.c
+++ b/mm/vmscan.c
@@ -1946,6 +1946,44 @@ static int current_may_throttle(void)
        return !(current->flags & PF_LOCAL_THROTTLE);
 }
 
+static void handle_reclaim_writeback(unsigned long nr_taken,
+                                    struct pglist_data *pgdat,
+                                    struct scan_control *sc,
+                                    struct reclaim_stat *stat)
+{
+       /*
+        * If dirty folios are scanned that are not queued for IO, it
+        * implies that flushers are not doing their job. This can
+        * happen when memory pressure pushes dirty folios to the end of
+        * the LRU before the dirty limits are breached and the dirty
+        * data has expired. It can also happen when the proportion of
+        * dirty folios grows not through writes but through memory
+        * pressure reclaiming all the clean cache. And in some cases,
+        * the flushers simply cannot keep up with the allocation
+        * rate. Nudge the flusher threads in case they are asleep.
+        */
+       if (stat->nr_unqueued_dirty == nr_taken) {
+               wakeup_flusher_threads(WB_REASON_VMSCAN);
+               /*
+                * For cgroupv1 dirty throttling is achieved by waking up
+                * the kernel flusher here and later waiting on folios
+                * which are in writeback to finish (see shrink_folio_list()).
+                *
+                * Flusher may not be able to issue writeback quickly
+                * enough for cgroupv1 writeback throttling to work
+                * on a large system.
+                */
+               if (!writeback_throttling_sane(sc))
+                       reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
+       }
+
+       sc->nr.dirty += stat->nr_dirty;
+       sc->nr.congested += stat->nr_congested;
+       sc->nr.writeback += stat->nr_writeback;
+       sc->nr.immediate += stat->nr_immediate;
+       sc->nr.taken += nr_taken;
+}
+
 /*
  * shrink_inactive_list() is a helper for shrink_node().  It returns the number
  * of reclaimed pages
@@ -2009,39 +2047,7 @@ static unsigned long shrink_inactive_list(unsigned long nr_to_scan,
        lruvec_lock_irq(lruvec);
        lru_note_cost_unlock_irq(lruvec, file, stat.nr_pageout,
                                        nr_scanned - nr_reclaimed);
-
-       /*
-        * If dirty folios are scanned that are not queued for IO, it
-        * implies that flushers are not doing their job. This can
-        * happen when memory pressure pushes dirty folios to the end of
-        * the LRU before the dirty limits are breached and the dirty
-        * data has expired. It can also happen when the proportion of
-        * dirty folios grows not through writes but through memory
-        * pressure reclaiming all the clean cache. And in some cases,
-        * the flushers simply cannot keep up with the allocation
-        * rate. Nudge the flusher threads in case they are asleep.
-        */
-       if (stat.nr_unqueued_dirty == nr_taken) {
-               wakeup_flusher_threads(WB_REASON_VMSCAN);
-               /*
-                * For cgroupv1 dirty throttling is achieved by waking up
-                * the kernel flusher here and later waiting on folios
-                * which are in writeback to finish (see shrink_folio_list()).
-                *
-                * Flusher may not be able to issue writeback quickly
-                * enough for cgroupv1 writeback throttling to work
-                * on a large system.
-                */
-               if (!writeback_throttling_sane(sc))
-                       reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
-       }
-
-       sc->nr.dirty += stat.nr_dirty;
-       sc->nr.congested += stat.nr_congested;
-       sc->nr.writeback += stat.nr_writeback;
-       sc->nr.immediate += stat.nr_immediate;
-       sc->nr.taken += nr_taken;
-
+       handle_reclaim_writeback(nr_taken, pgdat, sc, &stat);
        trace_mm_vmscan_lru_shrink_inactive(pgdat->node_id,
                        nr_scanned, nr_reclaimed, &stat, sc->priority, file);
        return nr_reclaimed;
@@ -4833,26 +4839,13 @@ static int evict_folios(unsigned long nr_to_scan, struct lruvec *lruvec,
 retry:
        reclaimed = shrink_folio_list(&list, pgdat, sc, &stat, false, memcg);
        sc->nr_reclaimed += reclaimed;
+       /* Retry pass is only meant for clean folios without new isolation */
+       if (isolated)
+               handle_reclaim_writeback(isolated, pgdat, sc, &stat);
        trace_mm_vmscan_lru_shrink_inactive(pgdat->node_id,
                        type_scanned, reclaimed, &stat, sc->priority,
                        type ? LRU_INACTIVE_FILE : LRU_INACTIVE_ANON);
 
-       /*
-        * If too many file cache in the coldest generation can't be evicted
-        * due to being dirty, wake up the flusher.
-        */
-       if (stat.nr_unqueued_dirty == isolated) {
-               wakeup_flusher_threads(WB_REASON_VMSCAN);
-
-               /*
-                * For cgroupv1 dirty throttling is achieved by waking up
-                * the kernel flusher here and later waiting on folios
-                * which are in writeback to finish (see shrink_folio_list()).
-                */
-               if (!writeback_throttling_sane(sc))
-                       reclaim_throttle(pgdat, VMSCAN_THROTTLE_WRITEBACK);
-       }
-
        list_for_each_entry_safe_reverse(folio, next, &list, lru) {
                DEFINE_MIN_SEQ(lruvec);
 
@@ -4895,6 +4888,7 @@ retry:
 
        if (!list_empty(&list)) {
                skip_retry = true;
+               isolated = 0;
                goto retry;
        }