Mojo struct

MLA_Decode_Pack

@register_passable(trivial) struct MLA_Decode_Pack[ValidLengthType: OptionalPointer, MaskType: MHAMask, ScoreModType: ScoreModTrait, SplitAccumType: OptionalPointer]

Fields

mask (MaskType):
score_mod (ScoreModType):
valid_length (ValidLengthType):
lse_accum_split_ptr (SplitAccumType):

Implemented traits

AnyType, Copyable, DevicePassable, ImplicitlyCopyable, ImplicitlyDestructible, Movable, RegisterPassable, TrivialRegisterPassable

`comptime` members

`__copy_ctor_is_trivial`

comptime __copy_ctor_is_trivial = (((MaskType.__copy_ctor_is_trivial & ScoreModType.__copy_ctor_is_trivial) & ValidLengthType.__copy_ctor_is_trivial) & SplitAccumType.__copy_ctor_is_trivial)

`delis_trivial`

comptime __del__is_trivial = (((MaskType.__del__is_trivial & ScoreModType.__del__is_trivial) & ValidLengthType.__del__is_trivial) & SplitAccumType.__del__is_trivial)

`__move_ctor_is_trivial`

comptime __move_ctor_is_trivial = (((MaskType.__move_ctor_is_trivial & ScoreModType.__move_ctor_is_trivial) & ValidLengthType.__move_ctor_is_trivial) & SplitAccumType.__move_ctor_is_trivial)

`device_type`

comptime device_type = MLA_Decode_Pack[ValidLengthType, MaskType, ScoreModType, SplitAccumType]

Methods

`init`

__init__(mask: MaskType, score_mod: ScoreModType, valid_length: ValidLengthType, lse_accum_split_ptr: SplitAccumType) -> Self

`get_type_name`

static get_type_name() -> String

Returns:

String

`get_device_type_name`

static get_device_type_name() -> String

Returns:

String

Fields​

Implemented traits​

comptime members​

__copy_ctor_is_trivial​

__del__is_trivial​

__move_ctor_is_trivial​

device_type​

Methods​

__init__​

get_type_name​

get_device_type_name​