JakeHillion/scx-upstream
1
0
Fork 0
mirror of https://github.com/sched-ext/scx.git synced 2024-11-25 20:20:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor refactor of scx_rusty

Browse Source
This commit is contained in:
Jordan Rome 2023-12-14 06:41:13 -08:00
parent ca05e28dde
commit ba35b97bb7
1 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
scheds/rust-user/scx_rusty/src/bpf/main.bpf.c
View File

@ -425,10 +425,13 @@ s32 BPF_STRUCT_OPS(rusty_select_cpu, struct task_struct *p, s32 prev_cpu,
if (!(taskc = lookup_task_ctx(p)) || !(p_cpumask = taskc->cpumask))
goto enoent;
if (kthreads_local &&
(p->flags & PF_KTHREAD) && p->nr_cpus_allowed == 1) {
if (p->nr_cpus_allowed == 1) {
cpu = prev_cpu;
stat_add(RUSTY_STAT_DIRECT_DISPATCH, 1);
if (kthreads_local && (p->flags & PF_KTHREAD)) {
stat_add(RUSTY_STAT_DIRECT_DISPATCH, 1);
} else {
stat_add(RUSTY_STAT_PINNED, 1);
}
goto direct;
}
@ -436,7 +439,7 @@ s32 BPF_STRUCT_OPS(rusty_select_cpu, struct task_struct *p, s32 prev_cpu,
* If WAKE_SYNC and the machine isn't fully saturated, wake up @p to the
* local dsq of the waker.
*/
if (p->nr_cpus_allowed > 1 && (wake_flags & SCX_WAKE_SYNC)) {
if (wake_flags & SCX_WAKE_SYNC) {
struct task_struct *current = (void *)bpf_get_current_task();
if (!(BPF_CORE_READ(current, flags) & PF_EXITING) &&
@ -475,13 +478,6 @@ s32 BPF_STRUCT_OPS(rusty_select_cpu, struct task_struct *p, s32 prev_cpu,
}
}
/* If only one CPU is allowed, dispatch */
if (p->nr_cpus_allowed == 1) {
stat_add(RUSTY_STAT_PINNED, 1);
cpu = prev_cpu;
goto direct;
}
has_idle_cores = !bpf_cpumask_empty(idle_smtmask);
/* did @p get pulled out to a foreign domain by e.g. greedy execution? */