56
56
using namespace mlir ;
57
57
using namespace mlir ::linalg;
58
58
59
-
60
59
SmallVector<int64_t > computeInterchangeFromDimPos (ArrayRef<int64_t > dimsPos,
61
- int64_t rank) {
60
+ int64_t rank) {
62
61
SmallVector<int64_t > interchangeVector;
63
62
interchangeVector.reserve (dimsPos.size ());
64
63
// First map dims and their position. For example, dims_pos = [2, 0] will map
@@ -94,7 +93,6 @@ SmallVector<T> interchange(ArrayRef<T> elements,
94
93
return vec;
95
94
}
96
95
97
-
98
96
// / Return a `memref.dim` or `tensor.dim` for the shape of `v` at `dim`.
99
97
static OpFoldResult getDimValue (OpBuilder &builder, Location loc, Value v,
100
98
int64_t dim) {
@@ -4860,9 +4858,8 @@ static void generatePackOpScalarImplementationBody(PackOp packOp,
4860
4858
Value idx =
4861
4859
getValueOrCreateConstantIndexOp (builder, loc, sourceIndices[dim]);
4862
4860
Value dimValue = getValueOrCreateConstantIndexOp (
4863
- builder, loc, getDimValue (builder, loc, packOp.getInput (), dim));
4864
- Value cond = arithBuilder.slt (
4865
- idx, dimValue);
4861
+ builder, loc, getDimValue (builder, loc, packOp.getInput (), dim));
4862
+ Value cond = arithBuilder.slt (idx, dimValue);
4866
4863
isInBounds = dim == 0 ? cond : arithBuilder._and (isInBounds, cond);
4867
4864
}
4868
4865
scalar = builder
@@ -4929,7 +4926,6 @@ LogicalResult PackOp::generateScalarImplementation(OpBuilder &builder,
4929
4926
return success ();
4930
4927
}
4931
4928
4932
-
4933
4929
Value PackOp::createDestinationTensor (OpBuilder &b, Location loc, Value source,
4934
4930
ArrayRef<OpFoldResult> innerTileSizes,
4935
4931
ArrayRef<int64_t > innerDimsPos,
@@ -5266,8 +5262,8 @@ LogicalResult UnPackOp::generateScalarImplementation(OpBuilder &builder,
5266
5262
}
5267
5263
if (outputShape.size () != 1 || outputShape[0 ].size () != getOutputRank ()) {
5268
5264
return getOperation ()->emitError (
5269
- " expected shape of one result value of rank" )
5270
- << getOutputRank ();
5265
+ " expected shape of one result value of rank" )
5266
+ << getOutputRank ();
5271
5267
}
5272
5268
5273
5269
DenseMap<int64_t , OpFoldResult> dimAndTileMapping = getDimAndTileMapping ();
@@ -5281,7 +5277,7 @@ LogicalResult UnPackOp::generateScalarImplementation(OpBuilder &builder,
5281
5277
affine::DivModValue divMod =
5282
5278
affine::getDivMod (builder, loc, ivs[dim],
5283
5279
getValueOrCreateConstantIndexOp (
5284
- builder, loc, dimAndTileMapping[dim]));
5280
+ builder, loc, dimAndTileMapping[dim]));
5285
5281
inputIvsPointLoops.push_back (divMod.remainder );
5286
5282
inputIvs.push_back (divMod.quotient );
5287
5283
} else {
@@ -5312,7 +5308,6 @@ LogicalResult UnPackOp::generateScalarImplementation(OpBuilder &builder,
5312
5308
return success ();
5313
5309
}
5314
5310
5315
-
5316
5311
LogicalResult
5317
5312
UnPackOp::reifyResultShapes (OpBuilder &builder,
5318
5313
ReifiedRankedShapedTypeDims &reifiedReturnShapes) {
0 commit comments