IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /max/get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
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 struct

AMDScheduleHints

struct AMDScheduleHints

Hardware expansion factors for AMD schedule_group_barrier hints.

Each body op (COMPUTE, FRAGMENT_LOAD, STORE_SMEM) expands to multiple hardware instructions. These factors tell the framework how many, so it can compute the interleaving ratios automatically.

Example for a 4x4x2 MMA config with 16x16 tiles: AMDScheduleHints(m_mmas=4, n_mmas=4, k_mmas=2, mma_m=16, mma_n=16, a_loads=2, b_loads=2)

Fields​

  • ​m_mmas (Int):
  • ​n_mmas (Int):
  • ​k_mmas (Int):
  • ​mma_m (Int):
  • ​mma_n (Int):
  • ​a_loads_per_thread (Int):
  • ​b_loads_per_thread (Int):

Implemented traits​

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDeletable, Movable