tilelang.backend.host_codegen¶

Host codegen registry shared by backend packages.

Attributes¶

HostCodegenFunc

HostCodegenHookFunc

TargetPredicate

Classes¶

HostCodegen

Host codegen entry point for one host target variant.

HostCodegenHook

Device-backend hook applied before host codegen build.

Functions¶

global_func_host_codegen(global_func_name)

Create a host codegen callback backed by a TVM global function.

register_host_codegen(target_host_kind, codegen, *[, ...])

Register a host codegen entry for a host target kind.

register_lazy_host_codegen(target_host_kind, import_path)

Register a backend module to import when a host target kind is first used.

register_host_codegen_hook(target_kind, hook, *[, ...])

Register a device-backend hook for host codegen preparation.

register_lazy_host_codegen_hooks(target_kind, import_path)

Register a backend module to import before applying host codegen hooks.

allowed_host_codegens_for_target(target_host)

Return matching host codegen names for a host target.

apply_host_codegen_hooks(mod, target_host, target)

Apply device-backend host codegen hooks.

resolve_host_codegen(target_host)

Resolve a host codegen entry from a TVM host target.

Module Contents¶

tilelang.backend.host_codegen.HostCodegenFunc¶
tilelang.backend.host_codegen.HostCodegenHookFunc¶
tilelang.backend.host_codegen.TargetPredicate¶
tilelang.backend.host_codegen.global_func_host_codegen(global_func_name)¶

Create a host codegen callback backed by a TVM global function.

Parameters:

global_func_name (str)

Return type:

HostCodegenFunc

class tilelang.backend.host_codegen.HostCodegen¶

Host codegen entry point for one host target variant.

name: str¶
build: HostCodegenFunc¶
supports_target: TargetPredicate | None = None¶
matches(target_host)¶
Parameters:

target_host (tvm.target.Target)

Return type:

bool

lower(mod, target_host)¶
Parameters:

  • mod (tvm.IRModule)

  • target_host (tvm.target.Target)

Return type:

tvm.IRModule

class tilelang.backend.host_codegen.HostCodegenHook¶

Device-backend hook applied before host codegen build.

name: str¶
apply: HostCodegenHookFunc¶
supports_target: TargetPredicate | None = None¶
matches(target)¶
Parameters:

target (tvm.target.Target)

Return type:

bool

lower(mod, target_host, target)¶
Parameters:

  • mod (tvm.IRModule)

  • target_host (tvm.target.Target)

  • target (tvm.target.Target)

Return type:

tvm.IRModule

tilelang.backend.host_codegen.register_host_codegen(target_host_kind, codegen, *, override=False)¶

Register a host codegen entry for a host target kind.

Parameters:
Return type:

HostCodegen

tilelang.backend.host_codegen.register_lazy_host_codegen(target_host_kind, import_path)¶

Register a backend module to import when a host target kind is first used.

Parameters:

  • target_host_kind (str)

  • import_path (str)

Return type:

None

tilelang.backend.host_codegen.register_host_codegen_hook(target_kind, hook, *, override=False)¶

Register a device-backend hook for host codegen preparation.

Parameters:
Return type:

HostCodegenHook

tilelang.backend.host_codegen.register_lazy_host_codegen_hooks(target_kind, import_path)¶

Register a backend module to import before applying host codegen hooks.

Parameters:

  • target_kind (str)

  • import_path (str)

Return type:

None

tilelang.backend.host_codegen.allowed_host_codegens_for_target(target_host)¶

Return matching host codegen names for a host target.

Parameters:

target_host (tvm.target.Target)

Return type:

list[str]

tilelang.backend.host_codegen.apply_host_codegen_hooks(mod, target_host, target)¶

Apply device-backend host codegen hooks.

Parameters:

  • mod (tvm.IRModule)

  • target_host (tvm.target.Target)

  • target (tvm.target.Target | None)

Return type:

tvm.IRModule

tilelang.backend.host_codegen.resolve_host_codegen(target_host)¶

Resolve a host codegen entry from a TVM host target.

Parameters:

target_host (tvm.target.Target)

Return type:

HostCodegen