tilelang.backend.device_codegen

Device codegen registry shared by backend packages.

Attributes

DeviceCodegenFunc
TargetPredicate

Classes

DeviceCodegen

Device codegen entry points for one backend target variant.

Functions

global_func_device_codegen(global_func_name)	Create a device codegen callback backed by a TVM global function.
register_device_codegen(target_kind, codegen, *[, ...])	Register a device codegen entry for a target kind.
register_lazy_device_codegen(target_kind, import_path)	Register a backend module to import when its target kind is first used.
allowed_device_codegens_for_target(target)	Return matching device codegen names for a target.
resolve_device_codegen(target)	Resolve a device codegen entry from a TVM target.

Module Contents

tilelang.backend.device_codegen.DeviceCodegenFunc

tilelang.backend.device_codegen.TargetPredicate

tilelang.backend.device_codegen.global_func_device_codegen(global_func_name)

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

Parameters:

global_func_name (str)

Return type:

DeviceCodegenFunc

class tilelang.backend.device_codegen.DeviceCodegen

Device codegen entry points for one backend target variant.

name: str

build: DeviceCodegenFunc | None = None

build_without_compile: DeviceCodegenFunc | None = None

supports_target: TargetPredicate | None = None

matches(target)

Parameters:

target (tvm.target.Target)

Return type:

bool

lower(mod, target, *, compile_device)

Parameters:

• mod (tvm.IRModule)

• target (tvm.target.Target)

• compile_device (bool)

Return type:

tvm.IRModule

tilelang.backend.device_codegen.register_device_codegen(target_kind, codegen, *, override=False)

Register a device codegen entry for a target kind.

Parameters:

• target_kind (str)

• codegen (DeviceCodegen)

• override (bool)

Return type:

DeviceCodegen

tilelang.backend.device_codegen.register_lazy_device_codegen(target_kind, import_path)

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

Parameters:

• target_kind (str)

• import_path (str)

Return type:

None

tilelang.backend.device_codegen.allowed_device_codegens_for_target(target)

Return matching device codegen names for a target.

Parameters:

target (tvm.target.Target)

Return type:

list[str]

tilelang.backend.device_codegen.resolve_device_codegen(target)

Resolve a device codegen entry from a TVM target.

Parameters:

target (tvm.target.Target)

Return type:

DeviceCodegen