Skip to main content

hetorch.passes.pipeline

Pass pipeline for executing transformation passes

Classes​

PassPipeline(passes: typing.List[hetorch.passes.base.TransformationPass])​

Ordered sequence of transformation passes

The pipeline executes passes in order, validating preconditions and tracking which passes have been executed.

Methods:

__init__(self, passes: List[hetorch.passes.base.TransformationPass])​

Initialize pipeline

Args: passes: List of pass instances

reset(self) -\> None​

Reset execution tracking

run(self, graph_module: torch.fx.graph_module.GraphModule, context: hetorch.compiler.context.CompilationContext) -\> torch.fx.graph_module.GraphModule​

Execute all passes in order

Args: graph_module: Input graph module context: Compilation context

Returns: Transformed graph module

Raises: PassValidationError: If any pass validation fails