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)¶
Parameters:

pannel_width (int)

Return type:

str

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

  • func_or_module (Union[tvm.tir.PrimFunc, tvm.IRModule])

  • arch (tilelang.carver.arch.TileDevice)

  • topk (int)

  • tensorcore_only (bool)

  • allow_gemv (bool)

Return type:

Optional[List[tilelang.carver.roller.hint.Hint]]

tilelang.carver.utils.get_roller_hints_from_output_nodes(output_nodes, arch, topk=10, extra_tags=None)¶
Parameters:
Return type:

Optional[List[tilelang.carver.roller.hint.Hint]]

tilelang.carver.utils.retrieve_func_from_module(ir_module)¶
Parameters:

ir_module (tvm.IRModule)

Return type:

tvm.tir.PrimFunc