Mojo struct
FlashAttentionAlgorithm
@register_passable(trivial)
struct FlashAttentionAlgorithm
Implemented traits
AnyType
,
Copyable
,
ExplicitlyCopyable
,
Movable
,
Stringable
,
UnknownDestructibility
,
Writable
Aliases
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
@implicit
__init__(value: Int) -> Self
__eq__
__eq__(self, other: Self) -> Bool
__ne__
__ne__(self, other: Self) -> Bool
__str__
__str__(self) -> String
write_to
write_to[W: Writer](self, mut writer: W)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!