.. _apigemm:

GEMM
=========

For framework integrations that already target DeepGEMM style Python APIs,
prefer the ``deep-gemm`` wrapper first. Use the MATE APIs below when wrapper
coverage is not enough.

MoE GEMM
-------------------------

.. currentmodule:: mate.gemm
.. autofunction:: ragged_m_moe_gemm_8bit
.. autofunction:: ragged_k_moe_gemm_8bit
.. autofunction:: masked_moe_gemm_8bit

Dense GEMM
-------------------------

.. autofunction:: bmm_fp16
.. autofunction:: bmm_fp8
.. autofunction:: gemm_fp8_nt_groupwise

DeepGemm Lighting Indexer
-------------------------

.. currentmodule:: mate.deep_gemm
.. autofunction:: get_paged_mqa_logits_metadata
.. autofunction:: fp8_paged_mqa_logits