tilelang.intrinsics.mma_layout
==============================

.. py:module:: tilelang.intrinsics.mma_layout


Attributes
----------

.. autoapisummary::

   tilelang.intrinsics.mma_layout.shared_16x16_to_mma_32x8_layout
   tilelang.intrinsics.mma_layout.shared_16x16_to_mma_32x8_layout_trans


Functions
---------

.. autoapisummary::

   tilelang.intrinsics.mma_layout.ldmatrix_32x8_to_shared_16x16_layout
   tilelang.intrinsics.mma_layout.ldmatrix_trans_32x8_to_shared_16x16_layout
   tilelang.intrinsics.mma_layout.ldmatrix_16x32_to_shared_16x32_layout_a
   tilelang.intrinsics.mma_layout.ldmatrix_16x32_to_shared_16x32_layout_b
   tilelang.intrinsics.mma_layout.ldmatrix_32x16_to_shared_16x32_layout_a
   tilelang.intrinsics.mma_layout.ldmatrix_32x16_to_shared_16x32_layout_b
   tilelang.intrinsics.mma_layout.mma_store_32x8_to_shared_16x16_layout
   tilelang.intrinsics.mma_layout.shared_16x16_to_mma_32x8_layout_sr
   tilelang.intrinsics.mma_layout.shared_16x16_to_mma_32x8_layout_rs
   tilelang.intrinsics.mma_layout.shared_16x32_to_mma_32x16_layout
   tilelang.intrinsics.mma_layout.shared_32x16_to_mma_32x16_layout
   tilelang.intrinsics.mma_layout.mma_32x8_to_shared_16x16_layout
   tilelang.intrinsics.mma_layout.shared_16x16_to_mma_32x8_smoothlayout
   tilelang.intrinsics.mma_layout.shared_16x32_to_mma_32x16_smoothlayout
   tilelang.intrinsics.mma_layout.shared_32x16_to_mma_32x16_smoothlayout
   tilelang.intrinsics.mma_layout.get_swizzle_layout
   tilelang.intrinsics.mma_layout.make_mma_swizzle_layout


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

.. py:function:: ldmatrix_32x8_to_shared_16x16_layout(thread_id, local_id)

.. py:function:: ldmatrix_trans_32x8_to_shared_16x16_layout(thread_id, local_id)

.. py:function:: ldmatrix_16x32_to_shared_16x32_layout_a(thread_id, local_id)

.. py:function:: ldmatrix_16x32_to_shared_16x32_layout_b(thread_id, local_id)

.. py:function:: ldmatrix_32x16_to_shared_16x32_layout_a(thread_id, local_id)

.. py:function:: ldmatrix_32x16_to_shared_16x32_layout_b(thread_id, local_id)

.. py:function:: mma_store_32x8_to_shared_16x16_layout(thread_id, local_id)

.. py:function:: shared_16x16_to_mma_32x8_layout_sr(i, j)

.. py:function:: shared_16x16_to_mma_32x8_layout_rs(i, j)

.. py:data:: shared_16x16_to_mma_32x8_layout

.. py:data:: shared_16x16_to_mma_32x8_layout_trans

.. py:function:: shared_16x32_to_mma_32x16_layout(i, j)

.. py:function:: shared_32x16_to_mma_32x16_layout(i, j)

.. py:function:: mma_32x8_to_shared_16x16_layout(thread_id, local_id)

.. py:function:: shared_16x16_to_mma_32x8_smoothlayout(i, j)

.. py:function:: shared_16x32_to_mma_32x16_smoothlayout(i, j)

.. py:function:: shared_32x16_to_mma_32x16_smoothlayout(i, j)

.. py:function:: get_swizzle_layout(row_idx, col_idx, row_size, dtype, swizzle_bytes=None)

.. py:function:: make_mma_swizzle_layout(shared_buf, is_smooth = False)