tilelang.jit.adapter.cutedsl.adapter¶

Attributes¶

logger

Classes¶

CuTeDSLKernelAdapter

Helper class that provides a standard way to create an ABC using

Module Contents¶

tilelang.jit.adapter.cutedsl.adapter.logger¶
class tilelang.jit.adapter.cutedsl.adapter.CuTeDSLKernelAdapter(params, result_idx, target, func_or_mod, host_mod=None, device_mod=None, host_kernel_source=None, device_kernel_source=None, verbose=False, pass_configs=None, compile_flags=None)¶

Bases: tilelang.jit.adapter.base.BaseKernelAdapter

Helper class that provides a standard way to create an ABC using inheritance.

Parameters:

  • params (list[tilelang.engine.param.KernelParam])

  • result_idx (list[int])

  • target (str | tvm.target.Target)

  • func_or_mod (tvm.tir.PrimFunc | tilelang.tvm.IRModule)

  • host_mod (tilelang.tvm.IRModule | None)

  • device_mod (tilelang.tvm.IRModule | None)

  • host_kernel_source (str | None)

  • device_kernel_source (str | None)

  • verbose (bool)

  • pass_configs (dict[str, Any] | None)

  • compile_flags (list[str] | None)

pymodule = None¶
params¶
result_idx¶
host_kernel_source = None¶
device_kernel_source = None¶
param_dtypes¶
param_shapes = []¶
target¶
verbose = False¶
wrapper¶
host_func¶
function_names¶
tma_cpp_init_code¶
tma_lib_name¶
launcher_cpp_code¶
launcher_lib_name¶
lib_generator¶
libpath¶
classmethod from_database(params, result_idx, target, func_or_mod, host_kernel_source, device_kernel_source, kernel_lib_path, verbose=False, pass_configs=None, compile_flags=None)¶
Parameters:

  • params (list[tilelang.engine.param.KernelParam])

  • result_idx (list[int])

  • target (str)

  • func_or_mod (tvm.tir.PrimFunc | tilelang.tvm.IRModule)

  • host_kernel_source (str)

  • device_kernel_source (str)

  • kernel_lib_path (str)

  • verbose (bool)

  • pass_configs (dict[str, Any] | None)

  • compile_flags (list[str] | None)

get_kernel_source(kernel_only=True)¶

Get the CUDA kernel source code.

Returns:

The kernel source code, or None if not available

Return type:

str | None

Parameters:

kernel_only (bool)

property prim_func: tvm.tir.PrimFunc¶

Returns the primary TIR function from the IR module.

Return type:

tvm.tir.PrimFunc