Mojo function
get_q_head_idx
get_q_head_idx[BM: Int, BN: Int, depth: Int, padded_depth: Int, num_heads: Int, group: Int, decoding: Bool, //](position: MHAPosition[BM, BN, depth, padded_depth, num_heads, group, decoding], lane: UInt32) -> StaticTuple[UInt32, MHAPosition[BM, BN, depth, padded_depth, num_heads, group, decoding].num_q_heads_per_thread]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!