Skip to main content

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:

List

transform_kernel

transform_kernel(self, ker: List[ScheduleEntry], body: List[OpDesc]) -> List[ScheduleEntry]

Append AMD schedule_group_barrier hints to kernel entries.

Returns:

List

Was this page helpful?