Mojo function
dispatch_materialized_mask
dispatch_materialized_mask[dtype: DType, layout: Layout, //, callback_fn: def[mask_t: MHAMask](mask: mask_t) raises capturing -> None](mask_nd: LayoutTensor[dtype, layout, mask_nd.origin], start_pos_nd: OptionalReg[LayoutTensor[DType.uint32, Layout.row_major(-1), ImmutAnyOrigin]] = None)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!