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?
Thank you! We'll create more content like this.
Thank you for helping us improve!