ANDROID: sched/fair: call set_next_entity() on the replaced next task

android_rvh_replace_next_task_fair hook allows vendor modules to
override the next task selected by CFS. However the current code is
not calling set_next_entity() on the hierarchy of the replaced task
in the case where the previous task is CFS. Fix this issue.

Bug: 184720311
Change-Id: If6c35b1ddefd0829cd236dd821e5ac8aef7347c6
Signed-off-by: Pavankumar Kondeti <quic_pkondeti@quicinc.com>

kernel/sched/fair.c

@@ -7211,8 +7211,11 @@ simple:
 		put_prev_task(rq, prev);
 
 	trace_android_rvh_replace_next_task_fair(rq, &p, &se, &repick, true);
-	if (repick)
+	if (repick) {
+		for_each_sched_entity(se)
+			set_next_entity(cfs_rq_of(se), se);
 		goto done;
+	}
 
 	do {
 		se = pick_next_entity(cfs_rq, NULL);