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

scx_lavd: replace get_task_cpu_id() to scx_bpf_task_cpu()

Browse Source 
Signed-off-by: Changwoo Min <changwoo@igalia.com>
This commit is contained in:
Changwoo Min 2024-08-20 09:10:35 +09:00
parent bf67b83561
commit 9368c6881d
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scheds/rust/scx_lavd/src/bpf/main.bpf.c
View File

@ -362,14 +362,6 @@ static struct task_ctx *get_task_ctx(struct task_struct *p)
return taskc; return taskc;
} }
static s32 get_task_cpu_id(struct task_struct *p)
{
/*
* This code assumes ONFIG_THREAD_INFO_IN_TASK is on in the kernel.
*/
return READ_ONCE(p->thread_info.cpu);
}
static struct cpu_ctx *get_cpu_ctx(void) static struct cpu_ctx *get_cpu_ctx(void)
{ {
const u32 idx = 0; const u32 idx = 0;
@ -1822,7 +1814,7 @@ void BPF_STRUCT_OPS(lavd_enqueue, struct task_struct *p, u64 enq_flags)
* always put the task to the global DSQ, so any idle CPU can pick it * always put the task to the global DSQ, so any idle CPU can pick it
* up. * up.
*/ */
cpu_id = get_task_cpu_id(p); cpu_id = scx_bpf_task_cpu(p);
taskc = get_task_ctx(p); taskc = get_task_ctx(p);
cpuc_task = get_cpu_ctx_id(cpu_id); cpuc_task = get_cpu_ctx_id(cpu_id);
cpuc_cur = get_cpu_ctx(); cpuc_cur = get_cpu_ctx();