Skip to main content

Module optimization

Module optimization 

Source
Expand description

Graph optimization and fusion detection utilities.

This module provides utilities for analyzing and optimizing EinsumGraph structures:

  • Fusion opportunities detection (combining adjacent operations)
  • Dead node elimination
  • Redundant computation detection
  • Operation reordering for better cache locality

Structs§

FusionOpportunity
Fusion opportunity between two nodes
FusionPlanner
Fusion planner for actually applying fusion transformations
GraphOptimizer
Graph optimizer for detecting optimization opportunities
OptimizationResult
Optimization pass result

Enums§

FusionType
Type of fusion that can be applied