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!