Skip to main content

Mojo module

fa4_config

FA4 (Flash Attention 4) configuration for SM100 (Blackwell) kernels.

comptime values

EnableEarlyAdd

comptime EnableEarlyAdd = get_defined_bool["FA4AddEarly", False]()

EnableForcedOrdering

comptime EnableForcedOrdering = get_defined_bool["FA4ForcedSoftmaxOrdering", False]()

Structs

Was this page helpful?