Skip to main content
Log in

Mojo struct

FlashAttentionAlgorithm

@register_passable(trivial) struct FlashAttentionAlgorithm

Aliases

  • NAIVE = FlashAttentionAlgorithm(0):
  • FLASH_ATTENTION_1 = FlashAttentionAlgorithm(1):
  • FLASH_ATTENTION_2 = FlashAttentionAlgorithm(2):
  • FLASH_ATTENTION_3 = FlashAttentionAlgorithm(3):

Implemented traits

AnyType, Copyable, ExplicitlyCopyable, Movable, Stringable, UnknownDestructibility, Writable

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)