Skip to main content

Mojo function

build_schedule

build_schedule[is_fp8: Bool, lgkm_a: Int, lgkm_b: Int](config: ScheduleConfig = ScheduleConfig(scheduling=SchedulingStrategy.CSP, sched_barrier_mask=85, auto_waits=True, drain_lgkm_mask=0, auto_drain=False, lds_contention_penalty=0, wait_lgkm_first=8, wait_vm_last=6, lgkm_per_load_a=0, lgkm_per_load_b=0, lgkm_after_last=False), target: TargetProfile = mi355x_target(4, 4, 1)) -> ScheduleCompiler

Compile the declarative constraint-based schedule.

The caller provides:

  • ScheduleConfig: tuning knobs (scheduling strategy, auto_waits)
  • TargetProfile: unified hardware target (cost model + pipeline config)

The framework automatically derives the execution ordering and wait counts from op constraints. This is the constraint-based alternative to build_interleaved_schedule().

Returns:

ScheduleCompiler

Was this page helpful?