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 module
phase_derivation
Phase derivation: recipes, default prologue/kernel/epilogue, and edge rules.
Contains:
- PhaseAction, PhaseStep: declarative recipe primitives
- apply_phase_recipe: recipe evaluator
- single_buffer_prologue_recipe, single_buffer_epilogue_recipe
- pipe_to_list, default_prologue, default_kernel, default_epilogue
- _strip_drain_fuse_blocks, _expand_epilogue_blocks
- derive_epilogue_from_program
- double_buffer_edge_rules, single_buffer_edge_rules, apply_edge_rules
- derive_edges_from_ops
- default_warp_stagger, default_warp_stagger_double_buffer
- derive_prologue_from_program
Structs
-
PhaseAction: Action type for a recipe step. -
PhaseStep: One step in a phase recipe.
Functions
-
apply_edge_rules: Apply declarative edge rules to a loop body, producing DepEdge list. -
apply_phase_recipe: Evaluate a phase recipe against a body, producing ScheduleEntries. -
default_epilogue: Derive epilogue for single-buffer pipeline (2 drain passes). -
default_kernel: Direct 1:1 mapping of body ops to kernel phase entries. -
default_prologue: Derive prologue for single-buffer pipeline from body + config. -
default_warp_stagger: Default warp stagger: right after prologue. -
default_warp_stagger_double_buffer: Count stage-0 prefetch loads in body for double-buffer warp stagger. -
derive_cross_stage_rotation_edges: Returns cross-partition FRAG→MMA + same-partition MMA→FRAG ANTI edges for cross-stage rotation patterns. -
derive_edges_from_ops: Derive all dependency edges from op metadata and pipeline config. -
derive_epilogue_from_program: Auto-derive epilogue from a kernel program. -
derive_prologue_from_program: Derive prologue from a finalized PipelineProgram. -
double_buffer_edge_rules: 5 rules encoding the 4 phases of double-buffer edge derivation. -
filter_spurious_cross_stage_flow: Drops spurious same-partition FLOW edges that the default Phase 1 rule emits for cross-stage frags. -
pipe_to_list: Convert a Pipe to a List[OpDesc]. -
single_buffer_edge_rules: 8 rules encoding single-buffer structural dependencies. -
single_buffer_epilogue_recipe: Recipe for single-buffer epilogue (2 drain passes). -
single_buffer_prologue_recipe: Recipe for single-buffer prologue.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!