Mojo struct
FlashAttentionAlgorithm
struct FlashAttentionAlgorithm
Implemented traits
AnyType,
Copyable,
Defaultable,
ImplicitlyCopyable,
ImplicitlyDestructible,
Movable,
RegisterPassable,
TrivialRegisterPassable,
Writable
comptime members
FLASH_ATTENTION_1
comptime FLASH_ATTENTION_1 = FlashAttentionAlgorithm(1)
FLASH_ATTENTION_2
comptime FLASH_ATTENTION_2 = FlashAttentionAlgorithm(2)
FLASH_ATTENTION_3
comptime FLASH_ATTENTION_3 = FlashAttentionAlgorithm(3)
NAIVE
comptime NAIVE = FlashAttentionAlgorithm(0)
Methods
__init__
__init__() -> Self
__init__(value: Int) -> Self
__eq__
__ne__
init
init(self, dtype: DType) -> Self
write_to
write_to(self, mut writer: T)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!