pub struct FusionConfig {
pub enable_memory_coalescing: bool,
pub enable_vectorization: bool,
pub batch_size: usize,
pub enable_kernel_fusion: bool,
pub buffer_size: usize,
pub enable_async_updates: bool,
}Expand description
Configuration for fused optimizer operations
Fields§
§enable_memory_coalescing: boolEnable memory bandwidth optimization
enable_vectorization: boolUse vectorized operations when possible
batch_size: usizeBatch size for parameter updates
enable_kernel_fusion: boolEnable kernel fusion for compatible operations
buffer_size: usizeBuffer size for batched operations
enable_async_updates: boolEnable asynchronous updates
Trait Implementations§
Source§impl Clone for FusionConfig
impl Clone for FusionConfig
Source§fn clone(&self) -> FusionConfig
fn clone(&self) -> FusionConfig
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for FusionConfig
impl Debug for FusionConfig
Source§impl Default for FusionConfig
impl Default for FusionConfig
Source§impl<'de> Deserialize<'de> for FusionConfig
impl<'de> Deserialize<'de> for FusionConfig
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for FusionConfig
impl RefUnwindSafe for FusionConfig
impl Send for FusionConfig
impl Sync for FusionConfig
impl Unpin for FusionConfig
impl UnwindSafe for FusionConfig
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> ConfigSerializable for Twhere
T: Serialize + for<'de> Deserialize<'de>,
impl<T> ConfigSerializable for Twhere
T: Serialize + for<'de> Deserialize<'de>,
Source§fn save_to_file(&self, path: &Path) -> Result<(), TrustformersError>
fn save_to_file(&self, path: &Path) -> Result<(), TrustformersError>
Save to file
Source§fn load_from_file(path: &Path) -> Result<Self, TrustformersError>where
Self: Sized,
fn load_from_file(path: &Path) -> Result<Self, TrustformersError>where
Self: Sized,
Load from file
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
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 more