For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo function
determine_thread_role
def determine_thread_role[producer_a_warps: Int, producer_b_warps: Int]() -> Tuple[ThreadRole, Int]
Returns (role, consumer_warp_id within role group).
Parameters:
- producer_a_warps (
Int): Number of warps assigned to loading matrix A tiles. - producer_b_warps (
Int): Number of warps assigned to loading matrix B tiles.
Returns: