tilelang.carver.utils module

tilelang.carver.utils.get_rasterization_code(pannel_width: int = 8) → str

tilelang.carver.utils.get_roller_hints_from_func(func_or_module: Union[PrimFunc, IRModule], arch: TileDevice, topk: int = 10, tensorcore_only: bool = False, allow_gemv: bool = False) → Optional[List[Hint]]

tilelang.carver.utils.get_roller_hints_from_output_nodes(output_nodes: List[OutputNode], arch: TileDevice, topk: int = 10, extra_tags: Optional[List[str]] = None) → Optional[List[Hint]]

tilelang.carver.utils.retrieve_func_from_module(ir_module: IRModule) → PrimFunc