Skip to main content

Mojo struct

FlashAttentionAlgorithm

@register_passable(trivial) struct FlashAttentionAlgorithm

Implemented traits

AnyType, Defaultable, ExplicitlyCopyable, ImplicitlyCopyable, Movable, Stringable, UnknownDestructibility, Writable

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

FLASH_ATTENTION_1

alias FLASH_ATTENTION_1 = FlashAttentionAlgorithm(1)

FLASH_ATTENTION_2

alias FLASH_ATTENTION_2 = FlashAttentionAlgorithm(2)

FLASH_ATTENTION_3

alias FLASH_ATTENTION_3 = FlashAttentionAlgorithm(3)

NAIVE

alias NAIVE = FlashAttentionAlgorithm(0)

Methods

__init__

__init__() -> Self

__init__(value: Int) -> Self

__eq__

__eq__(self, other: Self) -> Bool

Returns:

Bool

__eq__(self, version: Int) -> Bool

Returns:

Bool

__ne__

__ne__(self, other: Self) -> Bool

Returns:

Bool

__str__

__str__(self) -> String

Returns:

String

init

init(self, type: DType) -> Self

write_to

write_to(self, mut writer: T)

Was this page helpful?