Struct WorkDistributor 

pub struct WorkDistributor { /* private fields */ }

Work distributor for multi-GPU task scheduling.

Implementations

impl WorkDistributor

pub fn new( manager: Arc<MultiGpuManager>, strategy: DistributionStrategy, ) -> Self

Create a new work distributor.

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

Set affinity group.

pub fn distribute_work<T>(&self, items: Vec<T>) -> Vec<(usize, Vec<T>)>

Distribute work items across GPUs.