tilelang.backend.device_codegen¶

Device codegen component types and helpers.

Attributes¶

DeviceCodegenFunc

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.

Module Contents¶

tilelang.backend.device_codegen.DeviceCodegenFunc¶
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¶
lower(mod, target, *, compile_device)¶
Parameters:

  • mod (tvm.IRModule)

  • target (tvm.target.Target)

  • compile_device (bool)

Return type:

tvm.IRModule