Mojo struct
SingleBufferSchedule
struct SingleBufferSchedule[num_k_tiles: Int]
Declarative schedule for the default single-buffer matmul.
The algorithm declares logical ops (tag + buffer metadata only). The target cost model supplies resource, latency, and role. The framework derives the pipeline ordering and dependency edges.
Fields
- hints (
AMDScheduleHints):
Implemented traits
AnyType,
ImplicitlyDestructible,
PipelineSchedule
Methods
__init__
__init__(out self, hints: AMDScheduleHints, target: TargetProfile = mi355x_single_buffer_target())
config
config(self) -> PipelineConfig
Returns:
PipelineConfig
build_body
build_body(self) -> List[OpDesc]
Declare logical ops, annotate with cost model, reorder for pipeline.
Returns:
transform_kernel
transform_kernel(self, ker: List[ScheduleEntry], body: List[OpDesc]) -> List[ScheduleEntry]
Append AMD schedule_group_barrier hints to kernel entries.
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!