Skip to content

Commit 186ae8c

Browse files
[NFC][mlir][linalg] Make conv_3d_ncdhw_fcdhw consistent with 2D variant (#129547)
Other convolutions such as conv_2d_nchw_fchw have a output affine map with no permutations and the input and the filter map are adjusted accordingly. This makes conv_3d_ncdhw_fcdhw a stand out so this PR changes the affine map to be consistent with other variants.
1 parent da293b8 commit 186ae8c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4072,12 +4072,12 @@ structured_op: !LinalgStructuredOpConfig
40724072
indexing_maps: !LinalgIndexingMapsConfig
40734073
static_indexing_maps:
40744074
- affine_map<(d0, d1, d2, d3, d4, d5, d6, d7, d8)[s0, s1, s2, s3, s4, s5, s6,
4075-
s7, s8, s9, s10, s11, s12, s13, s14] -> (d0, d8, d1 * s3 + d5 * s5, d2 * s7
4076-
+ d6 * s9, d3 * s11 + d7 * s13)>
4075+
s7, s8, s9, s10, s11, s12, s13, s14] -> (d0, d5, d2 * s3 + d6 * s5, d3 * s7
4076+
+ d7 * s9, d4 * s11 + d8 * s13)>
40774077
- affine_map<(d0, d1, d2, d3, d4, d5, d6, d7, d8)[s0, s1, s2, s3, s4, s5, s6,
4078-
s7, s8, s9, s10, s11, s12, s13, s14] -> (d4, d8, d5, d6, d7)>
4078+
s7, s8, s9, s10, s11, s12, s13, s14] -> (d1, d5, d6, d7, d8)>
40794079
- affine_map<(d0, d1, d2, d3, d4, d5, d6, d7, d8)[s0, s1, s2, s3, s4, s5, s6,
4080-
s7, s8, s9, s10, s11, s12, s13, s14] -> (d0, d4, d1, d2, d3)>
4080+
s7, s8, s9, s10, s11, s12, s13, s14] -> (d0, d1, d2, d3, d4)>
40814081
iterator_types:
40824082
- parallel
40834083
- parallel

0 commit comments

Comments
 (0)