Mojo function
kv_coord
kv_coord[*, depth: Int, swizzle_granularity: Int](row: UInt32, head_idx: UInt32) -> StaticTuple[UInt32, 4 if _should_split_last_dim(depth, swizzle_granularity) else 3]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!