tilelang.backend.gemm
=====================

.. py:module:: tilelang.backend.gemm


Attributes
----------

.. autoapisummary::

   tilelang.backend.gemm.GemmTargetPredicate


Classes
-------

.. autoapisummary::

   tilelang.backend.gemm.GemmImplEntry


Functions
---------

.. autoapisummary::

   tilelang.backend.gemm.register_gemm_impl
   tilelang.backend.gemm.resolve_gemm_impl


Module Contents
---------------

.. py:data:: GemmTargetPredicate

.. py:class:: GemmImplEntry

   .. py:attribute:: name
      :type:  str


   .. py:attribute:: inst_name
      :type:  str


   .. py:attribute:: predicate
      :type:  GemmTargetPredicate


   .. py:attribute:: impl_class
      :type:  type


.. py:function:: register_gemm_impl(name, inst_name, predicate, impl_class)

   Register a backend-specific GEMM implementation class.


.. py:function:: resolve_gemm_impl(gemm_inst, target)

   Resolve the registered implementation class for a GEMM instruction key.