tilelang.backend.host_codegen

Host codegen component types and shared implementations.

Attributes

HostCodegenFunc
HostCodegenHookFunc
STANDARD_HOST_CODEGENS

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.

Module Contents

tilelang.backend.host_codegen.HostCodegenFunc

tilelang.backend.host_codegen.HostCodegenHookFunc

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

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

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.STANDARD_HOST_CODEGENS: collections.abc.Mapping[str, HostCodegen]