For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo package
pipeline
Generic compile-time software pipeline scheduling framework.
Import from submodules directly:
from pipeline.types import OpDesc, ResourceKind, OpRole
from pipeline.config import PipelineConfig, ScheduleConfig
from pipeline.compiler import PipelineSchedule, compile_schedule
from pipeline.schedulers import optimal_schedule_with_halves
from pipeline.program_builder import verify_schedule, build_kernel_programModules
-
compiler: Pipeline schedule compiler: PipelineSchedule trait, ScheduleCompiler struct, and compile_schedule function. -
config: Pipeline configuration structs: LoopCarriedSpec, BlockSizing, FragOrder, SchedulingStrategy, ScheduleConfig, WarpStaggerRule, PipelineConfig, and TargetProfile. -
debug: Diagnostic tooling for compiled pipeline schedules. -
dependency_graph: Loop Dependency Graph (LDG) types: OpNode and LoopBody. -
geometry: Kernel-geometry-derived scheduling constants. -
phase_derivation: Phase derivation: recipes, default prologue/kernel/epilogue, and edge rules. -
pipeline_dsl: Pipeline DSL: ScheduleEntry, EntryBuilder, Pipe, pipe, annotate_pipe. -
program: MMABlockSpec and PipelineProgram — declarative pipeline program representation. -
program_builder: Program builder: constructs PipelineProgram from a loop body. -
schedulers: Scheduling algorithms: greedy and within-iteration optimal schedulers. -
strategies: Strategy structs that group relatedScheduleConfigflags. -
types: Core types for the pipeline scheduling framework.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!