Update docs

Author: github-actions[bot]

_sources/autoapi/tilelang/autotuner/tuner/index.rst.txt

@@ -34,6 +34,7 @@ Classes
 .. autoapisummary::
 
    tilelang.autotuner.tuner.AutoTuner
+   tilelang.autotuner.tuner.AutoTuneImpl
 
 
 Functions
@@ -201,6 +202,120 @@ Module Contents
 
 
 
+.. py:class:: AutoTuneImpl
+
+   Bases: :py:obj:`Generic`\ [\ :py:obj:`_P`\ , :py:obj:`_T`\ ]
+
+
+   Abstract base class for generic types.
+
+   A generic type is typically declared by inheriting from
+   this class parameterized with one or more type variables.
+   For example, a generic mapping type might be defined as::
+
+     class Mapping(Generic[KT, VT]):
+         def __getitem__(self, key: KT) -> VT:
+             ...
+         # Etc.
+
+   This class can then be used as follows::
+
+     def lookup_name(mapping: Mapping[KT, VT], key: KT, default: VT) -> VT:
+         try:
+             return mapping[key]
+         except KeyError:
+             return default
+
+
+   .. py:attribute:: jit_impl
+      :type:  tilelang.jit.JITImpl
+
+
+   .. py:attribute:: warmup
+      :type:  int
+      :value: 25
+
+
+
+   .. py:attribute:: rep
+      :type:  int
+      :value: 100
+
+
+
+   .. py:attribute:: timeout
+      :type:  int
+      :value: 100
+
+
+
+   .. py:attribute:: configs
+      :type:  dict | Callable
+      :value: None
+
+
+
+   .. py:attribute:: supply_type
+      :type:  tilelang.TensorSupplyType
+
+
+   .. py:attribute:: ref_prog
+      :type:  Callable
+      :value: None
+
+
+
+   .. py:attribute:: supply_prog
+      :type:  Callable
+      :value: None
+
+
+
+   .. py:attribute:: rtol
+      :type:  float
+      :value: 0.01
+
+
+
+   .. py:attribute:: atol
+      :type:  float
+      :value: 0.01
+
+
+
+   .. py:attribute:: max_mismatched_ratio
+      :type:  float
+      :value: 0.01
+
+
+
+   .. py:attribute:: skip_check
+      :type:  bool
+      :value: False
+
+
+
+   .. py:attribute:: manual_check_prog
+      :type:  Callable
+      :value: None
+
+
+
+   .. py:attribute:: cache_input_tensors
+      :type:  bool
+      :value: False
+
+
+
+   .. py:method:: __post_init__()
+
+
+   .. py:method:: get_tunner()
+
+
+   .. py:method:: __call__(*args, **kwargs)
+
+
 .. py:function:: autotune(func = None, *, configs, warmup = 25, rep = 100, timeout = 100, supply_type = tilelang.TensorSupplyType.Auto, ref_prog = None, supply_prog = None, rtol = 0.01, atol = 0.01, max_mismatched_ratio = 0.01, skip_check = False, manual_check_prog = None, cache_input_tensors = False)
 
    Just-In-Time (JIT) compiler decorator for TileLang functions.

_sources/autoapi/tilelang/jit/index.rst.txt

@@ -31,12 +31,21 @@ Attributes
    tilelang.jit.logger
 
 
+Classes
+-------
+
+.. autoapisummary::
+
+   tilelang.jit.JITImpl
+
+
 Functions
 ---------
 
 .. autoapisummary::
 
    tilelang.jit.compile
+   tilelang.jit.par_compile
    tilelang.jit.jit
 
 
@@ -65,7 +74,112 @@ Package Contents
    :type pass_configs: dict, optional
 
 
-.. py:function:: jit(func = None, *, out_idx = None, target = 'auto', target_host = None, execution_backend = 'cython', verbose = False, pass_configs = None, debug_root_path = None, compile_flags = None)
+.. py:function:: par_compile(funcs, out_idx = None, execution_backend = 'cython', target = 'auto', target_host = None, verbose = False, pass_configs = None, compile_flags = None, num_workers = None, ignore_error = False)
+
+   Parallel compile multiple TileLang PrimFunc with TVM and build JITKernels.
+   :param funcs: The TileLang TIR functions to compile and wrap.
+   :type funcs: Iterable[tvm.tir.PrimFunc]
+   :param out_idx: Index(es) of the output tensors to return (default: None).
+   :type out_idx: Union[List[int], int], optional
+   :param execution_backend: Execution backend to use for kernel execution (default: "cython").
+   :type execution_backend: Literal["dlpack", "ctypes", "cython", "nvrtc"], optional
+   :param target: Compilation target, either as a string or a TVM Target object (default: "auto").
+   :type target: Union[str, Target], optional
+   :param target_host: Target host for cross-compilation (default: None).
+   :type target_host: Union[str, Target], optional
+   :param verbose: Whether to enable verbose output (default: False).
+   :type verbose: bool, optional
+   :param pass_configs: Additional keyword arguments to pass to the Compiler PassContext.
+                        Refer to `tilelang.transform.PassConfigKey` for supported options.
+   :type pass_configs: dict, optional
+
+
+.. py:class:: JITImpl
+
+   Bases: :py:obj:`Generic`\ [\ :py:obj:`_P`\ , :py:obj:`_KP`\ , :py:obj:`_T`\ ]
+
+
+   Abstract base class for generic types.
+
+   A generic type is typically declared by inheriting from
+   this class parameterized with one or more type variables.
+   For example, a generic mapping type might be defined as::
+
+     class Mapping(Generic[KT, VT]):
+         def __getitem__(self, key: KT) -> VT:
+             ...
+         # Etc.
+
+   This class can then be used as follows::
+
+     def lookup_name(mapping: Mapping[KT, VT], key: KT, default: VT) -> VT:
+         try:
+             return mapping[key]
+         except KeyError:
+             return default
+
+
+   .. py:attribute:: func
+      :type:  Callable[_P, _T] | tilelang.language.v2.PrimFunc[_KP, _T]
+
+
+   .. py:attribute:: out_idx
+      :type:  list[int] | int | None
+
+
+   .. py:attribute:: execution_backend
+      :type:  Literal['dlpack', 'ctypes', 'cython']
+
+
+   .. py:attribute:: target
+      :type:  str | tvm.target.Target
+
+
+   .. py:attribute:: target_host
+      :type:  str | tvm.target.Target
+
+
+   .. py:attribute:: verbose
+      :type:  bool
+
+
+   .. py:attribute:: pass_configs
+      :type:  dict[str, Any] | None
+
+
+   .. py:attribute:: debug_root_path
+      :type:  str | None
+
+
+   .. py:attribute:: compile_flags
+      :type:  list[str] | str | None
+
+
+   .. py:attribute:: func_source
+      :type:  str
+
+
+   .. py:attribute:: signature
+      :type:  inspect.Signature
+
+
+   .. py:method:: __post_init__()
+
+
+   .. py:method:: get_tir(*args, **kwargs)
+
+
+   .. py:method:: par_compile(configs, num_workers = None, ignore_error = False)
+
+
+   .. py:method:: compile(*args, **kwargs)
+
+
+   .. py:method:: __call__(*args, **kwargs)
+
+
+.. py:function:: jit(func: Callable[_P, tilelang.language.v2.PrimFunc[_KP, _T]]) -> JITImpl[_P, _KP, _T]
+                 jit(*, out_idx: Any = None, target: str | tvm.target.Target = 'auto', target_host: str | tvm.target.Target = None, execution_backend: Literal['dlpack', 'ctypes', 'cython', 'nvrtc'] = 'cython', verbose: bool = False, pass_configs: dict[str, Any] | None = None, debug_root_path: str | None = None, compile_flags: list[str] | str | None = None) -> Callable[[Callable[_P, tilelang.language.v2.PrimFunc[_KP, _T]]], JITImpl[_P, _KP, _T]]
 
    Just-In-Time (JIT) compiler decorator for TileLang functions.
 

_sources/autoapi/tilelang/jit/kernel/index.rst.txt

@@ -27,6 +27,9 @@ Module Contents
 
 .. py:class:: JITKernel(func = None, out_idx = None, execution_backend = 'cython', target = 'auto', target_host = None, verbose = False, pass_configs = None, from_database = False, compile_flags = None)
 
+   Bases: :py:obj:`Generic`\ [\ :py:obj:`_P`\ , :py:obj:`_T`\ ]
+
+
    A wrapper class for compiling and invoking TileLang (TVM TIR) functions as PyTorch-compatible functions.
 
    .. attribute:: artifact

_sources/autoapi/tilelang/language/index.rst.txt

@@ -39,6 +39,7 @@ Submodules
    /autoapi/tilelang/language/symbolics/index
    /autoapi/tilelang/language/tir/index
    /autoapi/tilelang/language/utils/index
+   /autoapi/tilelang/language/v2/index
    /autoapi/tilelang/language/warpgroup/index