Mojo struct

STMatrixOffsets

@register_passable(trivial) struct STMatrixOffsets[BM: Int, BN: Int, *, num_threads: Int, accum_type_size: Int, curr_repeat: Int, cumulative_repeat: Int, m_mma: Int]

Implemented traits

AnyType, Copyable, ImplicitlyCopyable, Movable, UnknownDestructibility

Aliases

__copyinit__is_trivial

alias __copyinit__is_trivial = True

__del__is_trivial

alias __del__is_trivial = True

__moveinit__is_trivial

alias __moveinit__is_trivial = True

b32_per_repeat

alias b32_per_repeat = ((4 * accum_type_size) // 4)

local_frag_size_b32

alias local_frag_size_b32 = (curr_repeat * ((4 * accum_type_size) // 4))

ptr_offset

alias ptr_offset = (((4 * accum_type_size) // 4) * (((BN // 8) * m_mma) + cumulative_repeat))

STLayout

alias STLayout = STMatrixLayout[BM, BN, num_threads=num_threads, accum_type_size=accum_type_size]

tmem_col_offset

alias tmem_col_offset = ((cumulative_repeat * 2) * 4)

tmem_offset

alias tmem_offset = (((16 * m_mma) << 16) + ((cumulative_repeat * 2) * 4))

tmem_row_offset

alias tmem_row_offset = (16 * m_mma)

Methods

__init__

__init__() -> Self

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