tilelang.carver.analysis module#
Analysis on TIR blocks, loops and functions.
- class tilelang.carver.analysis.BlockInfo(name: str, iters: List[IterInfo], block_rv: BlockRV, reduction_block: bool = False)#
Bases:
objectInformation about a TIR block.
- block_rv: BlockRV#
- dom() List[Union[int, PrimExpr]]#
The iteration domain of the block.
- dom_kind() str#
The iteration domain kind of the block, for example, SSSS, SSSR.
- is_elementwise(sch: Schedule) bool#
Whether the block is elementwise, i.e. trivial mapping between read/write region
- is_gemm() bool#
Whether the block is a GEMM workload.
- is_gemv() bool#
Whether the block is a GEMV workload.
- is_injective() bool#
Whether the block is injective, i.e. all its iteration domains are injective.
- is_reduction() bool#
Whether the block is a reduction workload.
- name: str#
- class tilelang.carver.analysis.IterInfo(kind: Literal['S', 'R', 'O'], var: Var, dom: PrimExpr, loop_rv: LoopRV)#
Bases:
objectInformation about a loop/iter var.
- property dom: Union[int, PrimExpr]#
The iteration domain of the loop.
- kind: Literal['S', 'R', 'O']#
- loop_rv: LoopRV#
- var: Var#
- tilelang.carver.analysis.check_func_with_dynamic(func)#
- tilelang.carver.analysis.collect_block_iter_vars_used_in_access_region(block: Block, region: List[Range]) Set[Var]#
Collect the block iter variables used in the access region of a buffer region.
- tilelang.carver.analysis.collect_vars_used_in_prim_expr(expr: PrimExpr) Set[Var]#
Collect the variables used in the PrimExpr.
- tilelang.carver.analysis.detect_dominant_read(block: Block) PrimExpr#
Detect the dominant read indices in the block.
- tilelang.carver.analysis.find_var_from_func(func, var: str)#
- tilelang.carver.analysis.get_coalesced_veclen(block_stmt: Block, target_bits: int = 128) int#
- tilelang.carver.analysis.get_max_threads_per_block(target: Target) int#
- tilelang.carver.analysis.get_reduction_blocks(sch: Schedule, blocks: List[BlockRV]) List[BlockRV]#
- tilelang.carver.analysis.get_root_block(sch: Schedule, func_name: str = 'main') BlockRV#
- tilelang.carver.analysis.is_broadcast_epilogue(sch: Schedule, block: BlockRV, epilogue: BlockRV) bool#
Check if the epilogue block is a broadcast pattern