Skip to main content

Module gpu

Module gpu 

Source

Re-exportsยง

use std::collections::HashMap; ๐Ÿ”’
use std::collections::HashSet; ๐Ÿ”’
use std::path::Path; ๐Ÿ”’
use std::path::PathBuf; ๐Ÿ”’
use anyhow::Result; ๐Ÿ”’
use log::debug; ๐Ÿ”’
use crate::cgroup::CgroupReader; ๐Ÿ”’

Functionsยง

direct_gpu_processes ๐Ÿ”’
Keep only NVML processes whose GPUs resolve to one NUMA node.
expand_gpu_processes ๐Ÿ”’
Expand NVML processes to peer processes in the same exact cgroup.
expand_gpu_processes_with ๐Ÿ”’
fit_gpu_processes ๐Ÿ”’
Fit workload hints into the BPF map without partially expanding a cgroup.