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,
ImplicitlyDestructible,
Movable
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!