Mojo module

tile_scheduler_splitk

Structs

• ​AdvanceAfterWorkContextSplitK: Context for warps that do work THEN advance (Load/Scheduler/Epilogue).
• ​PrefetchBeforeWorkContextSplitK: Context for MMA warp that prefetches BEFORE work (software pipelining).
• ​SchedulerWorkIteratorSplitK: Work iterator for Scheduler warp (split-K) - owns work_info and both states. Throttle pipeline is obtained from the scheduler.
• ​TileScheduler:
• ​WorkInfo:
• ​WorkIteratorSplitK: Per-warp work iterator for split-K that owns work_info and pipeline state. Throttle pipeline is obtained from the scheduler.

Functions

• ​get_num_tiles:
• ​get_required_locks_buffer_size_bytes:

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit