tilelang.analysis.fragment_loop_checker¶
Functions¶
|
Collect local buffer accesses in the loop body. |
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.
- Parameters:
statement – The TIR statement to analyze
- Returns:
Tuple of buffer accesses in the loop body.
- Return type:
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.
The range of loop can not be symbolic.
- Returns:
A prim_func_pass that applies the transformation