Mojo function

build_schedule

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.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit