tilelang.metal.codegen

Attributes

build_metal
build_metal_without_compile

Functions

mark_host_metal_context(mod, target_host, target)

Module Contents

tilelang.metal.codegen.build_metal

tilelang.metal.codegen.build_metal_without_compile

tilelang.metal.codegen.mark_host_metal_context(mod, target_host, target)

Parameters:

• mod (tvm.IRModule)

• target_host (tvm.target.Target)

• target (tvm.target.Target)

Return type:

tvm.IRModule