[PATCH 2/2] drm/xe: Update shrinker batch size based on average BO size

["Thomas Hellström" <thomas.hellstrom@linux.intel.com>]

Update our preferred vmscan batch size on each count pass to avoid
invoking scan_objects for requests too small to free even a single
average-sized GEM object. Our rough estimate for an effective batch
is twice the average number of pages per populated ttm_tt across all
shrinkable and purgeable objects. The factor of two provides headroom
so that most scan invocations can free at least one GEM object despite
variability in object sizes.

The batch value is updated as an exponential moving average,
(old_batch + avg) / 2, to smooth out sudden changes in the
object population. It is floored at 128 pages, the kernel default
SHRINK_BATCH, to ensure the shrinker remains responsive when there
are very few objects.

The populated_tts counter introduced in the previous commit provides
the object count needed for the average. We inherit the same
justification as the analogous mechanism in i915: shrinking a GEM
object has non-trivial locking overhead, so firing the shrinker for
requests smaller than a single object is wasteful.

Assisted-by: GitHub_Copilot:claude-sonnet-4.6
Signed-off-by: Thomas Hellström <thomas.hellstrom@linux.intel.com>
---
 drivers/gpu/drm/xe/xe_shrinker.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/drivers/gpu/drm/xe/xe_shrinker.c b/drivers/gpu/drm/xe/xe_shrinker.c
index cded230f5459..a15f568d8186 100644
--- a/drivers/gpu/drm/xe/xe_shrinker.c
+++ b/drivers/gpu/drm/xe/xe_shrinker.c
@@ -146,6 +146,7 @@ xe_shrinker_count(struct shrinker *shrink, struct shrink_control *sc)
 {
 	struct xe_shrinker *shrinker = to_xe_shrinker(shrink);
 	unsigned long num_pages;
+	unsigned long populated_tts;
 	bool can_backup = !!(sc->gfp_mask & __GFP_FS);
 
 	num_pages = ttm_backup_bytes_avail() >> PAGE_SHIFT;
@@ -157,8 +158,24 @@ xe_shrinker_count(struct shrinker *shrink, struct shrink_control *sc)
 		num_pages = 0;
 
 	num_pages += shrinker->purgeable_pages;
+	populated_tts = shrinker->populated_tts;
 	read_unlock(&shrinker->lock);
 
+	/*
+	 * Update our preferred vmscan batch size for the next pass.
+	 * Our rough guess for an effective batch size is twice the average
+	 * number of pages per GEM object. That is, we don't want the
+	 * shrinker to fire until the request is large enough to justify
+	 * the overhead of freeing at least one GEM object.
+	 */
+	if (populated_tts) {
+		unsigned long avg = 2 * num_pages / populated_tts;
+
+		shrinker->shrink->batch =
+			max((shrinker->shrink->batch + avg) >> 1,
+			    128UL /* default SHRINK_BATCH */);
+	}
+
 	return num_pages ? num_pages : SHRINK_EMPTY;
 }
 
-- 
2.54.0