tilelang.carver.utils

Attributes

logger

Functions

get_rasterization_code([pannel_width])

get_roller_hints_from_func(func_or_module, arch[, ...])

get_roller_hints_from_output_nodes(output_nodes, arch)

retrieve_func_from_module(ir_module)

Module Contents

tilelang.carver.utils.logger
tilelang.carver.utils.get_rasterization_code(pannel_width=8)
参数:

pannel_width (int)

返回类型:

str

tilelang.carver.utils.get_roller_hints_from_func(func_or_module, arch, topk=10, tensorcore_only=False, allow_gemv=False)
参数:

  • func_or_module (tvm.tir.PrimFunc | tvm.IRModule)

  • arch (tilelang.carver.arch.TileDevice)

  • topk (int)

  • tensorcore_only (bool)

  • allow_gemv (bool)

返回类型:

list[tilelang.carver.roller.hint.Hint] | None

tilelang.carver.utils.get_roller_hints_from_output_nodes(output_nodes, arch, topk=10, extra_tags=None)
参数:
返回类型:

list[tilelang.carver.roller.hint.Hint] | None

tilelang.carver.utils.retrieve_func_from_module(ir_module)
参数:

ir_module (tvm.IRModule)

返回类型:

tvm.tir.PrimFunc