Mojo struct

MMAConfig

struct MMAConfig[InType: DType, OutType: DType, mma_shape: IndexList[3], transpose_b: Bool = True]

Implemented traits

AnyType, UnknownDestructibility

Aliases

`delis_trivial`

alias __del__is_trivial = True

`k_group_size_a`

alias k_group_size_a = (simd_width_of[InType]() // num_matrix_reg[mma_shape.__getitem__[3, DType.int64, Int](0), mma_shape.__getitem__[3, DType.int64, Int](2)]())

`k_group_size_b`

alias k_group_size_b = (simd_width_of[InType]() // num_matrix_reg[mma_shape.__getitem__[3, DType.int64, Int](1), mma_shape.__getitem__[3, DType.int64, Int](2)]())

`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, DType.int64, Int](0), mma_shape.__getitem__[3, DType.int64, Int](2)]()

`registers_per_thread_b`

alias registers_per_thread_b = num_matrix_reg[mma_shape.__getitem__[3, DType.int64, Int](1), mma_shape.__getitem__[3, DType.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​

__del__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​