Struct MultiGpuBuilder
pub struct MultiGpuBuilder {
config: MultiGpuConfig,
}Expand description
Builder for multi-GPU coordinator.
Fields§
§config: MultiGpuConfigImplementations§
§impl MultiGpuBuilder
impl MultiGpuBuilder
pub fn new() -> MultiGpuBuilder
pub fn new() -> MultiGpuBuilder
Create a new builder.
pub fn load_balancing(self, strategy: LoadBalancingStrategy) -> MultiGpuBuilder
pub fn load_balancing(self, strategy: LoadBalancingStrategy) -> MultiGpuBuilder
Set load balancing strategy.
pub fn auto_select_device(self, enable: bool) -> MultiGpuBuilder
pub fn auto_select_device(self, enable: bool) -> MultiGpuBuilder
Set auto device selection.
pub fn max_kernels_per_device(self, max: usize) -> MultiGpuBuilder
pub fn max_kernels_per_device(self, max: usize) -> MultiGpuBuilder
Set max kernels per device.
pub fn enable_p2p(self, enable: bool) -> MultiGpuBuilder
pub fn enable_p2p(self, enable: bool) -> MultiGpuBuilder
Enable P2P transfers.
pub fn preferred_devices(self, devices: Vec<usize>) -> MultiGpuBuilder
pub fn preferred_devices(self, devices: Vec<usize>) -> MultiGpuBuilder
Set preferred devices.
pub fn build(self) -> Arc<MultiGpuCoordinator>
pub fn build(self) -> Arc<MultiGpuCoordinator>
Build the coordinator.
Trait Implementations§
§impl Default for MultiGpuBuilder
impl Default for MultiGpuBuilder
§fn default() -> MultiGpuBuilder
fn default() -> MultiGpuBuilder
Returns the “default value” for a type. Read more
Auto Trait Implementations§
impl Freeze for MultiGpuBuilder
impl RefUnwindSafe for MultiGpuBuilder
impl Send for MultiGpuBuilder
impl Sync for MultiGpuBuilder
impl Unpin for MultiGpuBuilder
impl UnwindSafe for MultiGpuBuilder
Blanket Implementations§
§impl<T> ArchivePointee for T
impl<T> ArchivePointee for T
§type ArchivedMetadata = ()
type ArchivedMetadata = ()
The archived version of the pointer metadata for this type.
§fn pointer_metadata(
_: &<T as ArchivePointee>::ArchivedMetadata,
) -> <T as Pointee>::Metadata
fn pointer_metadata( _: &<T as ArchivePointee>::ArchivedMetadata, ) -> <T as Pointee>::Metadata
Converts some archived metadata to the pointer metadata for itself.
§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
§impl<F, W, T, D> Deserialize<With<T, W>, D> for F
impl<F, W, T, D> Deserialize<With<T, W>, D> for F
§fn deserialize(
&self,
deserializer: &mut D,
) -> Result<With<T, W>, <D as Fallible>::Error>
fn deserialize( &self, deserializer: &mut D, ) -> Result<With<T, W>, <D as Fallible>::Error>
Deserializes using the given deserializer
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
§impl<T> LayoutRaw for T
impl<T> LayoutRaw for T
§fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
Gets the layout of the type.