Index C | D | F | G | M | U 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) M magi_attention.api module magi_attn_flex_dispatch() (in module magi_attention.api.magi_attn_interface) magi_attn_flex_key() (in module magi_attention.api.magi_attn_interface) magi_attn_varlen_dispatch() (in module magi_attention.api.magi_attn_interface) magi_attn_varlen_key() (in module magi_attention.api.magi_attn_interface) module magi_attention.api U undispatch() (in module magi_attention.api.magi_attn_interface)