[Mlir-commits] [mlir] [mlir][tensor][NFC] Improve documentation of pack/unpack (PR #72201)
lorenzo chelini
llvmlistbot at llvm.org
Tue Nov 14 00:15:56 PST 2023
================
@@ -1751,42 +1751,66 @@ def Tensor_PackOp : Tensor_RelayoutOp<"pack", [
AttrSizedOperandSegments]> {
let summary = "tensor pack operation";
let description = [{
- The pack operation converts an input tensor to a higher-dimensional tensor
- with a tiled and packed layout. The mandatory `inner_dims_pos` attribute
- specifies a permutation for the original dimensions, while `inner_tiles` is the
- tiling factor for each dimension. The optional attribute `outer_dims_perm`
- specifies the order for the tiled data dimension, while the attribute
- `padding_value` specifies a padding value at the boundary on non-perfectly
- divisible dimensions. Padding is optional:
- - If absent, it is UB if the tile does not perfectly divide the dimension.
- - If present, it will pad along high dimensions (high-padding) to make the
- tile complete.
+ The "pack" operation converts a source tensor of rank `n` into a result
+ tensor of rank `n + k` with a tiled and packed layout (maybe with padding)
+ and optionally transposes some dimensions.
----------------
chelini wrote:
"some dimensions." I would clarify that we can only transpose the original tiled loop dimensions. We cannot interchange the new point loop dimensions.
https://github.com/llvm/llvm-project/pull/72201
More information about the Mlir-commits
mailing list