tilelang.transform.add_bufstore_wrapper¶
Functions¶
Creates a TVM pass that wraps single buffer stores with parallel loops. |
Module Contents¶
- tilelang.transform.add_bufstore_wrapper.AddWrapperForSingleBufStore()¶
Creates a TVM pass that wraps single buffer stores with parallel loops.
This transformation adds T.Parallel wrappers around buffer stores that: 1. Access fragment buffers with index 0 2. Are not inside existing tile operations or thread bindings 3. Don’t access fragment buffers with non-zero indices
- Returns:
A prim_func_pass that applies the transformation