Struct MultiGpuBuilder 

pub struct MultiGpuBuilder { /* private fields */ }

Builder for multi-GPU coordinator.

Implementations

impl MultiGpuBuilder

pub fn new() -> Self

Create a new builder.

pub fn load_balancing(self, strategy: LoadBalancingStrategy) -> Self

Set load balancing strategy.

pub fn auto_select_device(self, enable: bool) -> Self

Set auto device selection.

pub fn max_kernels_per_device(self, max: usize) -> Self

Set max kernels per device.

pub fn enable_p2p(self, enable: bool) -> Self

Enable P2P transfers.

pub fn preferred_devices(self, devices: Vec<usize>) -> Self

Set preferred devices.

pub fn build(self) -> Arc<MultiGpuCoordinator>

Build the coordinator.

Trait Implementations

impl Default for MultiGpuBuilder

fn default() -> Self

Returns the “default value” for a type.