Add missing load update in couple functions: trigger_loadbalancer

and task_dead.
hamadmarri · Nov 13, 2021 · b1e1202 · b1e1202
1 parent 970fef0
commit b1e1202
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/kernel/sched/bs.c b/kernel/sched/bs.c
@@ -1012,6 +1012,7 @@ void trigger_load_balance(struct rq *this_rq)
 	if (unlikely(on_null_domain(this_rq) || !cpu_active(cpu_of(this_rq))))
 		return;
 
+	update_blocked_averages(this_rq->cpu);
 	nohz_balancer_kick(this_rq);
 }
 

diff --git a/kernel/sched/bs.h b/kernel/sched/bs.h
@@ -58,12 +58,16 @@ static void rq_online_fair(struct rq *rq) {}
 static void rq_offline_fair(struct rq *rq) {}
 static void task_dead_fair(struct task_struct *p)
 {
+#ifdef CONFIG_TT_ACCOUNTING_STATS
+	remove_entity_load_avg(&p->se);
+#else
 	struct cfs_rq *cfs_rq = cfs_rq_of(&p->se);
 	unsigned long flags;
 
 	raw_spin_lock_irqsave(&cfs_rq->removed.lock, flags);
 	++cfs_rq->removed.nr;
 	raw_spin_unlock_irqrestore(&cfs_rq->removed.lock, flags);
+#endif
 }
 
 #endif /** CONFIG_SMP */
-Original file line number
+Diff line change
@@ Expand Up / @@ -1012,6 +1012,7 @@ void trigger_load_balance(struct rq *this_rq) @@
     	if (unlikely(on_null_domain(this_rq) || !cpu_active(cpu_of(this_rq))))
     		return;
+    	update_blocked_averages(this_rq->cpu);
     	nohz_balancer_kick(this_rq);
     }
@@ Expand Down @@