tilelang.engine.lower module

The compiler for TL programs.

tilelang.engine.lower.canon_target_host(target: Union[str, Target], target_host: Optional[Union[str, Target]])

tilelang.engine.lower.device_codegen(device_mod: IRModule, target: Target) → IRModule

tilelang.engine.lower.device_codegen_without_compile(device_mod: IRModule, target: Target) → IRModule

tilelang.engine.lower.extrac_params(func: PrimFunc) → List[KernelParam]

tilelang.engine.lower.get_device_call(is_device_c: bool = False) → Callable[[PrimFunc], bool]

tilelang.engine.lower.get_host_call(is_device_c: bool = False) → Callable[[PrimFunc], bool]

tilelang.engine.lower.has_device_kernel_launch(attrs) → bool

Check if the attributes indicate a device kernel launch.

tilelang.engine.lower.host_codegen(host_mod: IRModule, target_host: Target) → IRModule

tilelang.engine.lower.is_cpu_device_backend(target: Target)

tilelang.engine.lower.is_device_call(func: PrimFunc)

tilelang.engine.lower.is_device_call_c_device(func: PrimFunc)

tilelang.engine.lower.lower(func_or_mod: Union[PrimFunc, IRModule], target: Union[str, Target] = 'auto', target_host: Optional[Union[str, Target]] = None, runtime_only=False, enable_host_codegen=False, enable_device_compile=False) → CompiledArtifact

enable_host_codegen: whether to enable host codegen, default is False, as we have our own host codegen implementation in jit. enable_device_compile: whether to enable device codegen, default is False, as we have our own device codegen implementation in jit.