tilelang.jit.adapter.nvrtc.adapter module

class tilelang.jit.adapter.nvrtc.adapter.NVRTCKernelAdapter(params: List[KernelParam], result_idx: List[int], target: Union[str, Target], func_or_mod: Union[PrimFunc, IRModule], host_mod: Optional[IRModule] = None, device_mod: Optional[IRModule] = None, kernel_global_source: Optional[str] = None, verbose: bool = False, pass_configs: Optional[Dict[str, Any]] = None)

Bases: BaseKernelAdapter

classmethod from_database(params: List[KernelParam], result_idx: List[int], target: str, func_or_mod: Union[PrimFunc, IRModule], kernel_global_source: str, kernel_lib_path: str, verbose: bool = False, pass_configs: Optional[Dict[str, Any]] = None)

get_kernel_source()

kernels = {}

property prim_func: PrimFunc

Returns the primary TIR function from the IR module.

pymodule = None