tilelang.jit.adapter.nvrtc.adapter¶

Attributes¶

logger

is_nvrtc_available

NVRTC_UNAVAILABLE_WARNING

is_nvrtc_available

Classes¶

NVRTCKernelAdapter

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

Module Contents¶

tilelang.jit.adapter.nvrtc.adapter.logger¶
tilelang.jit.adapter.nvrtc.adapter.is_nvrtc_available = False¶
tilelang.jit.adapter.nvrtc.adapter.NVRTC_UNAVAILABLE_WARNING = 'cuda-python is not available, nvrtc backend cannot be used. Please install cuda-python via `pip...¶
tilelang.jit.adapter.nvrtc.adapter.is_nvrtc_available = True¶
class tilelang.jit.adapter.nvrtc.adapter.NVRTCKernelAdapter(params, result_idx, target, func_or_mod, host_mod=None, device_mod=None, kernel_global_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 (Union[str, tvm.target.Target])

  • func_or_mod (Union[tvm.tir.PrimFunc, tilelang.tvm.IRModule])

  • host_mod (Optional[tilelang.tvm.IRModule])

  • device_mod (Optional[tilelang.tvm.IRModule])

  • kernel_global_source (Optional[str])

  • verbose (bool)

  • pass_configs (Optional[Dict[str, Any]])

  • compile_flags (Optional[List[str]])

pymodule = None¶
kernels¶
params¶
result_idx¶
kernel_global_source = None¶
param_dtypes¶
param_shapes = []¶
dynamic_symbolic_map¶
target¶
verbose = False¶
wrapper¶
lib_generator¶
libpath¶
classmethod from_database(params, result_idx, target, func_or_mod, kernel_global_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 (Union[tvm.tir.PrimFunc, tilelang.tvm.IRModule])

  • kernel_global_source (str)

  • kernel_lib_path (str)

  • verbose (bool)

  • pass_configs (Optional[Dict[str, Any]])

  • compile_flags (Optional[List[str]])

get_kernel_source()¶
property prim_func: tvm.tir.PrimFunc¶

Returns the primary TIR function from the IR module.

Return type:

tvm.tir.PrimFunc