Index A | C | D | F | G | I | M | P | S | U A apply_padding() (in module magi_attention.api.functools) C calc_attn() (in module magi_attention.api.magi_attn_interface) compute_pad_size() (in module magi_attention.api.functools) D dispatch() (in module magi_attention.api.magi_attn_interface) F flex_flash_attn_func() (in module magi_attention.functional.flex_flash_attn) G get_most_recent_key() (in module magi_attention.api.magi_attn_interface) get_position_ids() (in module magi_attention.api.magi_attn_interface) I infer_attn_mask_from_cu_seqlens() (in module magi_attention.api.functools) infer_attn_mask_from_sliding_window() (in module magi_attention.api.functools) infer_varlen_mask_from_batch() (in module magi_attention.api.functools) M magi_attention.api module magi_attn_flex_key() (in module magi_attention.api.magi_attn_interface) magi_attn_varlen_key() (in module magi_attention.api.magi_attn_interface) make_flex_key_for_new_mask_after_dispatch() (in module magi_attention.api.magi_attn_interface) make_varlen_key_for_new_mask_after_dispatch() (in module magi_attention.api.magi_attn_interface) module magi_attention.api P pad_at_dim() (in module magi_attention.api.functools) S squash_batch_dim() (in module magi_attention.api.functools) U undispatch() (in module magi_attention.api.magi_attn_interface) unpad_at_dim() (in module magi_attention.api.functools)
