tilelang.engine.param
=====================

.. py:module:: tilelang.engine.param

.. autoapi-nested-parse::

   The profiler and convert to torch utils



Classes
-------

.. autoapisummary::

   tilelang.engine.param.KernelParam
   tilelang.engine.param.CompiledArtifact


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

.. py:class:: KernelParam

   Represents parameters for a kernel operation, storing dtype and shape information.
   Used to describe tensor or scalar parameters in TVM/PyTorch interop.


   .. py:attribute:: dtype
      :type:  torch.dtype


   .. py:attribute:: shape
      :type:  List[Union[int, tvm.tir.Var]]


   .. py:method:: from_buffer(buffer)
      :classmethod:


      Creates a KernelParam instance from a TVM Buffer object.

      :param buffer: TVM Buffer object containing dtype and shape information

      :returns: KernelParam instance with converted dtype and shape

      :raises ValueError: If dimension type is not supported (not IntImm or Var)



   .. py:method:: from_var(var)
      :classmethod:


      Creates a KernelParam instance from a TVM Variable object.
      Used for scalar parameters.

      :param var: TVM Variable object containing dtype information

      :returns: KernelParam instance representing a scalar (empty shape)



   .. py:method:: is_scalar()

      Checks if the parameter represents a scalar value.

      :returns: True if parameter has no dimensions (empty shape), False otherwise
      :rtype: bool



   .. py:method:: is_unsigned()

      Checks if the parameter represents an unsigned integer type.

      :returns: True if parameter is an unsigned integer type, False otherwise
      :rtype: bool



   .. py:method:: is_float8()

      Checks if the parameter represents a float8 type.

      :returns: True if parameter is a float8 type, False otherwise
      :rtype: bool



   .. py:method:: is_boolean()

      Checks if the parameter represents a boolean type.

      :returns: True if parameter is a boolean type, False otherwise
      :rtype: bool



.. py:class:: CompiledArtifact

   Represents a compiled kernel artifact containing both host and device code.
   Stores all necessary components for kernel execution in the TVM runtime.


   .. py:attribute:: host_mod
      :type:  tilelang.tvm.IRModule


   .. py:attribute:: device_mod
      :type:  tilelang.tvm.IRModule


   .. py:attribute:: params
      :type:  List[KernelParam]


   .. py:attribute:: kernel_source
      :type:  str


   .. py:attribute:: rt_mod
      :type:  Optional[tilelang.tvm.runtime.Module]
      :value: None