Skip to content

Commit 68415e2

Browse files
authored
[MLIR][Linalg] fix linalg.pack description (#167045)
Fix formula describing the mapping from inner_tiles to result shape. When the comment says "tiles are added to the end of the result tensor" it is meant to say tile dimensions are after the dimensions that have been tiled.
1 parent 91e6dee commit 68415e2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/include/mlir/Dialect/Linalg/IR/LinalgRelayoutOps.td

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def Linalg_PackOp : Linalg_RelayoutOp<"pack", [
109109
within [0, n).
110110
- The tiled dimensions (of size `inner_tiles`) are added to the end of the
111111
result tensor in the order in which they appear, i.e.
112-
`shape(result)[rank(result) + i] = inner_tiles[i]` for `0 <= i < k`.
112+
`shape(result)[rank(source) + i] = inner_tiles[i]` for `0 <= i < k`.
113113
- The following relationship for the tiled dimensions holds:
114114
`shape(result)[inner_dims_pos[i]] = shape(source)[inner_dims_pos[i]] / inner_tiles[i]`,
115115
where (⌈/⌉ indicates CeilDiv).

0 commit comments

Comments
 (0)