pub struct GpuConfig {
pub device_id: usize,
pub memory_pool_size: Option<usize>,
pub enable_memory_optimization: bool,
pub batch_size: usize,
pub use_half_precision: bool,
pub enable_async: bool,
pub tensor_cores: TensorCoresConfig,
pub memory_strategy: MemoryStrategy,
pub dynamic_batching: bool,
pub graph_optimization: GraphOptimizationLevel,
}Expand description
GPU device configuration
Fields§
§device_id: usizeDevice ID to use
memory_pool_size: Option<usize>Memory pool size in bytes
enable_memory_optimization: boolEnable memory optimization
batch_size: usizeBatch size for GPU operations
use_half_precision: boolUse half precision (FP16) for faster computation
enable_async: boolEnable asynchronous execution
tensor_cores: TensorCoresConfigTensor cores configuration
memory_strategy: MemoryStrategyMemory allocation strategy
dynamic_batching: boolEnable dynamic batch sizing
graph_optimization: GraphOptimizationLevelGraph optimization level
Trait Implementations§
Auto Trait Implementations§
impl Freeze for GpuConfig
impl RefUnwindSafe for GpuConfig
impl Send for GpuConfig
impl Sync for GpuConfig
impl Unpin for GpuConfig
impl UnwindSafe for GpuConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.