Skip to content

Commit b06ba75

Browse files
Abhishek-VarmaAbhishek-Varma
committed
Updated maps
1 parent 535f7e9 commit b06ba75

File tree

1 file changed

+33
-33
lines changed

1 file changed

+33
-33
lines changed

‎mlir/lib/Dialect/Linalg/Utils/Utils.cpp‎

Lines changed: 33 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -909,9 +909,9 @@ bool isaPoolingNhwcMaxOp(LinalgOp op) {
909909
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
910910
Value yieldVal = yieldOp.getOperand(0);
911911
unsigned iIndex = 0, oIndex = 2;
912-
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1 + d4, d2 + d5, d3)>
913-
// #map3 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d4, d5)>
914-
// #map4 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d2, d3)>
912+
// #map = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1 + d4, d2 + d5, d3)>
913+
// #map1 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d4, d5)>
914+
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d2, d3)>
915915
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
916916
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
917917
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/1, /*oDim=*/2) &&
@@ -931,9 +931,9 @@ bool isaPoolingNhwcMinOp(LinalgOp op) {
931931
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
932932
Value yieldVal = yieldOp.getOperand(0);
933933
unsigned iIndex = 0, oIndex = 2;
934-
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1 + d4, d2 + d5, d3)>
935-
// #map3 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d4, d5)>
936-
// #map4 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d2, d3)>
934+
// #map = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1 + d4, d2 + d5, d3)>
935+
// #map1 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d4, d5)>
936+
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d2, d3)>
937937
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
938938
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
939939
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/1, /*oDim=*/2) &&
@@ -953,9 +953,9 @@ bool isaPoolingNhwcSumOp(LinalgOp op) {
953953
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
954954
Value yieldVal = yieldOp.getOperand(0);
955955
unsigned iIndex = 0, oIndex = 2;
956-
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1 + d4, d2 + d5, d3)>
957-
// #map3 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d4, d5)>
958-
// #map4 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d2, d3)>
956+
// #map = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1 + d4, d2 + d5, d3)>
957+
// #map1 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d4, d5)>
958+
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5) -> (d0, d1, d2, d3)>
959959
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
960960
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
961961
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/1, /*oDim=*/2) &&
@@ -1019,9 +1019,9 @@ bool isaPoolingNcwMaxOp(LinalgOp op) {
10191019
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
10201020
Value yieldVal = yieldOp.getOperand(0);
10211021
unsigned iIndex = 0, oIndex = 2;
1022-
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2 + d3)>
1023-
// #map3 = affine_map<(d0, d1, d2, d3) -> (d3)>
1024-
// #map4 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
1022+
// #map = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2 + d3)>
1023+
// #map1 = affine_map<(d0, d1, d2, d3) -> (d3)>
1024+
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
10251025
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
10261026
matchConvDimExprPattern(indexingMaps, iIndex, 1, oIndex, 1) &&
10271027
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/0, /*oDim=*/2) &&
@@ -1040,9 +1040,9 @@ bool isaPoolingNcwSumOp(LinalgOp op) {
10401040
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
10411041
Value yieldVal = yieldOp.getOperand(0);
10421042
unsigned iIndex = 0, oIndex = 2;
1043-
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2 + d3)>
1044-
// #map3 = affine_map<(d0, d1, d2, d3) -> (d3)>
1045-
// #map4 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
1043+
// #map = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2 + d3)>
1044+
// #map1 = affine_map<(d0, d1, d2, d3) -> (d3)>
1045+
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
10461046
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
10471047
matchConvDimExprPattern(indexingMaps, iIndex, 1, oIndex, 1) &&
10481048
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/0, /*oDim=*/2) &&
@@ -1061,9 +1061,9 @@ bool isaPoolingNwcMaxOp(LinalgOp op) {
10611061
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
10621062
Value yieldVal = yieldOp.getOperand(0);
10631063
unsigned iIndex = 0, oIndex = 2;
1064-
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1 + d3, d2)>
1065-
// #map3 = affine_map<(d0, d1, d2, d3) -> (d3)>
1066-
// #map4 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
1064+
// #map = affine_map<(d0, d1, d2, d3) -> (d0, d1 + d3, d2)>
1065+
// #map1 = affine_map<(d0, d1, d2, d3) -> (d3)>
1066+
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
10671067
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
10681068
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
10691069
matchConvDimExprPattern(indexingMaps, iIndex, 2, oIndex, 2) &&
@@ -1082,9 +1082,9 @@ bool isaPoolingNwcMinOp(LinalgOp op) {
10821082
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
10831083
Value yieldVal = yieldOp.getOperand(0);
10841084
unsigned iIndex = 0, oIndex = 2;
1085-
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1 + d3, d2)>
1086-
// #map3 = affine_map<(d0, d1, d2, d3) -> (d3)>
1087-
// #map4 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
1085+
// #map = affine_map<(d0, d1, d2, d3) -> (d0, d1 + d3, d2)>
1086+
// #map1 = affine_map<(d0, d1, d2, d3) -> (d3)>
1087+
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
10881088
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
10891089
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
10901090
matchConvDimExprPattern(indexingMaps, iIndex, 2, oIndex, 2) &&
@@ -1103,9 +1103,9 @@ bool isaPoolingNwcSumOp(LinalgOp op) {
11031103
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
11041104
Value yieldVal = yieldOp.getOperand(0);
11051105
unsigned iIndex = 0, oIndex = 2;
1106-
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1 + d3, d2)>
1107-
// #map3 = affine_map<(d0, d1, d2, d3) -> (d3)>
1108-
// #map4 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
1106+
// #map = affine_map<(d0, d1, d2, d3) -> (d0, d1 + d3, d2)>
1107+
// #map1 = affine_map<(d0, d1, d2, d3) -> (d3)>
1108+
// #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
11091109
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
11101110
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
11111111
matchConvDimExprPattern(indexingMaps, iIndex, 2, oIndex, 2) &&
@@ -1124,9 +1124,9 @@ bool isaPoolingNdhwcMaxOp(LinalgOp op) {
11241124
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
11251125
Value yieldVal = yieldOp.getOperand(0);
11261126
unsigned iIndex = 0, oIndex = 2;
1127-
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1 + d5, d2 + d6, d3 + d7, d4)>
1128-
// #map3 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d5, d6, d7)>
1129-
// #map4 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1, d2, d3, d4)>
1127+
// #map = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1 + d5, d2 + d6, d3 + d7, d4)>
1128+
// #map1 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d5, d6, d7)>
1129+
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1, d2, d3, d4)>
11301130
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
11311131
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
11321132
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/1, /*oDim=*/2) &&
@@ -1147,9 +1147,9 @@ bool isaPoolingNdhwcMinOp(LinalgOp op) {
11471147
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
11481148
Value yieldVal = yieldOp.getOperand(0);
11491149
unsigned iIndex = 0, oIndex = 2;
1150-
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1 + d5, d2 + d6, d3 + d7, d4)>
1151-
// #map3 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d5, d6, d7)>
1152-
// #map4 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1, d2, d3, d4)>
1150+
// #map = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1 + d5, d2 + d6, d3 + d7, d4)>
1151+
// #map1 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d5, d6, d7)>
1152+
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1, d2, d3, d4)>
11531153
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
11541154
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
11551155
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/1, /*oDim=*/2) &&
@@ -1170,9 +1170,9 @@ bool isaPoolingNdhwcSumOp(LinalgOp op) {
11701170
auto yieldOp = cast<linalg::YieldOp>(body->getTerminator());
11711171
Value yieldVal = yieldOp.getOperand(0);
11721172
unsigned iIndex = 0, oIndex = 2;
1173-
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1 + d5, d2 + d6, d3 + d7, d4)>
1174-
// #map3 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d5, d6, d7)>
1175-
// #map4 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1, d2, d3, d4)>
1173+
// #map = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1 + d5, d2 + d6, d3 + d7, d4)>
1174+
// #map1 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d5, d6, d7)>
1175+
// #map2 = affine_map<(d0, d1, d2, d3, d4, d5, d6, d7) -> (d0, d1, d2, d3, d4)>
11761176
return (matchConvDimExprPattern(indexingMaps, iIndex, 0, oIndex, 0) &&
11771177
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/1, /*fDim=*/0, /*oDim=*/1) &&
11781178
matchConvDimAddExprPattern(indexingMaps, /*iDim=*/2, /*fDim=*/1, /*oDim=*/2) &&

0 commit comments

Comments
 (0)