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

select_cpu metrics

Browse Source
This commit is contained in:
Emil Tsalapatis 2024-11-25 11:13:51 -08:00
parent 7a87c24c7d
commit b692c415f1
3 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scheds/c/scx_sdt.bpf.c
View File

@ -22,6 +22,8 @@ __u64 stat_##metric; \
DEFINE_SDT_STAT(enqueue); DEFINE_SDT_STAT(enqueue);
DEFINE_SDT_STAT(init); DEFINE_SDT_STAT(init);
DEFINE_SDT_STAT(exit); DEFINE_SDT_STAT(exit);
DEFINE_SDT_STAT(select_idle_cpu);
DEFINE_SDT_STAT(select_busy_cpu);
static SDT_TASK_FN_ATTRS void static SDT_TASK_FN_ATTRS void
stat_global_update(struct sdt_stats __arena *stats) stat_global_update(struct sdt_stats __arena *stats)
@ -30,16 +32,28 @@ stat_global_update(struct sdt_stats __arena *stats)
__sync_fetch_and_add(&stat_enqueue, stats->enqueue); __sync_fetch_and_add(&stat_enqueue, stats->enqueue);
__sync_fetch_and_add(&stat_init, stats->init); __sync_fetch_and_add(&stat_init, stats->init);
__sync_fetch_and_add(&stat_exit, stats->exit); __sync_fetch_and_add(&stat_exit, stats->exit);
__sync_fetch_and_add(&stat_select_idle_cpu, stats->select_idle_cpu);
__sync_fetch_and_add(&stat_select_busy_cpu, stats->select_busy_cpu);
} }
s32 BPF_STRUCT_OPS(sdt_select_cpu, struct task_struct *p, s32 prev_cpu, u64 wake_flags) s32 BPF_STRUCT_OPS(sdt_select_cpu, struct task_struct *p, s32 prev_cpu, u64 wake_flags)
{ {
struct sdt_stats __arena *stats;
bool is_idle = false; bool is_idle = false;
s32 cpu; s32 cpu;
stats = sdt_task_retrieve(p);
if (!stats) {
bpf_printk("%s: no stats for pid %d", p->pid);
return 0;
}
cpu = scx_bpf_select_cpu_dfl(p, prev_cpu, wake_flags, &is_idle); cpu = scx_bpf_select_cpu_dfl(p, prev_cpu, wake_flags, &is_idle);
if (is_idle) { if (is_idle) {
stat_inc_select_idle_cpu(stats);
scx_bpf_dispatch(p, SCX_DSQ_LOCAL, SCX_SLICE_DFL, 0); scx_bpf_dispatch(p, SCX_DSQ_LOCAL, SCX_SLICE_DFL, 0);
} else {
stat_inc_select_busy_cpu(stats);
} }
return cpu; return cpu;

5
scheds/c/scx_sdt.c
View File

@ -73,6 +73,11 @@ restart:
printf("inits=%llu\t", skel->bss->stat_init); printf("inits=%llu\t", skel->bss->stat_init);
printf("exits=%llu\t", skel->bss->stat_exit); printf("exits=%llu\t", skel->bss->stat_exit);
printf("\n"); printf("\n");
printf("select_idle_cpu=%llu\t", skel->bss->stat_select_idle_cpu);
printf("select_busy_cpu=%llu\t", skel->bss->stat_select_busy_cpu);
printf("\n");
fflush(stdout); fflush(stdout);
sleep(1); sleep(1);
} }

2
scheds/c/scx_sdt.h
View File

@ -5,4 +5,6 @@ struct sdt_stats {
__u64 init; __u64 init;
__u64 enqueue; __u64 enqueue;
__u64 exit; __u64 exit;
__u64 select_idle_cpu;
__u64 select_busy_cpu;
}; };