wash/mango
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

kernel: add functions to query the number of available/online CPUs

Browse Source
This commit is contained in:
max 🍓
2023-05-05 15:25:44 +01:00
parent 2ca1bd0b06
commit cec6b644ac
2 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/socks/cpu.h
View File

@@ -39,6 +39,9 @@ extern bool cpu_is_online(unsigned int cpu_id);
extern void cpu_set_available(unsigned int cpu_id); extern void cpu_set_available(unsigned int cpu_id);
extern void cpu_set_online(unsigned int cpu_id); extern void cpu_set_online(unsigned int cpu_id);
extern unsigned int cpu_nr_available(void);
extern unsigned int cpu_nr_online(void);
extern cycles_t get_cycles(void); extern cycles_t get_cycles(void);
static inline cycles_t cycles_diff(cycles_t then, cycles_t now) static inline cycles_t cycles_diff(cycles_t then, cycles_t now)
{ {

10
kernel/cpu.c
View File

@@ -58,6 +58,16 @@ void cpu_set_online(unsigned int cpu_id)
bitmap_set(cpu_online, cpu_id); bitmap_set(cpu_online, cpu_id);
} }
unsigned int cpu_nr_available(void)
{
return bitmap_count_set(cpu_available, CPU_MAX);
}
unsigned int cpu_nr_online(void)
{
return bitmap_count_set(cpu_online, CPU_MAX);
}
void preempt_disable(void) void preempt_disable(void)
{ {
ml_cpu_block *ml_cpu = ml_this_cpu(); ml_cpu_block *ml_cpu = ml_this_cpu();