Skip to contentSkip to conversation

memcg: no refill for offlined objcg

PATCH In review Shakeel Butt 2 participants 3 messages Apr 10, 2025 – Apr 14, 2025

Subsystems (3)
lore
Share & tools
lore

Shakeel Butt

author maintainer
#0 lore raw

PATCH 1/1 memcg: no refill for offlined objcg

In our fleet, we are observing refill_obj_stock() spending a lot of cpu in obj_cgroup_get() and on further inspection it seems like the given objcg is offlined and the kernel has to take the slow path i.e. atomic operations for objcg reference counting.

Other than expensive atomic operations, refilling stock of an offlined objcg is a waster as there will not be new allocations for the offlined objcg. In addition, refilling triggers flush of the previous objcg which might be used in future. So, let's just avoid refilling the stock with the offlined objcg.

---
 mm/memcontrol.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
mm/memcontrol.c +18 -0
index 2178a051bd09..23c62ae6a8c6 100644
@@ -2474,6 +2474,17 @@ static inline void __mod_objcg_mlstate(struct obj_cgroup *objcg,
 	rcu_read_unlock();
 }
 
+static inline void mod_objcg_mlstate(struct obj_cgroup *objcg,
+				     struct pglist_data *pgdat,
+				     enum node_stat_item idx, int nr)
+{
+	unsigned long flags;
+
+	local_irq_save(flags);
+	__mod_objcg_mlstate(objcg, pgdat, idx, nr);
+	local_irq_restore(flags);
+}
+
 static __always_inline
 struct mem_cgroup *mem_cgroup_from_obj_folio(struct folio *folio, void *p)
 {
@@ -2925,6 +2936,13 @@ static void refill_obj_stock(struct obj_cgroup *objcg, unsigned int nr_bytes,
 	unsigned long flags;
 	unsigned int nr_pages = 0;
 
+	if (unlikely(percpu_ref_is_dying(&objcg->refcnt))) {
+		atomic_add(nr_bytes, &objcg->nr_charged_bytes);
+		if (pgdat)
+			mod_objcg_mlstate(objcg, pgdat, idx, nr_acct);
+		return;
+	}
+
 	local_lock_irqsave(&memcg_stock.stock_lock, flags);
 
 	stock = this_cpu_ptr(&memcg_stock);
Signature · 1 line
2.47.1
maintainer
#1 lore raw
Quoted 10 lines · Shakeel Butt

In our fleet, we are observing refill_obj_stock() spending a lot of cpu in obj_cgroup_get() and on further inspection it seems like the given objcg is offlined and the kernel has to take the slow path i.e. atomic operations for objcg reference counting.

Other than expensive atomic operations, refilling stock of an offlined objcg is a waster as there will not be new allocations for the offlined objcg. In addition, refilling triggers flush of the previous objcg which might be used in future. So, let's just avoid refilling the stock with the offlined objcg.

Hm, but on the other side if there are multiple uncharges in a row, refilling obj stocks might be still cheaper?

In general I think that switching to atomic css refcnt on memcg offlining is a mistake - it makes memory reclaim generally more expensive. We can simple delay it until the approximate refcnt number reaches some low value, e.g. 100 objects.

author maintainer
#2 lore raw
Quoted 7 lines · Roman Gushchin
Quoted 5 lines · Shakeel Butt

In our fleet, we are observing refill_obj_stock() spending a lot of cpu in obj_cgroup_get() and on further inspection it seems like the given objcg is offlined and the kernel has to take the slow path i.e. atomic operations for objcg reference counting.

Quoted 9 lines
Quoted 5 lines

Other than expensive atomic operations, refilling stock of an offlined objcg is a waster as there will not be new allocations for the offlined objcg. In addition, refilling triggers flush of the previous objcg which might be used in future. So, let's just avoid refilling the stock with the offlined objcg.

Hm, but on the other side if there are multiple uncharges in a row, refilling obj stocks might be still cheaper?

Thanks for the review. I looked at the fleet data again and what you are suspecting i.e. multiple objects of same objcg getting freed closeby is possible. I think I should be optimizing/batching at the upper layer. I will be looking at rcu freeing side which is the most obvious batching opportunity.

Quoted 4 lines

In general I think that switching to atomic css refcnt on memcg offlining is a mistake - it makes memory reclaim generally more expensive. We can simple delay it until the approximate refcnt number reaches some low value, e.g. 100 objects.

This is a good idea. For the memcg, I think after Muchun's LRU reparenting, we should not have zombie refcnt slowdown but for objcg this idea might help. Anyways this is for later.