tilelang.tileop.gemm.gemm_wmma
==============================

.. py:module:: tilelang.tileop.gemm.gemm_wmma

.. autoapi-nested-parse::

   GEMM implementation using AMD RDNA WMMA instructions (gfx11/gfx12).



Classes
-------

.. autoapisummary::

   tilelang.tileop.gemm.gemm_wmma.GemmWMMA


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

.. py:class:: GemmWMMA

   Bases: :py:obj:`tilelang.tileop.gemm.gemm_base.GemmBase`


   GEMM using AMD RDNA WMMA instructions (16×16×16, warp-size=32).


   .. py:method:: infer_layout(target, thread_nums)


   .. py:method:: lower(layout_map, target, thread_bounds, thread_var)


   .. py:method:: is_gemm_ss()


   .. py:method:: is_gemm_sr()


   .. py:method:: is_gemm_rs()


   .. py:method:: is_gemm_rr()