Struct FusionConfig 

pub struct FusionConfig {
    pub logger: LoggerConfig<FusionLogLevel>,
    pub beam_search: BeamSearchConfig,
    pub max_graph_size: Option<usize>,
    pub growth_patience: usize,
}

Configuration for operation fusion in Burn.

Fields

logger: LoggerConfig<FusionLogLevel>

Logger configuration for fusion logs.

beam_search: BeamSearchConfig

Beam search configuration used when exploring fusion opportunities.

max_graph_size: Option<usize>

Maximum number of operations in a single client-cached graph (router graph caching).

The greedy graph fuser keeps accumulating ops into one cached graph; once it reaches this many ops the graph is closed and dispatched. None (the default) sets no size cap, leaving growth_patience to decide when a graph stops being worth growing; set a value to also bound the per-graph memory and the cost of building and replaying any single graph.

growth_patience: usize

Close the current graph once its fusion score hasn’t reached a new maximum for this many consecutive ops (router graph caching).

The fuser scores the accumulated ops as they grow and tracks the best score so far; once this many ops have been added without beating it, the graph has stopped getting more worth caching and is closed. Higher values keep growing the graph longer in search of a better score.

Trait Implementations

impl Clone for FusionConfig

fn clone(&self) -> FusionConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for FusionConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for FusionConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for FusionConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for FusionConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.