Mojo function
fa4_correction
fa4_correction[qkv_type: DType, MaskType: MHAMask, config: FA4Config, page_size: Int](mbars: FA4MiscMBars[num_qk_stages=config.num_qk_stages, num_pv_stages=config.num_pv_stages, num_kv_stages=config.num_kv_stages, use_order_barriers=EnableForcedOrdering], score_row: UInt32, num_keys: UInt32, mask: MaskType)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!