tilelang.analysis.fragment_loop_checker

Functions

collect_local_buffer_accesses(statement)	Collect local buffer accesses in the loop body.
FragmentLoopChecker()	When using T.Parallel over a local/fragment buffer, there are several restrictions:

Module Contents

tilelang.analysis.fragment_loop_checker.collect_local_buffer_accesses(statement)

Collect local buffer accesses in the loop body.

参数:

statement -- The TIR statement to analyze

返回:

Tuple of buffer accesses in the loop body.

返回类型:

list[tvm.tir.BufferLoad | tvm.tir.BufferStore]

tilelang.analysis.fragment_loop_checker.FragmentLoopChecker()

When using T.Parallel over a local/fragment buffer, there are several restrictions: to ensure that the parallelization is valid.

1. The range of loop can not be symbolic.

返回:

A prim_func_pass that applies the transformation