tilelang.analysis.fragment_loop_checker

函式

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