skainet-compile-hlo/sk.ainet.compile.hlo/StableHloOptimizer

StableHloOptimizer

class StableHloOptimizer(source)

Optimizer for StableHLO MLIR modules.

This class provides optimization passes for StableHLO code, including operation fusion, constant folding, and other transformations.

Constructors

StableHloOptimizer

constructor()

Types

object Companion

Create a default optimizer with common optimization passes

Functions

fun addPass(pass: OptimizationPass)

Add an optimization pass

fun optimize(module: StableHloModule): StableHloModule

Optimize a StableHLO module by applying all registered passes