Mojo struct

MMAConfig

@register_passable(trivial) struct MMAConfig[InType: DType, OutType: DType, mma_shape: IndexList[3], transpose_b: Bool = True]

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

`k_group_size_a`

alias k_group_size_a = (MMAConfig[InType, OutType, mma_shape, transpose_b].simd_width // MMAConfig[InType, OutType, mma_shape, transpose_b].registers_per_thread_a)

`k_group_size_b`

alias k_group_size_b = (MMAConfig[InType, OutType, mma_shape, transpose_b].simd_width // MMAConfig[InType, OutType, mma_shape, transpose_b].registers_per_thread_b)

`mma`

alias mma = TensorCore[OutType, InType, mma_shape, transpose_b]()

`registers_per_thread_a`

alias registers_per_thread_a = num_matrix_reg[mma_shape.__getitem__[3, int64, Int](0), mma_shape.__getitem__[3, int64, Int](2)]()

`registers_per_thread_b`

alias registers_per_thread_b = num_matrix_reg[mma_shape.__getitem__[3, int64, Int](1), mma_shape.__getitem__[3, int64, Int](2)]()

`simd_width`

alias simd_width = simd_width_of[InType]()

Methods

`adjusted_mma_k_shape_a`

static adjusted_mma_k_shape_a() -> Int

Returns:

Int

`adjusted_mma_k_shape_b`

static adjusted_mma_k_shape_b() -> Int

Returns:

Int

Implemented traits​

Aliases​

__copyinit__is_trivial​

__del__is_trivial​

__moveinit__is_trivial​

k_group_size_a​

k_group_size_b​

mma​

registers_per_thread_a​

registers_per_thread_b​

simd_width​

Methods​

adjusted_mma_k_shape_a​

adjusted_mma_k_shape_b​