tilelang.carver.common_schedules

Common schedule strategies for TIR.

Functions

get_block(sch, blocks, name)	Get the target block from a schedule.
get_output_blocks(sch, blocks)	Get the output blocks of a schedule.
try_inline(sch, blocks)	Try to inline as many blocks as possible, and return the remaining blocks.
try_inline_contiguous_spatial(sch, block_infos)	Try to inline contiguous spatial blocks in a schedule

Module Contents

tilelang.carver.common_schedules.get_block(sch, blocks, name)

Get the target block from a schedule.

参数:

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

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

• blocks (list[tilelang.carver.analysis.BlockInfo])

返回:

target_block -- The target block.

返回类型:

BlockRV

tilelang.carver.common_schedules.get_output_blocks(sch, blocks)

Get the output blocks of a schedule.

参数:

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

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

返回:

output_blocks -- The output blocks.

返回类型:

List[BlockInfo]

tilelang.carver.common_schedules.try_inline(sch, blocks)

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

参数:

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

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

返回:

remaining -- The remaining blocks that cannot be inlined.

返回类型:

List[BlockInfo]

tilelang.carver.common_schedules.try_inline_contiguous_spatial(sch, block_infos)

Try to inline contiguous spatial blocks in a schedule

参数:

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

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

返回:

remaining -- The remaining blocks that cannot be inlined.

返回类型:

List[BlockInfo]