Struct AffinityManager 

pub struct AffinityManager { /* private fields */ }

GPU affinity manager for thread-GPU binding

Implementations

impl AffinityManager

pub fn new(devices: Vec<Arc<GpuDevice>>) -> Self

Create a new affinity manager

pub fn pin_thread(&self, gpu_index: usize) -> Result<(), String>

Pin current thread to a specific GPU

pub fn unpin_thread(&self)

Unpin current thread

pub fn get_thread_gpu(&self) -> Option<usize>

Get GPU index for current thread (if pinned)

pub fn get_thread_device(&self) -> Option<Arc<GpuDevice>>

Get device for current thread (if pinned)

pub fn get_gpu_threads(&self, gpu_index: usize) -> Vec<ThreadId>

Get all threads pinned to a GPU

pub fn auto_pin_thread(&self) -> Result<usize, String>

Auto-pin current thread to least loaded GPU

pub fn get_stats(&self) -> AffinityStats

Get affinity statistics

pub fn print_affinity_info(&self)

Print affinity information

pub fn clear_all(&self)

Clear all affinity mappings

pub fn device_count(&self) -> usize

Get total number of devices