mm, slub: skip percpu sheaves for remote object freeing

Since we don't control the NUMA locality of objects in percpu sheaves,
allocations with node restrictions bypass them. Allocations without
restrictions may however still expect to get local objects with high
probability, and the introduction of sheaves can decrease it due to
freed object from a remote node ending up in percpu sheaves.

The fraction of such remote frees seems low (5% on an 8-node machine)
but it can be expected that some cache or workload specific corner cases
exist. We can either conclude that this is not a problem due to the low
fraction, or we can make remote frees bypass percpu sheaves and go
directly to their slabs. This will make the remote frees more expensive,
but if if's only a small fraction, most frees will still benefit from
the lower overhead of percpu sheaves.

This patch thus makes remote object freeing bypass percpu sheaves,
including bulk freeing, and kfree_rcu() via the rcu_free sheaf. However
it's not intended to be 100% guarantee that percpu sheaves will only
contain local objects. The refill from slabs does not provide that
guarantee in the first place, and there might be cpu migrations
happening when we need to unlock the local_lock. Avoiding all that could
be possible but complicated so we can leave it for later investigation
whether it would be worth it. It can be expected that the more selective
freeing will itself prevent accumulation of remote objects in percpu
sheaves so any such violations would have only short-term effects.

Signed-off-by: Vlastimil Babka <vbabka@suse.cz>

diff --git a/mm/slab_common.c b/mm/slab_common.c
index 2d806e02568532a1000fd3912db6978e945dcfa8..f466f68a5bd82030a987baf849a98154cd48ef23 100644 (file)
--- a/mm/slab_common.c
+++ b/mm/slab_common.c
@@ -1623,8 +1623,11 @@ static bool kfree_rcu_sheaf(void *obj)
 
        slab = folio_slab(folio);
        s = slab->slab_cache;
-       if (s->cpu_sheaves)
-               return __kfree_rcu_sheaf(s, obj);
+       if (s->cpu_sheaves) {
+               if (likely(!IS_ENABLED(CONFIG_NUMA) ||
+                          slab_nid(slab) == numa_node_id()))
+                       return __kfree_rcu_sheaf(s, obj);
+       }
 
        return false;
 }

diff --git a/mm/slub.c b/mm/slub.c
index 339d91c6ea29be99a14a8914117fab0e3e6ed26b..50fc35b8fc9b3101821c338e9469c134677ded51 100644 (file)
--- a/mm/slub.c
+++ b/mm/slub.c
@@ -455,6 +455,7 @@ struct slab_sheaf {
        };
        struct kmem_cache *cache;
        unsigned int size;
+       int node; /* only used for rcu_sheaf */
        void *objects[];
 };
 
@@ -5682,7 +5683,7 @@ static void rcu_free_sheaf(struct rcu_head *head)
         */
        __rcu_free_sheaf_prepare(s, sheaf);
 
-       barn = get_node(s, numa_mem_id())->barn;
+       barn = get_node(s, sheaf->node)->barn;
 
        /* due to slab_free_hook() */
        if (unlikely(sheaf->size == 0))
@@ -5765,10 +5766,12 @@ do_free:
 
        rcu_sheaf->objects[rcu_sheaf->size++] = obj;
 
-       if (likely(rcu_sheaf->size < s->sheaf_capacity))
+       if (likely(rcu_sheaf->size < s->sheaf_capacity)) {
                rcu_sheaf = NULL;
-       else
+       } else {
                pcs->rcu_free = NULL;
+               rcu_sheaf->node = numa_mem_id();
+       }
 
        local_unlock(&s->cpu_sheaves->lock);
 
@@ -5794,7 +5797,11 @@ static void free_to_pcs_bulk(struct kmem_cache *s, size_t size, void **p)
        struct slab_sheaf *main, *empty;
        bool init = slab_want_init_on_free(s);
        unsigned int batch, i = 0;
+       void *remote_objects[PCS_BATCH_MAX];
+       unsigned int remote_nr = 0;
+       int node = numa_mem_id();
 
+next_remote_batch:
        while (i < size) {
                struct slab *slab = virt_to_slab(p[i]);
 
@@ -5804,7 +5811,15 @@ static void free_to_pcs_bulk(struct kmem_cache *s, size_t size, void **p)
                if (unlikely(!slab_free_hook(s, p[i], init, false))) {
                        p[i] = p[--size];
                        if (!size)
-                               return;
+                               goto flush_remote;
+                       continue;
+               }
+
+               if (unlikely(IS_ENABLED(CONFIG_NUMA) && slab_nid(slab) != node)) {
+                       remote_objects[remote_nr] = p[i];
+                       p[i] = p[--size];
+                       if (++remote_nr >= PCS_BATCH_MAX)
+                               goto flush_remote;
                        continue;
                }
 
@@ -5872,6 +5887,15 @@ no_empty:
         */
 fallback:
        __kmem_cache_free_bulk(s, size, p);
+
+flush_remote:
+       if (remote_nr) {
+               __kmem_cache_free_bulk(s, remote_nr, &remote_objects[0]);
+               if (i < size) {
+                       remote_nr = 0;
+                       goto next_remote_batch;
+               }
+       }
 }
 
 #ifndef CONFIG_SLUB_TINY
@@ -5963,8 +5987,14 @@ void slab_free(struct kmem_cache *s, struct slab *slab, void *object,
        if (unlikely(!slab_free_hook(s, object, slab_want_init_on_free(s), false)))
                return;
 
-       if (!s->cpu_sheaves || !free_to_pcs(s, object))
-               do_slab_free(s, slab, object, object, 1, addr);
+       if (s->cpu_sheaves && likely(!IS_ENABLED(CONFIG_NUMA) ||
+                                    slab_nid(slab) == numa_mem_id())) {
+               if (likely(free_to_pcs(s, object))) {
+                       return;
+               }
+       }
+
+       do_slab_free(s, slab, object, object, 1, addr);
 }
 
 #ifdef CONFIG_MEMCG