tilelang.carver.utils

Attributes

logger

函式

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