Skip to content

Commit e91ee96

Browse files
committed
[MLIR][Linalg] fix linalg.pack description
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 dimension that have been tiled.
1 parent 5256db3 commit e91ee96

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
@@ -108,7 +108,7 @@ def Linalg_PackOp : Linalg_RelayoutOp<"pack", [
108108
within [0, n).
109109
- The tiled dimensions (of size `inner_tiles`) are added to the end of the
110110
result tensor in the order in which they appear, i.e.
111-
`shape(result)[rank(result) + i] = inner_tiles[i]` for `0 <= i < k`.
111+
`shape(result)[rank(source) + i] = inner_tiles[i]` for `0 <= i < k`.
112112
- The following relationship for the tiled dimensions holds:
113113
`shape(result)[inner_dims_pos[i]] = shape(source)[inner_dims_pos[i]] / inner_tiles[i]`,
114114
where (⌈/⌉ indicates CeilDiv).

0 commit comments

Comments
 (0)