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 function

build_schedule

def build_schedule[geometry: KernelGeometry](config: ScheduleConfig = Pipeline4Wave._default_schedule_config(), target: TargetProfile = mi355x_target(4, 4, 1)) -> ScheduleCompiler

Compiles the 4-wave pipeline schedule.

Takes a KernelGeometry (kernel-shape-derived constants) as its only template parameter; the framework reads lgkm_per_load_* from it directly. Use as a comptime value; the kernel reads entries via comptime for.

Parameters:

geometry (KernelGeometry): Kernel-shape-derived constants (lgkm/vm costs, etc.).

Args:

config (ScheduleConfig): Schedule-level overrides; defaults to the cross-stage rotation invariants.
target (TargetProfile): Target hardware profile; defaults to MI355X.

Returns:

ScheduleCompiler: A ScheduleCompiler with prologue, kernel, and epilogue lists.