Struct Scheduler 

pub(crate) struct Scheduler<'a> {
    pub(crate) skel: BpfSkel<'a>,
    pub(crate) opts: &'a Opts,
    pub(crate) struct_ops: Option<Link>,
    pub(crate) stats_server: StatsServer<(), Metrics>,
    pub(crate) gpu_index_to_node: Option<HashMap<u32, u32>>,
    pub(crate) previous_gpu_pids: Option<HashMap<u32, u32>>,
    pub(crate) nvml: Option<Nvml>,
    pub(crate) perf_threshold_state: Option<DynamicThresholdState>,
    pub(crate) perf_sticky_threshold_state: Option<DynamicThresholdState>,
}

Fields

skel: BpfSkel<'a>

opts: &'a Opts

struct_ops: Option<Link>

stats_server: StatsServer<(), Metrics>

gpu_index_to_node: Option<HashMap<u32, u32>>

GPU device index -> NUMA node (for NVML PID sync). Only set when –gpu and NUMA enabled.

previous_gpu_pids: Option<HashMap<u32, u32>>

Previous (pid, node) set so we can remove PIDs that stopped using the GPU.

nvml: Option<Nvml>

Reused NVML handle to avoid re-initializing on every sync (expensive).

perf_threshold_state: Option<DynamicThresholdState>

Dynamic threshold state for perf event migrations (when –perf-threshold is 0/dynamic).

perf_sticky_threshold_state: Option<DynamicThresholdState>

Dynamic threshold state for sticky perf events (when –perf-sticky-threshold is 0/dynamic).

Implementations

impl<'a> Scheduler<'a>

pub(crate) fn init( opts: &'a Opts, open_object: &'a mut MaybeUninit<OpenObject>, ) -> Result<Self>

pub(crate) fn sync_gpu_pids(&mut self) -> Result<()>

Sync PID -> GPU (node) map from NVML. When gpu_util_threshold > 0, only PIDs with GPU utilization (SM or memory) >= threshold are added. Map is keyed by task pid. Only processes using a single GPU are added; multi-GPU processes are excluded.

pub(crate) fn add_running_gpu_processes_to_set( device: &Device<'_>, node: u32, pid_to_nodes: &mut HashMap<u32, HashSet<u32>>, )

Record running compute/graphics process PIDs and the GPU node in pid_to_nodes.

pub(crate) fn task_tids(pid: u32) -> Vec<u32>

Return all thread IDs (tids) of the process with the given pid (tgid).

pub(crate) fn enable_primary_cpu( skel: &mut BpfSkel<'_>, cpu: i32, ) -> Result<(), u32>

pub(crate) fn enable_sibling_cpu( skel: &mut BpfSkel<'_>, cpu: usize, sibling_cpu: usize, ) -> Result<(), u32>

pub(crate) fn init_smt_domains( skel: &mut BpfSkel<'_>, topo: &Topology, ) -> Result<(), Error>

pub(crate) fn get_metrics(&self) -> Metrics

pub fn exited(&mut self) -> bool

pub(crate) fn compute_user_cpu_pct( prev: &CpuTimes, curr: &CpuTimes, ) -> Option<u64>

pub(crate) fn read_per_cpu_cpu_times(nr_cpus: usize) -> Option<Vec<CpuTimes>>

Read per-CPU times from /proc/stat (lines “cpu0”, “cpu1”, …). Returns one CpuTimes per CPU, in order.

pub(crate) fn run(&mut self, shutdown: Arc<AtomicBool>) -> Result<UserExitInfo>

Trait Implementations

impl Drop for Scheduler<'_>

fn drop(&mut self)

Executes the destructor for this type.