Mojo module

tile_scheduler_splitk

Structs

• ​SchedulerWorkIteratorSplitK: Work iterator for Scheduler warp (split-K) using next-style iteration.
• ​TileScheduler:
• ​WaitAndAdvanceContextSplitK: Context for waiting on CLC barrier and advancing work iterator (Split-K).
• ​WorkInfo:
• ​WorkIteratorSplitK: Per-warp work iterator for split-K using next-style iteration.

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