Mojo struct

TileScheduler

@register_passable(trivial) struct TileScheduler[M: Int, tile_shape: IndexList[3], cluster: IndexList[3] = Index(1, 1, 1), cta_group: Int = 1, swizzle: Bool = False]

Fields

num_active_experts (Int):
group_offsets (NDBuffer[DType.uint32, 1, MutableAnyOrigin]):
current_iter (Int32):
current_group_idx (UInt32):
current_n_cumsum (UInt32):
block_idx_start (UInt32):

Implemented traits

AnyType, Copyable, ImplicitlyCopyable, Movable, UnknownDestructibility

Aliases

`copyinitis_trivial`

alias __copyinit__is_trivial = True

`delis_trivial`

alias __del__is_trivial = True

`moveinitis_trivial`

alias __moveinit__is_trivial = True

`div_block_n`

alias div_block_n = FastDiv[DType.uint32]((tile_shape.__getitem__[3, DType.int64, Int](1) * cta_group))

`kNum1DBlocksPerGroup`

alias kNum1DBlocksPerGroup = 16

`num_m_blocks`

alias num_m_blocks = SIMD[DType.uint32, 1](ceildiv(M, tile_shape.__getitem__[3, DType.int64, Int](0)))

`tile_n`

alias tile_n = (tile_shape.__getitem__[3, DType.int64, Int](1) * cta_group)

Methods

`init`

__init__(num_active_experts: Int, group_offsets: NDBuffer[DType.uint32, 1, MutableAnyOrigin]) -> Self

`fetch_next_work`

fetch_next_work(mut self) -> WorkInfo

Returns:

WorkInfo

Fields​

Implemented traits​

Aliases​

__copyinit__is_trivial​

__del__is_trivial​

__moveinit__is_trivial​

div_block_n​

kNum1DBlocksPerGroup​

num_m_blocks​

tile_n​

Methods​

__init__​

fetch_next_work​