pub struct FusionOptimizer { /* private fields */ }Expand description
Kernel fusion analyzer and optimizer.
Implementations§
Source§impl FusionOptimizer
impl FusionOptimizer
Sourcepub fn new(config: FusionConfig) -> Self
pub fn new(config: FusionConfig) -> Self
Create a new fusion optimizer.
Sourcepub fn with_cost_model(
config: FusionConfig,
cost_model: FusionCostModel,
) -> Self
pub fn with_cost_model( config: FusionConfig, cost_model: FusionCostModel, ) -> Self
Create with custom cost model.
Sourcepub fn analyze(&mut self, graph: &EinsumGraph) -> Vec<FusionCandidate>
pub fn analyze(&mut self, graph: &EinsumGraph) -> Vec<FusionCandidate>
Analyze graph and identify fusion opportunities.
Sourcepub fn apply_fusions(
&self,
graph: &EinsumGraph,
_candidates: &[FusionCandidate],
) -> Result<EinsumGraph, FusionError>
pub fn apply_fusions( &self, graph: &EinsumGraph, _candidates: &[FusionCandidate], ) -> Result<EinsumGraph, FusionError>
Apply fusion candidates to create optimized graph.
Sourcepub fn stats(&self) -> FusionStats
pub fn stats(&self) -> FusionStats
Get fusion statistics.
Auto Trait Implementations§
impl Freeze for FusionOptimizer
impl RefUnwindSafe for FusionOptimizer
impl Send for FusionOptimizer
impl Sync for FusionOptimizer
impl Unpin for FusionOptimizer
impl UnwindSafe for FusionOptimizer
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