Update docs

Author: github-actions[bot]

_sources/autoapi/tilelang/contrib/nvcc/index.rst.txt

@@ -15,6 +15,9 @@ Functions
 .. autoapisummary::
 
    tilelang.contrib.nvcc.compile_cuda
+   tilelang.contrib.nvcc.default_compile_options
+   tilelang.contrib.nvcc.get_ptx_from_source
+   tilelang.contrib.nvcc.get_sass_from_source
    tilelang.contrib.nvcc.find_cuda_path
    tilelang.contrib.nvcc.get_cuda_version
    tilelang.contrib.nvcc.tilelang_callback_cuda_compile
@@ -56,6 +59,52 @@ Module Contents
    :rtype: bytearray
 
 
+.. py:function:: default_compile_options(compile_flags = None)
+
+   Build a set of default NVCC compile options for TileLang generated sources.
+
+   Includes C++ standard and common include paths (TileLang templates, CUTLASS,
+   CUDA include). Merges user-provided compile flags if given.
+
+   :param compile_flags: Additional flags to include. Items are split on whitespace.
+   :type compile_flags: Optional[List[str]]
+
+   :returns: A list of flags suitable for NVCC's command line.
+   :rtype: List[str]
+
+
+.. py:function:: get_ptx_from_source(code, compile_flags = None, verbose = False)
+
+   Compile CUDA C++ source to PTX using NVCC and return as text.
+
+   :param code: CUDA C++ kernel source code.
+   :type code: str
+   :param compile_flags: Additional flags merged with defaults.
+   :type compile_flags: Optional[List[str]]
+   :param verbose: Print NVCC output when True.
+   :type verbose: bool
+
+   :returns: PTX text.
+   :rtype: str
+
+
+.. py:function:: get_sass_from_source(code, compile_flags = None, verbose = False)
+
+   Compile CUDA C++ source to CUBIN and disassemble to SASS.
+
+   Uses nvdisasm if available; otherwise falls back to cuobjdump.
+
+   :param code: CUDA C++ kernel source code.
+   :type code: str
+   :param compile_flags: Additional flags merged with defaults.
+   :type compile_flags: Optional[List[str]]
+   :param verbose: Print tool outputs when True.
+   :type verbose: bool
+
+   :returns: SASS text.
+   :rtype: str
+
+
 .. py:function:: find_cuda_path()
 
    Utility function to find cuda path

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

@@ -191,6 +191,61 @@ Module Contents
    .. py:method:: run_once(func = None)
 
 
+   .. py:method:: show_source(which = 'kernel')
+
+      Print generated source code to stdout.
+
+      :param which: Select which source to print. Defaults to "kernel".
+      :type which: Literal["kernel", "host", "both"], optional
+
+      .. rubric:: Examples
+
+      >>> jit_kernel.show_source()            # print kernel source
+      >>> jit_kernel.show_source("host")      # print host source
+      >>> jit_kernel.show_source("both")      # print both sources
+
+
+
+   .. py:method:: export_sources(kernel_path = None, host_path = None)
+
+      Export generated source code to files.
+
+      :param kernel_path: Destination file path to write the kernel source. If None, skips writing kernel code.
+      :type kernel_path: Optional[str]
+      :param host_path: Destination file path to write the host source. If None, skips writing host code.
+      :type host_path: Optional[str]
+
+      .. rubric:: Examples
+
+      >>> jit_kernel.export_sources(kernel_path="/tmp/kernel.cu")
+      >>> jit_kernel.export_sources(host_path="/tmp/host.cc")
+      >>> jit_kernel.export_sources(
+      ...     kernel_path="/tmp/kernel.cu",
+      ...     host_path="/tmp/host.cc",
+      ... )
+
+
+
+   .. py:method:: print_source_code(which = 'kernel', file = None)
+
+      Deprecated: use show_source() or export_sources() instead.
+
+      :param which: Kept for backward compatibility with printing behavior.
+      :type which: Literal["kernel", "host", "both"], optional
+      :param file: If provided, behaves like export_sources(kernel_path=file).
+      :type file: Optional[str]
+
+      .. rubric:: Examples
+
+      >>> # New API (preferred)
+      >>> jit_kernel.show_source("both")
+      >>> jit_kernel.export_sources(kernel_path="/tmp/kernel.cu")
+
+      >>> # Old API (still works but deprecated)
+      >>> jit_kernel.print_source_code(file="/tmp/kernel.cu")
+
+
+
    .. py:method:: update_tuner_result(latency, config, ref_latency)
 
       Updates the tuning results for this kernel.
@@ -247,3 +302,49 @@ Module Contents
 
 
 
+   .. py:method:: show_ptx()
+
+      Print compiled PTX for the kernel (CUDA only).
+
+      .. rubric:: Examples
+
+      >>> jit_kernel.show_ptx()
+
+
+
+   .. py:method:: export_ptx(path)
+
+      Export compiled PTX to a file (CUDA only).
+
+      :param path: Destination file path to write PTX.
+      :type path: str
+
+      .. rubric:: Examples
+
+      >>> jit_kernel.export_ptx("/tmp/kernel.ptx")
+
+
+
+   .. py:method:: show_sass()
+
+      Print disassembled SASS for the kernel (CUDA only).
+
+      .. rubric:: Examples
+
+      >>> jit_kernel.show_sass()
+
+
+
+   .. py:method:: export_sass(path)
+
+      Export disassembled SASS to a file (CUDA only).
+
+      :param path: Destination file path to write SASS.
+      :type path: str
+
+      .. rubric:: Examples
+
+      >>> jit_kernel.export_sass("/tmp/kernel.sass")
+
+
+