tilelang.language.tir.ir
========================

.. py:module:: tilelang.language.tir.ir


Attributes
----------

.. autoapisummary::

   tilelang.language.tir.ir.abs
   tilelang.language.tir.ir.acos
   tilelang.language.tir.ir.acosh
   tilelang.language.tir.ir.address_of
   tilelang.language.tir.ir.asin
   tilelang.language.tir.ir.asinh
   tilelang.language.tir.ir.atan
   tilelang.language.tir.ir.atan2
   tilelang.language.tir.ir.atanh
   tilelang.language.tir.ir.bitwise_and
   tilelang.language.tir.ir.bitwise_not
   tilelang.language.tir.ir.bitwise_or
   tilelang.language.tir.ir.bitwise_xor
   tilelang.language.tir.ir.ceil
   tilelang.language.tir.ir.clz
   tilelang.language.tir.ir.copysign
   tilelang.language.tir.ir.cos
   tilelang.language.tir.ir.cosh
   tilelang.language.tir.ir.erf
   tilelang.language.tir.ir.exp
   tilelang.language.tir.ir.exp2
   tilelang.language.tir.ir.exp10
   tilelang.language.tir.ir.floor
   tilelang.language.tir.ir.ceildiv
   tilelang.language.tir.ir.floordiv
   tilelang.language.tir.ir.floormod
   tilelang.language.tir.ir.fmod
   tilelang.language.tir.ir.hypot
   tilelang.language.tir.ir.if_then_else
   tilelang.language.tir.ir.infinity
   tilelang.language.tir.ir.isfinite
   tilelang.language.tir.ir.isinf
   tilelang.language.tir.ir.isnan
   tilelang.language.tir.ir.isnullptr
   tilelang.language.tir.ir.ldexp
   tilelang.language.tir.ir.likely
   tilelang.language.tir.ir.log
   tilelang.language.tir.ir.log1p
   tilelang.language.tir.ir.log2
   tilelang.language.tir.ir.log10
   tilelang.language.tir.ir.lookup_param
   tilelang.language.tir.ir.max_value
   tilelang.language.tir.ir.min_value
   tilelang.language.tir.ir.nearbyint
   tilelang.language.tir.ir.nextafter
   tilelang.language.tir.ir.popcount
   tilelang.language.tir.ir.pow
   tilelang.language.tir.ir.q_multiply_shift
   tilelang.language.tir.ir.q_multiply_shift_per_axis
   tilelang.language.tir.ir.ret
   tilelang.language.tir.ir.round
   tilelang.language.tir.ir.rsqrt
   tilelang.language.tir.ir.shift_left
   tilelang.language.tir.ir.shift_right
   tilelang.language.tir.ir.sigmoid
   tilelang.language.tir.ir.sin
   tilelang.language.tir.ir.sinh
   tilelang.language.tir.ir.sqrt
   tilelang.language.tir.ir.tan
   tilelang.language.tir.ir.tanh
   tilelang.language.tir.ir.trunc
   tilelang.language.tir.ir.truncdiv
   tilelang.language.tir.ir.truncmod
   tilelang.language.tir.ir.tvm_access_ptr
   tilelang.language.tir.ir.tvm_throw_last_error
   tilelang.language.tir.ir.tvm_stack_alloca
   tilelang.language.tir.ir.tvm_stack_make_shape
   tilelang.language.tir.ir.tvm_stack_make_array
   tilelang.language.tir.ir.tvm_check_return
   tilelang.language.tir.ir.call_packed
   tilelang.language.tir.ir.call_cpacked
   tilelang.language.tir.ir.call_packed_lowered
   tilelang.language.tir.ir.call_cpacked_lowered
   tilelang.language.tir.ir.tvm_tuple
   tilelang.language.tir.ir.tvm_struct_set
   tilelang.language.tir.ir.tvm_struct_get
   tilelang.language.tir.ir.tvm_thread_invariant
   tilelang.language.tir.ir.tvm_thread_allreduce
   tilelang.language.tir.ir.tvm_load_matrix_sync
   tilelang.language.tir.ir.tvm_mma_sync
   tilelang.language.tir.ir.tvm_bmma_sync
   tilelang.language.tir.ir.tvm_fill_fragment
   tilelang.language.tir.ir.tvm_store_matrix_sync
   tilelang.language.tir.ir.tvm_storage_sync
   tilelang.language.tir.ir.tvm_warp_shuffle
   tilelang.language.tir.ir.tvm_warp_shuffle_up
   tilelang.language.tir.ir.tvm_warp_shuffle_down
   tilelang.language.tir.ir.tvm_warp_activemask
   tilelang.language.tir.ir.ptx_wait_group
   tilelang.language.tir.ir.ptx_commit_group
   tilelang.language.tir.ir.ptx_cp_async_barrier
   tilelang.language.tir.ir.ptx_init_barrier_thread_count
   tilelang.language.tir.ir.ptx_arrive_barrier
   tilelang.language.tir.ir.ptx_arrive_barrier_expect_tx
   tilelang.language.tir.ir.ptx_wait_barrier
   tilelang.language.tir.ir.create_barriers
   tilelang.language.tir.ir.assume
   tilelang.language.tir.ir.undef
   tilelang.language.tir.ir.TVMBackendAllocWorkspace
   tilelang.language.tir.ir.TVMBackendFreeWorkspace
   tilelang.language.tir.ir.start_profile_intrinsic
   tilelang.language.tir.ir.end_profile_intrinsic
   tilelang.language.tir.ir.anylist_getitem
   tilelang.language.tir.ir.anylist_resetitem
   tilelang.language.tir.ir.anylist_setitem_call_packed
   tilelang.language.tir.ir.anylist_setitem_call_cpacked
   tilelang.language.tir.ir.vscale
   tilelang.language.tir.ir.reinterpret
   tilelang.language.tir.ir.call_extern
   tilelang.language.tir.ir.call_intrin
   tilelang.language.tir.ir.call_llvm_intrin
   tilelang.language.tir.ir.call_llvm_pure_intrin
   tilelang.language.tir.ir.call_pure_extern
   tilelang.language.tir.ir.ptx_mma
   tilelang.language.tir.ir.ptx_mma_sp
   tilelang.language.tir.ir.ptx_ldmatrix
   tilelang.language.tir.ir.ptx_cp_async
   tilelang.language.tir.ir.ptx_cp_async_bulk
   tilelang.language.tir.ir.mma_store
   tilelang.language.tir.ir.mma_fill
   tilelang.language.tir.ir.vectorlow
   tilelang.language.tir.ir.vectorhigh
   tilelang.language.tir.ir.vectorcombine
   tilelang.language.tir.ir.tvm_mfma
   tilelang.language.tir.ir.tvm_mfma_store
   tilelang.language.tir.ir.tvm_rdna_wmma
   tilelang.language.tir.ir.tvm_rdna_wmma_store


Functions
---------

.. autoapisummary::

   tilelang.language.tir.ir.serial
   tilelang.language.tir.ir.parallel
   tilelang.language.tir.ir.vectorized
   tilelang.language.tir.ir.unroll
   tilelang.language.tir.ir.thread_binding
   tilelang.language.tir.ir.grid


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

.. py:function:: serial(start, stop = None, *, annotations = None)

   The serial For statement.

   :param start: The minimum value of iteration.
   :type start: PrimExpr
   :param stop: The maximum value of iteration.
   :type stop: PrimExpr
   :param annotations: The optional annotations of the For statement.
   :type annotations: Dict[str, Any]

   :returns: **res** -- The ForFrame.
   :rtype: frame.ForFrame


.. py:function:: parallel(start, stop = None, *, annotations = None)

   The parallel For statement.

   :param start: The minimum value of iteration.
   :type start: PrimExpr
   :param stop: The maximum value of iteration.
   :type stop: PrimExpr
   :param annotations: The optional annotations of the For statement.
   :type annotations: Dict[str, Any]

   :returns: **res** -- The ForFrame.
   :rtype: frame.ForFrame


.. py:function:: vectorized(start, stop = None, *, annotations = None)

   The vectorized For statement.

   :param start: The minimum value of iteration.
   :type start: PrimExpr
   :param stop: The maximum value of iteration.
   :type stop: PrimExpr
   :param annotations: The optional annotations of the For statement.
   :type annotations: Dict[str, Any]

   :returns: **res** -- The ForFrame.
   :rtype: frame.ForFrame


.. py:function:: unroll(start, stop = None, *, annotations = None)

   The unrolled For statement.

   :param start: The minimum value of iteration.
   :type start: PrimExpr
   :param stop: The maximum value of iteration.
   :type stop: PrimExpr
   :param annotations: The optional annotations of the For statement.
   :type annotations: Dict[str, Any]

   :returns: **res** -- The ForFrame.
   :rtype: frame.ForFrame


.. py:function:: thread_binding(start, stop = None, thread = None, *, annotations = None)

   The thread-binding For statement.

   :param start: The minimum value of iteration.
   :type start: PrimExpr
   :param stop: The maximum value of iteration.
   :type stop: PrimExpr
   :param thread: The thread for loop variable to bind.
   :type thread: str
   :param annotations: The optional annotations of the For statement.
   :type annotations: Dict[str, Any]

   :returns: **res** -- The ForFrame.
   :rtype: frame.ForFrame


.. py:function:: grid(*extents)

   The grid For statement.

   :param extents: The extents of the iteration.
   :type extents: PrimExpr

   :returns: **res** -- The ForFrame.
   :rtype: frame.ForFrame


.. py:data:: abs

.. py:data:: acos

.. py:data:: acosh

.. py:data:: address_of

.. py:data:: asin

.. py:data:: asinh

.. py:data:: atan

.. py:data:: atan2

.. py:data:: atanh

.. py:data:: bitwise_and

.. py:data:: bitwise_not

.. py:data:: bitwise_or

.. py:data:: bitwise_xor

.. py:data:: ceil

.. py:data:: clz

.. py:data:: copysign

.. py:data:: cos

.. py:data:: cosh

.. py:data:: erf

.. py:data:: exp

.. py:data:: exp2

.. py:data:: exp10

.. py:data:: floor

.. py:data:: ceildiv

.. py:data:: floordiv

.. py:data:: floormod

.. py:data:: fmod

.. py:data:: hypot

.. py:data:: if_then_else

.. py:data:: infinity

.. py:data:: isfinite

.. py:data:: isinf

.. py:data:: isnan

.. py:data:: isnullptr

.. py:data:: ldexp

.. py:data:: likely

.. py:data:: log

.. py:data:: log1p

.. py:data:: log2

.. py:data:: log10

.. py:data:: lookup_param

.. py:data:: max_value

.. py:data:: min_value

.. py:data:: nearbyint

.. py:data:: nextafter

.. py:data:: popcount

.. py:data:: pow

.. py:data:: q_multiply_shift

.. py:data:: q_multiply_shift_per_axis

.. py:data:: ret

.. py:data:: round

.. py:data:: rsqrt

.. py:data:: shift_left

.. py:data:: shift_right

.. py:data:: sigmoid

.. py:data:: sin

.. py:data:: sinh

.. py:data:: sqrt

.. py:data:: tan

.. py:data:: tanh

.. py:data:: trunc

.. py:data:: truncdiv

.. py:data:: truncmod

.. py:data:: tvm_access_ptr

.. py:data:: tvm_throw_last_error

.. py:data:: tvm_stack_alloca

.. py:data:: tvm_stack_make_shape

.. py:data:: tvm_stack_make_array

.. py:data:: tvm_check_return

.. py:data:: call_packed

.. py:data:: call_cpacked

.. py:data:: call_packed_lowered

.. py:data:: call_cpacked_lowered

.. py:data:: tvm_tuple

.. py:data:: tvm_struct_set

.. py:data:: tvm_struct_get

.. py:data:: tvm_thread_invariant

.. py:data:: tvm_thread_allreduce

.. py:data:: tvm_load_matrix_sync

.. py:data:: tvm_mma_sync

.. py:data:: tvm_bmma_sync

.. py:data:: tvm_fill_fragment

.. py:data:: tvm_store_matrix_sync

.. py:data:: tvm_storage_sync

.. py:data:: tvm_warp_shuffle

.. py:data:: tvm_warp_shuffle_up

.. py:data:: tvm_warp_shuffle_down

.. py:data:: tvm_warp_activemask

.. py:data:: ptx_wait_group

.. py:data:: ptx_commit_group

.. py:data:: ptx_cp_async_barrier

.. py:data:: ptx_init_barrier_thread_count

.. py:data:: ptx_arrive_barrier

.. py:data:: ptx_arrive_barrier_expect_tx

.. py:data:: ptx_wait_barrier

.. py:data:: create_barriers

.. py:data:: assume

.. py:data:: undef

.. py:data:: TVMBackendAllocWorkspace

.. py:data:: TVMBackendFreeWorkspace

.. py:data:: start_profile_intrinsic

.. py:data:: end_profile_intrinsic

.. py:data:: anylist_getitem

.. py:data:: anylist_resetitem

.. py:data:: anylist_setitem_call_packed

.. py:data:: anylist_setitem_call_cpacked

.. py:data:: vscale

.. py:data:: reinterpret

.. py:data:: call_extern

.. py:data:: call_intrin

.. py:data:: call_llvm_intrin

.. py:data:: call_llvm_pure_intrin

.. py:data:: call_pure_extern

.. py:data:: ptx_mma

.. py:data:: ptx_mma_sp

.. py:data:: ptx_ldmatrix

.. py:data:: ptx_cp_async

.. py:data:: ptx_cp_async_bulk

.. py:data:: mma_store

.. py:data:: mma_fill

.. py:data:: vectorlow

.. py:data:: vectorhigh

.. py:data:: vectorcombine

.. py:data:: tvm_mfma

.. py:data:: tvm_mfma_store

.. py:data:: tvm_rdna_wmma

.. py:data:: tvm_rdna_wmma_store