Hi, This is the v3 RFC to add an opt-in percpu array-based caching layer to SLUB. This is to publish accumulated fixes since v2 ahead of LSF/MM. I have also squashed the usage of localtry_lock to be used immediately (not converted to in a separate patch) as the lock should be going to 6.15 at this point. There's still a patch introducing it for this RFC to avoid depending on bpf-next tree. The name "sheaf" was invented by Matthew so we don't call it magazine like the original Bonwick paper. The per-NUMA-node cache of sheaves is thus called "barn". This may seem similar to the arrays in SLAB, but the main differences are: - opt-in, not used for every cache - does not distinguish NUMA locality, thus no "shared" arrays and no "alien" arrays that would need periodical flushing - improves kfree_rcu() handling - API for obtaining a preallocated sheaf that can be used for guaranteed and efficient allocations in a restricted context, when the upper bound for needed objects is known but rarely reached The motivation comes mainly from the ongoing work related to VMA scalability and the related maple tree operations. This is why maple tree nodes are sheaf-enabled in the RFC, but it's not a full conversion that would take benefits of the improved preallocation API. The VMA part is currently left out as it's expected that Suren will land the VMA TYPESAFE_BY_RCU conversion [3] soon and there would be conflict with that. With both series applied it means just adding a line to kmem_cache_args in proc_caches_init(). Some performance benefits were measured by Suren and Liam in previous versions. Suren's results in [5] look promising, so far except for the preallocation support as used by the refactored maple tree code. A sheaf-enabled cache has the following expected advantages: - Cheaper fast paths. For allocations, instead of local double cmpxchg, after Patch 5 it's preempt_disable() and no atomic operations. Same for freeing, which is normally a local double cmpxchg only for a short term allocations (so the same slab is still active on the same cpu when freeing the object) and a more costly locked double cmpxchg otherwise. The downside is the lack of NUMA locality guarantees for the allocated objects. - kfree_rcu() batching and recycling. kfree_rcu() will put objects to a separate percpu sheaf and only submit the whole sheaf to call_rcu() when full. After the grace period, the sheaf can be used for allocations, which is more efficient than freeing and reallocating individual slab objects (even with the batching done by kfree_rcu() implementation itself). In case only some cpus are allowed to handle rcu callbacks, the sheaf can still be made available to other cpus on the same node via the shared barn. The maple_node cache uses kfree_rcu() and thus can benefit from this. - Preallocation support. A prefilled sheaf can be privately borrowed for a short term operation that is not allowed to block in the middle and may need to allocate some objects. If an upper bound (worst case) for the number of allocations is known, but only much fewer allocations actually needed on average, borrowing and returning a sheaf is much more efficient then a bulk allocation for the worst case followed by a bulk free of the many unused objects. Maple tree write operations should benefit from this. Patch 1 is copied from the series "bpf, mm: Introduce try_alloc_pages()" [2] to introduce a variant of local_lock that has a trylock operation. Patch 2 implements the basic sheaf functionality as a caching layer on top of (per-cpu) slabs. Patch 3 adds the sheaf kfree_rcu() support. Patch 4 implements borrowing prefilled sheaves, with maple tree being the ancticipated user. Patch 5 seeks to reduce barn spinlock contention. Separately for possible evaluation. Patches 6 and 7 by Liam add testing stubs that maple tree will use in its userspace tests. Patch 9 enables sheaves for the maple tree node cache, but does not take advantage of prefilling yet. (RFC) LIMITATIONS: - with slub_debug enabled, objects in sheaves are considered allocated so allocation/free stacktraces may become imprecise and checking of e.g. redzone violations may be delayed GIT TREES: this series: https://git.kernel.org/vbabka/l/slub-percpu-sheaves-v3 To avoid conflicts, the series requires (and the branch above is based on) the kfree_rcu() code refactoring scheduled for 6.15: https://git.kernel.org/pub/scm/linux/kernel/git/vbabka/slab.git/log/?h=slab/for-6.15/kfree_rcu_tiny Vlastimil [1] https://lore.kernel.org/all/20241111205506.3404479-1-surenb@xxxxxxxxxx/ [2] https://lore.kernel.org/all/20250222024427.30294-2-alexei.starovoitov@xxxxxxxxx/ [3] https://lore.kernel.org/all/20250213224655.1680278-1-surenb@xxxxxxxxxx/ [4] https://www.infradead.org/git/?p=users/jedix/linux-maple.git;a=shortlog;h=refs/heads/slub-percpu-sheaves-v2 [5] https://lore.kernel.org/all/CAJuCfpFVopL%2BsMdU4bLRxs%2BHS_WPCmFZBdCmwE8qV2Dpa5WZnA@xxxxxxxxxxxxxx/ --- Changes in v3: - Squash localtry_lock conversion so it's used immediately. - Incorporate feedback and add tags from Suren and Harry - thanks! - Mostly adding comments and some refactoring. - Fixes for kfree_rcu_sheaf() vmalloc handling, cpu hotremove flushing. - Fix wrong condition in kmem_cache_return_sheaf() that may have affected performance negatively. - Refactoring of free_to_pcs() - Link to v2: https://lore.kernel.org/r/20250214-slub-percpu-caches-v2-0-88592ee0966a@xxxxxxx Changes in v2: - Removed kfree_rcu() destructors support as VMAs will not need it anymore after [3] is merged. - Changed to localtry_lock_t borrowed from [2] instead of an own implementation of the same idea. - Many fixes and improvements thanks to Liam's adoption for maple tree nodes. - Userspace Testing stubs by Liam. - Reduced limitations/todos - hooking to kfree_rcu() is complete, prefilled sheaves can exceed cache's sheaf_capacity. - Link to v1: https://lore.kernel.org/r/20241112-slub-percpu-caches-v1-0-ddc0bdc27e05@xxxxxxx --- Liam R. Howlett (2): tools: Add testing support for changes to rcu and slab for sheaves tools: Add sheafs support to testing infrastructure Sebastian Andrzej Siewior (1): locking/local_lock: Introduce localtry_lock_t Vlastimil Babka (5): slab: add opt-in caching layer of percpu sheaves slab: add sheaf support for batching kfree_rcu() operations slab: sheaf prefilling for guaranteed allocations slab: determine barn status racily outside of lock maple_tree: use percpu sheaves for maple_node_cache include/linux/local_lock.h | 70 ++ include/linux/local_lock_internal.h | 146 ++++ include/linux/slab.h | 50 ++ lib/maple_tree.c | 11 +- mm/slab.h | 4 + mm/slab_common.c | 29 +- mm/slub.c | 1463 +++++++++++++++++++++++++++++++-- tools/include/linux/slab.h | 65 +- tools/testing/shared/linux.c | 108 ++- tools/testing/shared/linux/rcupdate.h | 22 + 10 files changed, 1891 insertions(+), 77 deletions(-) --- base-commit: 379487e17ca406b47392e7ab6cf35d1c3bacb371 change-id: 20231128-slub-percpu-caches-9441892011d7 prerequisite-message-id: 20250203-slub-tiny-kfree_rcu-v1-0-d4428bf9a8a1@xxxxxxx prerequisite-patch-id: 1a4af92b5eb1b8bfc86bac8d7fc1ef0963e7d9d6 prerequisite-patch-id: f24a39c38103b7e09fbf2e6f84e6108499ab7980 prerequisite-patch-id: 23e90b23482f4775c95295821dd779ba4e3712e9 prerequisite-patch-id: 5c53a619477acdce07071abec0f40e79501ea40b Best regards, -- Vlastimil Babka <vbabka@xxxxxxx>
