Struct GpuAffinityManager 

pub struct GpuAffinityManager { /* private fields */ }

GPU affinity manager for NUMA-aware scheduling.

Implementations

impl GpuAffinityManager

pub fn new() -> Self

Create a new affinity manager.

pub fn set_affinity_group(&mut self, group_id: usize, devices: Vec<usize>)

Set devices in the same affinity group.

pub fn get_affinity_group(&self, device: usize) -> Vec<usize>

Get devices in the same affinity group.

pub fn same_affinity(&self, device_a: usize, device_b: usize) -> bool

Check if two devices are in the same affinity group.

pub fn optimal_device( &self, data_device: usize, available: &[usize], ) -> Option<usize>

Get optimal device for data locality.

Trait Implementations

impl Default for GpuAffinityManager

fn default() -> Self

Returns the “default value” for a type.