tilelang.carver.common_schedules module#

Common schedule strategies for TIR.

tilelang.carver.common_schedules.get_block(sch: Schedule, blocks: List[BlockInfo], name: str)#

Get the target block from a schedule.

Parameters:

  • sch (tir.Schedule) – The TIR schedule used to get target block.

  • name (str) – The name of the target block.

Returns:

target_block – The target block.

Return type:

BlockRV

tilelang.carver.common_schedules.get_output_blocks(sch: Schedule, blocks: List[BlockInfo])#

Get the output blocks of a schedule.

Parameters:

  • sch (tir.Schedule) – The TIR schedule used to get output blocks.

  • blocks (List[BlockInfo]) – The blocks to be analyzed.

Returns:

output_blocks – The output blocks.

Return type:

List[BlockInfo]

tilelang.carver.common_schedules.try_inline(sch: Schedule, blocks: List[BlockInfo]) → List[BlockInfo]#

Try to inline as many blocks as possible, and return the remaining blocks.

Parameters:

  • sch (tir.Schedule) – The TIR schedule used to inline blocks.

  • blocks (List[BlockInfo]) – The blocks to be inlined.

Returns:

remaining – The remaining blocks that cannot be inlined.

Return type:

List[BlockInfo]

tilelang.carver.common_schedules.try_inline_contiguous_spatial(sch: Schedule, block_infos: List[BlockInfo]) → List[BlockInfo]#

Try to inline contiguous spatial blocks in a schedule

Parameters:

  • sch (tir.Schedule) – The TIR schedule used to inline blocks.

  • block_infos (List[BlockInfo]) – The blocks to be try.

Returns:

remaining – The remaining blocks that cannot be inlined.

Return type:

List[BlockInfo]