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

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


Submodules
----------

.. toctree::
   :maxdepth: 1

   /autoapi/tilelang/tileop/gemm/gemm_base/index
   /autoapi/tilelang/tileop/gemm/gemm_mma/index


Classes
-------

.. autoapisummary::

   tilelang.tileop.gemm.GemmPy


Functions
---------

.. autoapisummary::

   tilelang.tileop.gemm.gemm_py_infer_layout
   tilelang.tileop.gemm.gemm_py_lower


Package Contents
----------------

.. py:function:: gemm_py_infer_layout(gemm_py, target, thread_bounds)

.. py:function:: gemm_py_lower(gemm_py, target, thread_bounds, thread_var)

.. py:class:: GemmPy

   Bases: :py:obj:`tvm.ir.base.Node`, :py:obj:`tvm.runtime.Scriptable`


   .. py:attribute:: A
      :type:  tvm.tir.Buffer


   .. py:attribute:: B
      :type:  tvm.tir.Buffer


   .. py:attribute:: C
      :type:  tvm.tir.Buffer


   .. py:attribute:: APtr
      :type:  tvm.tir.PrimExpr


   .. py:attribute:: BPtr
      :type:  tvm.tir.PrimExpr


   .. py:attribute:: CPtr
      :type:  tvm.tir.PrimExpr


   .. py:attribute:: M
      :type:  int


   .. py:attribute:: N
      :type:  int


   .. py:attribute:: K
      :type:  int


   .. py:attribute:: trans_A
      :type:  bool


   .. py:attribute:: trans_B
      :type:  bool


   .. py:attribute:: stride_A
      :type:  int


   .. py:attribute:: stride_B
      :type:  int


   .. py:attribute:: offset_A
      :type:  int


   .. py:attribute:: offset_B
      :type:  int


   .. py:attribute:: clear_accum
      :type:  bool


   .. py:attribute:: k_pack
      :type:  int


   .. py:attribute:: wg_wait
      :type:  int


   .. py:attribute:: policy
      :type:  tilelang.ir.GemmWarpPolicy


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


   .. py:method:: lower(target, thread_nums, thread_var)