.. _apiattention:

Attention
=========

FMHA
----

.. currentmodule:: mate.mha_interface

.. autofunction:: flash_attn_with_kvcache
.. autofunction:: flash_attn_varlen_func


MLA
---
.. currentmodule:: mate.flashmla

.. autofunction:: get_mla_metadata
.. autofunction:: flash_mla_with_kvcache