forked from OSchip/llvm-project
Remove useless nesting blok and dead return statement in TosaToLinalg.cpp (NFC)
Flagged by Coverity.
This commit is contained in:
Mehdi Amini
parent
a1e62aa75b
commit
e4e463e747
|
|
@ -1431,248 +1431,240 @@ public:
|
||||||
getNParallelLoopsAttrs(resultTy.getRank()));
|
getNParallelLoopsAttrs(resultTy.getRank()));
|
||||||
rewriter.replaceOp(op, genericOp.getResult(0));
|
rewriter.replaceOp(op, genericOp.getResult(0));
|
||||||
|
|
||||||
{
|
OpBuilder::InsertionGuard regionGuard(rewriter);
|
||||||
OpBuilder::InsertionGuard regionGuard(rewriter);
|
rewriter.createBlock(&genericOp.region(), genericOp.region().end(),
|
||||||
rewriter.createBlock(&genericOp.region(), genericOp.region().end(),
|
TypeRange({resultElementTy}));
|
||||||
TypeRange({resultElementTy}));
|
Value batch = rewriter.create<linalg::IndexOp>(loc, 0);
|
||||||
Value batch = rewriter.create<linalg::IndexOp>(loc, 0);
|
Value y = rewriter.create<linalg::IndexOp>(loc, 1);
|
||||||
Value y = rewriter.create<linalg::IndexOp>(loc, 1);
|
Value x = rewriter.create<linalg::IndexOp>(loc, 2);
|
||||||
Value x = rewriter.create<linalg::IndexOp>(loc, 2);
|
Value channel = rewriter.create<linalg::IndexOp>(loc, 3);
|
||||||
Value channel = rewriter.create<linalg::IndexOp>(loc, 3);
|
|
||||||
|
|
||||||
auto hwMin = rewriter.create<arith::ConstantOp>(
|
auto hwMin =
|
||||||
loc, rewriter.getI32IntegerAttr(0));
|
rewriter.create<arith::ConstantOp>(loc, rewriter.getI32IntegerAttr(0));
|
||||||
auto hMax = rewriter.create<arith::ConstantOp>(
|
auto hMax = rewriter.create<arith::ConstantOp>(
|
||||||
loc, rewriter.getI32IntegerAttr(imageH - 1));
|
loc, rewriter.getI32IntegerAttr(imageH - 1));
|
||||||
auto wMax = rewriter.create<arith::ConstantOp>(
|
auto wMax = rewriter.create<arith::ConstantOp>(
|
||||||
loc, rewriter.getI32IntegerAttr(imageW - 1));
|
loc, rewriter.getI32IntegerAttr(imageW - 1));
|
||||||
|
|
||||||
Value inY =
|
Value inY =
|
||||||
rewriter.create<arith::IndexCastOp>(loc, rewriter.getI32Type(), y);
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getI32Type(), y);
|
||||||
Value inX =
|
Value inX =
|
||||||
rewriter.create<arith::IndexCastOp>(loc, rewriter.getI32Type(), x);
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getI32Type(), x);
|
||||||
|
|
||||||
int32_t shift = op.shift();
|
int32_t shift = op.shift();
|
||||||
bool floatingPointMode = shift == 0;
|
bool floatingPointMode = shift == 0;
|
||||||
|
|
||||||
Value yStride, xStride, yOffset, xOffset;
|
Value yStride, xStride, yOffset, xOffset;
|
||||||
if (floatingPointMode) {
|
if (floatingPointMode) {
|
||||||
yStride = rewriter.create<arith::ConstantOp>(loc, op.stride_fp()[0]);
|
yStride = rewriter.create<arith::ConstantOp>(loc, op.stride_fp()[0]);
|
||||||
xStride = rewriter.create<arith::ConstantOp>(loc, op.stride_fp()[1]);
|
xStride = rewriter.create<arith::ConstantOp>(loc, op.stride_fp()[1]);
|
||||||
yOffset = rewriter.create<arith::ConstantOp>(loc, op.offset_fp()[0]);
|
yOffset = rewriter.create<arith::ConstantOp>(loc, op.offset_fp()[0]);
|
||||||
xOffset = rewriter.create<arith::ConstantOp>(loc, op.offset_fp()[1]);
|
xOffset = rewriter.create<arith::ConstantOp>(loc, op.offset_fp()[1]);
|
||||||
} else {
|
} else {
|
||||||
SmallVector<int32_t> stride, offset;
|
SmallVector<int32_t> stride, offset;
|
||||||
getValuesFromIntArrayAttribute(op.stride(), stride);
|
getValuesFromIntArrayAttribute(op.stride(), stride);
|
||||||
getValuesFromIntArrayAttribute(op.offset(), offset);
|
getValuesFromIntArrayAttribute(op.offset(), offset);
|
||||||
|
|
||||||
yStride = rewriter.create<arith::ConstantOp>(
|
yStride = rewriter.create<arith::ConstantOp>(
|
||||||
loc, rewriter.getI32IntegerAttr(stride[0]));
|
loc, rewriter.getI32IntegerAttr(stride[0]));
|
||||||
xStride = rewriter.create<arith::ConstantOp>(
|
xStride = rewriter.create<arith::ConstantOp>(
|
||||||
loc, rewriter.getI32IntegerAttr(stride[1]));
|
loc, rewriter.getI32IntegerAttr(stride[1]));
|
||||||
yOffset = rewriter.create<arith::ConstantOp>(
|
yOffset = rewriter.create<arith::ConstantOp>(
|
||||||
loc, rewriter.getI32IntegerAttr(offset[0]));
|
loc, rewriter.getI32IntegerAttr(offset[0]));
|
||||||
xOffset = rewriter.create<arith::ConstantOp>(
|
xOffset = rewriter.create<arith::ConstantOp>(
|
||||||
loc, rewriter.getI32IntegerAttr(offset[1]));
|
loc, rewriter.getI32IntegerAttr(offset[1]));
|
||||||
}
|
|
||||||
|
|
||||||
// Compute the the integer index and partial offset.
|
|
||||||
// x = x * stride + offset;
|
|
||||||
// ix = floor(x)
|
|
||||||
// dx = x - ix
|
|
||||||
Value ix, iy, dx, dy;
|
|
||||||
if (floatingPointMode) {
|
|
||||||
Value y =
|
|
||||||
rewriter.create<arith::UIToFPOp>(loc, rewriter.getF32Type(), inY);
|
|
||||||
Value x =
|
|
||||||
rewriter.create<arith::UIToFPOp>(loc, rewriter.getF32Type(), inX);
|
|
||||||
|
|
||||||
y = rewriter.create<arith::MulFOp>(loc, y, yStride);
|
|
||||||
x = rewriter.create<arith::MulFOp>(loc, x, xStride);
|
|
||||||
|
|
||||||
y = rewriter.create<arith::AddFOp>(loc, y, yOffset);
|
|
||||||
x = rewriter.create<arith::AddFOp>(loc, x, xOffset);
|
|
||||||
|
|
||||||
iy = rewriter.create<math::FloorOp>(loc, y);
|
|
||||||
ix = rewriter.create<math::FloorOp>(loc, x);
|
|
||||||
|
|
||||||
dy = rewriter.create<arith::SubFOp>(loc, y, iy);
|
|
||||||
dx = rewriter.create<arith::SubFOp>(loc, x, ix);
|
|
||||||
|
|
||||||
iy = rewriter.create<arith::FPToSIOp>(loc, rewriter.getI32Type(), iy);
|
|
||||||
ix = rewriter.create<arith::FPToSIOp>(loc, rewriter.getI32Type(), ix);
|
|
||||||
} else {
|
|
||||||
Value shiftVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getI32IntegerAttr(shift));
|
|
||||||
|
|
||||||
Value y = rewriter.create<arith::MulIOp>(loc, inY, yStride);
|
|
||||||
Value x = rewriter.create<arith::MulIOp>(loc, inX, xStride);
|
|
||||||
|
|
||||||
y = rewriter.create<arith::AddIOp>(loc, y, yOffset);
|
|
||||||
x = rewriter.create<arith::AddIOp>(loc, x, xOffset);
|
|
||||||
|
|
||||||
iy = rewriter.create<arith::ShRSIOp>(loc, y, shiftVal);
|
|
||||||
ix = rewriter.create<arith::ShRSIOp>(loc, x, shiftVal);
|
|
||||||
|
|
||||||
Value yTrunc = rewriter.create<arith::ShLIOp>(loc, iy, shiftVal);
|
|
||||||
Value xTrunc = rewriter.create<arith::ShLIOp>(loc, ix, shiftVal);
|
|
||||||
|
|
||||||
dy = rewriter.create<arith::SubIOp>(loc, y, yTrunc);
|
|
||||||
dx = rewriter.create<arith::SubIOp>(loc, x, xTrunc);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (op.mode() == "NEAREST_NEIGHBOR") {
|
|
||||||
Value yPred, xPred;
|
|
||||||
// Round the index position towards the closest pixel location.
|
|
||||||
if (floatingPointMode) {
|
|
||||||
auto halfVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getF32FloatAttr(0.5f));
|
|
||||||
yPred = rewriter.create<arith::CmpFOp>(loc, arith::CmpFPredicate::OGE,
|
|
||||||
dy, halfVal);
|
|
||||||
xPred = rewriter.create<arith::CmpFOp>(loc, arith::CmpFPredicate::OGE,
|
|
||||||
dx, halfVal);
|
|
||||||
} else {
|
|
||||||
auto halfVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getI32IntegerAttr(1 << (shift - 1)));
|
|
||||||
yPred = rewriter.create<arith::CmpIOp>(loc, arith::CmpIPredicate::sge,
|
|
||||||
dy, halfVal);
|
|
||||||
xPred = rewriter.create<arith::CmpIOp>(loc, arith::CmpIPredicate::sge,
|
|
||||||
dx, halfVal);
|
|
||||||
}
|
|
||||||
|
|
||||||
auto zeroVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getI32IntegerAttr(0));
|
|
||||||
auto oneVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getI32IntegerAttr(1));
|
|
||||||
|
|
||||||
auto yOffset =
|
|
||||||
rewriter.create<mlir::SelectOp>(loc, yPred, oneVal, zeroVal);
|
|
||||||
auto xOffset =
|
|
||||||
rewriter.create<mlir::SelectOp>(loc, xPred, oneVal, zeroVal);
|
|
||||||
|
|
||||||
iy = rewriter.create<arith::AddIOp>(loc, iy, yOffset);
|
|
||||||
ix = rewriter.create<arith::AddIOp>(loc, ix, xOffset);
|
|
||||||
|
|
||||||
// Clamp the to be within the bounds of the input image.
|
|
||||||
|
|
||||||
iy = clampHelper<arith::CmpIOp>(loc, iy, hwMin, hMax,
|
|
||||||
arith::CmpIPredicate::slt, rewriter);
|
|
||||||
ix = clampHelper<arith::CmpIOp>(loc, ix, hwMin, wMax,
|
|
||||||
arith::CmpIPredicate::slt, rewriter);
|
|
||||||
|
|
||||||
// Read the value from the input array.
|
|
||||||
iy = rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(),
|
|
||||||
iy);
|
|
||||||
ix = rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(),
|
|
||||||
ix);
|
|
||||||
|
|
||||||
Value result = rewriter.create<tensor::ExtractOp>(
|
|
||||||
loc, input, ValueRange{batch, iy, ix, channel});
|
|
||||||
|
|
||||||
rewriter.create<linalg::YieldOp>(loc, result);
|
|
||||||
|
|
||||||
return success();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (op.mode() == "BILINEAR") {
|
|
||||||
Value y0 = iy;
|
|
||||||
Value x0 = ix;
|
|
||||||
|
|
||||||
auto oneVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getI32IntegerAttr(1));
|
|
||||||
Value y1 = rewriter.create<arith::AddIOp>(loc, y0, oneVal);
|
|
||||||
Value x1 = rewriter.create<arith::AddIOp>(loc, x0, oneVal);
|
|
||||||
|
|
||||||
y0 = clampHelper<arith::CmpIOp>(loc, y0, hwMin, hMax,
|
|
||||||
arith::CmpIPredicate::slt, rewriter);
|
|
||||||
y1 = clampHelper<arith::CmpIOp>(loc, y1, hwMin, hMax,
|
|
||||||
arith::CmpIPredicate::slt, rewriter);
|
|
||||||
|
|
||||||
x0 = clampHelper<arith::CmpIOp>(loc, x0, hwMin, wMax,
|
|
||||||
arith::CmpIPredicate::slt, rewriter);
|
|
||||||
x1 = clampHelper<arith::CmpIOp>(loc, x1, hwMin, wMax,
|
|
||||||
arith::CmpIPredicate::slt, rewriter);
|
|
||||||
|
|
||||||
y0 = rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(),
|
|
||||||
y0);
|
|
||||||
y1 = rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(),
|
|
||||||
y1);
|
|
||||||
x0 = rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(),
|
|
||||||
x0);
|
|
||||||
x1 = rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(),
|
|
||||||
x1);
|
|
||||||
|
|
||||||
Value y0x0 = rewriter.create<tensor::ExtractOp>(
|
|
||||||
loc, input, ValueRange{batch, y0, x0, channel});
|
|
||||||
Value y0x1 = rewriter.create<tensor::ExtractOp>(
|
|
||||||
loc, input, ValueRange{batch, y0, x1, channel});
|
|
||||||
Value y1x0 = rewriter.create<tensor::ExtractOp>(
|
|
||||||
loc, input, ValueRange{batch, y1, x0, channel});
|
|
||||||
Value y1x1 = rewriter.create<tensor::ExtractOp>(
|
|
||||||
loc, input, ValueRange{batch, y1, x1, channel});
|
|
||||||
|
|
||||||
if (floatingPointMode) {
|
|
||||||
auto oneVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getF32FloatAttr(1.f));
|
|
||||||
Value rightPart = dx;
|
|
||||||
Value leftPart = rewriter.create<arith::SubFOp>(loc, oneVal, dx);
|
|
||||||
|
|
||||||
y0x0 = rewriter.create<arith::MulFOp>(loc, y0x0, leftPart);
|
|
||||||
y0x1 = rewriter.create<arith::MulFOp>(loc, y0x1, rightPart);
|
|
||||||
Value topAcc = rewriter.create<arith::AddFOp>(loc, y0x0, y0x1);
|
|
||||||
|
|
||||||
y1x0 = rewriter.create<arith::MulFOp>(loc, y1x0, leftPart);
|
|
||||||
y1x1 = rewriter.create<arith::MulFOp>(loc, y1x1, rightPart);
|
|
||||||
Value bottomAcc = rewriter.create<arith::AddFOp>(loc, y1x0, y1x1);
|
|
||||||
|
|
||||||
Value bottomPart = dy;
|
|
||||||
Value topPart = rewriter.create<arith::SubFOp>(loc, oneVal, dy);
|
|
||||||
topAcc = rewriter.create<arith::MulFOp>(loc, topAcc, topPart);
|
|
||||||
bottomAcc =
|
|
||||||
rewriter.create<arith::MulFOp>(loc, bottomAcc, bottomPart);
|
|
||||||
Value result = rewriter.create<arith::AddFOp>(loc, topAcc, bottomAcc);
|
|
||||||
|
|
||||||
rewriter.create<linalg::YieldOp>(loc, result);
|
|
||||||
return success();
|
|
||||||
}
|
|
||||||
y0x0 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y0x0);
|
|
||||||
y0x1 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y0x1);
|
|
||||||
y1x0 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y1x0);
|
|
||||||
y1x1 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y1x1);
|
|
||||||
|
|
||||||
if (resultElementTy.getIntOrFloatBitWidth() > 32) {
|
|
||||||
dx = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, dx);
|
|
||||||
dy = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, dy);
|
|
||||||
}
|
|
||||||
|
|
||||||
auto unitVal = rewriter.create<arith::ConstantOp>(
|
|
||||||
loc, rewriter.getIntegerAttr(resultElementTy, 1 << shift));
|
|
||||||
Value rightPart = dx;
|
|
||||||
Value leftPart = rewriter.create<arith::SubIOp>(loc, unitVal, dx);
|
|
||||||
|
|
||||||
y0x0 = rewriter.create<arith::MulIOp>(loc, y0x0, leftPart);
|
|
||||||
y0x1 = rewriter.create<arith::MulIOp>(loc, y0x1, rightPart);
|
|
||||||
Value topAcc = rewriter.create<arith::AddIOp>(loc, y0x0, y0x1);
|
|
||||||
|
|
||||||
y1x0 = rewriter.create<arith::MulIOp>(loc, y1x0, leftPart);
|
|
||||||
y1x1 = rewriter.create<arith::MulIOp>(loc, y1x1, rightPart);
|
|
||||||
Value bottomAcc = rewriter.create<arith::AddIOp>(loc, y1x0, y1x1);
|
|
||||||
|
|
||||||
Value bottomPart = dy;
|
|
||||||
Value topPart = rewriter.create<arith::SubIOp>(loc, unitVal, dy);
|
|
||||||
topAcc = rewriter.create<arith::MulIOp>(loc, topAcc, topPart);
|
|
||||||
bottomAcc =
|
|
||||||
rewriter.create<arith::MulIOp>(loc, bottomAcc, bottomPart);
|
|
||||||
Value result = rewriter.create<arith::AddIOp>(loc, topAcc, bottomAcc);
|
|
||||||
|
|
||||||
rewriter.create<linalg::YieldOp>(loc, result);
|
|
||||||
return success();
|
|
||||||
}
|
|
||||||
|
|
||||||
return failure();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return success();
|
// Compute the the integer index and partial offset.
|
||||||
|
// x = x * stride + offset;
|
||||||
|
// ix = floor(x)
|
||||||
|
// dx = x - ix
|
||||||
|
Value ix, iy, dx, dy;
|
||||||
|
if (floatingPointMode) {
|
||||||
|
Value y =
|
||||||
|
rewriter.create<arith::UIToFPOp>(loc, rewriter.getF32Type(), inY);
|
||||||
|
Value x =
|
||||||
|
rewriter.create<arith::UIToFPOp>(loc, rewriter.getF32Type(), inX);
|
||||||
|
|
||||||
|
y = rewriter.create<arith::MulFOp>(loc, y, yStride);
|
||||||
|
x = rewriter.create<arith::MulFOp>(loc, x, xStride);
|
||||||
|
|
||||||
|
y = rewriter.create<arith::AddFOp>(loc, y, yOffset);
|
||||||
|
x = rewriter.create<arith::AddFOp>(loc, x, xOffset);
|
||||||
|
|
||||||
|
iy = rewriter.create<math::FloorOp>(loc, y);
|
||||||
|
ix = rewriter.create<math::FloorOp>(loc, x);
|
||||||
|
|
||||||
|
dy = rewriter.create<arith::SubFOp>(loc, y, iy);
|
||||||
|
dx = rewriter.create<arith::SubFOp>(loc, x, ix);
|
||||||
|
|
||||||
|
iy = rewriter.create<arith::FPToSIOp>(loc, rewriter.getI32Type(), iy);
|
||||||
|
ix = rewriter.create<arith::FPToSIOp>(loc, rewriter.getI32Type(), ix);
|
||||||
|
} else {
|
||||||
|
Value shiftVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getI32IntegerAttr(shift));
|
||||||
|
|
||||||
|
Value y = rewriter.create<arith::MulIOp>(loc, inY, yStride);
|
||||||
|
Value x = rewriter.create<arith::MulIOp>(loc, inX, xStride);
|
||||||
|
|
||||||
|
y = rewriter.create<arith::AddIOp>(loc, y, yOffset);
|
||||||
|
x = rewriter.create<arith::AddIOp>(loc, x, xOffset);
|
||||||
|
|
||||||
|
iy = rewriter.create<arith::ShRSIOp>(loc, y, shiftVal);
|
||||||
|
ix = rewriter.create<arith::ShRSIOp>(loc, x, shiftVal);
|
||||||
|
|
||||||
|
Value yTrunc = rewriter.create<arith::ShLIOp>(loc, iy, shiftVal);
|
||||||
|
Value xTrunc = rewriter.create<arith::ShLIOp>(loc, ix, shiftVal);
|
||||||
|
|
||||||
|
dy = rewriter.create<arith::SubIOp>(loc, y, yTrunc);
|
||||||
|
dx = rewriter.create<arith::SubIOp>(loc, x, xTrunc);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (op.mode() == "NEAREST_NEIGHBOR") {
|
||||||
|
Value yPred, xPred;
|
||||||
|
// Round the index position towards the closest pixel location.
|
||||||
|
if (floatingPointMode) {
|
||||||
|
auto halfVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getF32FloatAttr(0.5f));
|
||||||
|
yPred = rewriter.create<arith::CmpFOp>(loc, arith::CmpFPredicate::OGE,
|
||||||
|
dy, halfVal);
|
||||||
|
xPred = rewriter.create<arith::CmpFOp>(loc, arith::CmpFPredicate::OGE,
|
||||||
|
dx, halfVal);
|
||||||
|
} else {
|
||||||
|
auto halfVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getI32IntegerAttr(1 << (shift - 1)));
|
||||||
|
yPred = rewriter.create<arith::CmpIOp>(loc, arith::CmpIPredicate::sge,
|
||||||
|
dy, halfVal);
|
||||||
|
xPred = rewriter.create<arith::CmpIOp>(loc, arith::CmpIPredicate::sge,
|
||||||
|
dx, halfVal);
|
||||||
|
}
|
||||||
|
|
||||||
|
auto zeroVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getI32IntegerAttr(0));
|
||||||
|
auto oneVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getI32IntegerAttr(1));
|
||||||
|
|
||||||
|
auto yOffset =
|
||||||
|
rewriter.create<mlir::SelectOp>(loc, yPred, oneVal, zeroVal);
|
||||||
|
auto xOffset =
|
||||||
|
rewriter.create<mlir::SelectOp>(loc, xPred, oneVal, zeroVal);
|
||||||
|
|
||||||
|
iy = rewriter.create<arith::AddIOp>(loc, iy, yOffset);
|
||||||
|
ix = rewriter.create<arith::AddIOp>(loc, ix, xOffset);
|
||||||
|
|
||||||
|
// Clamp the to be within the bounds of the input image.
|
||||||
|
|
||||||
|
iy = clampHelper<arith::CmpIOp>(loc, iy, hwMin, hMax,
|
||||||
|
arith::CmpIPredicate::slt, rewriter);
|
||||||
|
ix = clampHelper<arith::CmpIOp>(loc, ix, hwMin, wMax,
|
||||||
|
arith::CmpIPredicate::slt, rewriter);
|
||||||
|
|
||||||
|
// Read the value from the input array.
|
||||||
|
iy =
|
||||||
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(), iy);
|
||||||
|
ix =
|
||||||
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(), ix);
|
||||||
|
|
||||||
|
Value result = rewriter.create<tensor::ExtractOp>(
|
||||||
|
loc, input, ValueRange{batch, iy, ix, channel});
|
||||||
|
|
||||||
|
rewriter.create<linalg::YieldOp>(loc, result);
|
||||||
|
|
||||||
|
return success();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (op.mode() == "BILINEAR") {
|
||||||
|
Value y0 = iy;
|
||||||
|
Value x0 = ix;
|
||||||
|
|
||||||
|
auto oneVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getI32IntegerAttr(1));
|
||||||
|
Value y1 = rewriter.create<arith::AddIOp>(loc, y0, oneVal);
|
||||||
|
Value x1 = rewriter.create<arith::AddIOp>(loc, x0, oneVal);
|
||||||
|
|
||||||
|
y0 = clampHelper<arith::CmpIOp>(loc, y0, hwMin, hMax,
|
||||||
|
arith::CmpIPredicate::slt, rewriter);
|
||||||
|
y1 = clampHelper<arith::CmpIOp>(loc, y1, hwMin, hMax,
|
||||||
|
arith::CmpIPredicate::slt, rewriter);
|
||||||
|
|
||||||
|
x0 = clampHelper<arith::CmpIOp>(loc, x0, hwMin, wMax,
|
||||||
|
arith::CmpIPredicate::slt, rewriter);
|
||||||
|
x1 = clampHelper<arith::CmpIOp>(loc, x1, hwMin, wMax,
|
||||||
|
arith::CmpIPredicate::slt, rewriter);
|
||||||
|
|
||||||
|
y0 =
|
||||||
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(), y0);
|
||||||
|
y1 =
|
||||||
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(), y1);
|
||||||
|
x0 =
|
||||||
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(), x0);
|
||||||
|
x1 =
|
||||||
|
rewriter.create<arith::IndexCastOp>(loc, rewriter.getIndexType(), x1);
|
||||||
|
|
||||||
|
Value y0x0 = rewriter.create<tensor::ExtractOp>(
|
||||||
|
loc, input, ValueRange{batch, y0, x0, channel});
|
||||||
|
Value y0x1 = rewriter.create<tensor::ExtractOp>(
|
||||||
|
loc, input, ValueRange{batch, y0, x1, channel});
|
||||||
|
Value y1x0 = rewriter.create<tensor::ExtractOp>(
|
||||||
|
loc, input, ValueRange{batch, y1, x0, channel});
|
||||||
|
Value y1x1 = rewriter.create<tensor::ExtractOp>(
|
||||||
|
loc, input, ValueRange{batch, y1, x1, channel});
|
||||||
|
|
||||||
|
if (floatingPointMode) {
|
||||||
|
auto oneVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getF32FloatAttr(1.f));
|
||||||
|
Value rightPart = dx;
|
||||||
|
Value leftPart = rewriter.create<arith::SubFOp>(loc, oneVal, dx);
|
||||||
|
|
||||||
|
y0x0 = rewriter.create<arith::MulFOp>(loc, y0x0, leftPart);
|
||||||
|
y0x1 = rewriter.create<arith::MulFOp>(loc, y0x1, rightPart);
|
||||||
|
Value topAcc = rewriter.create<arith::AddFOp>(loc, y0x0, y0x1);
|
||||||
|
|
||||||
|
y1x0 = rewriter.create<arith::MulFOp>(loc, y1x0, leftPart);
|
||||||
|
y1x1 = rewriter.create<arith::MulFOp>(loc, y1x1, rightPart);
|
||||||
|
Value bottomAcc = rewriter.create<arith::AddFOp>(loc, y1x0, y1x1);
|
||||||
|
|
||||||
|
Value bottomPart = dy;
|
||||||
|
Value topPart = rewriter.create<arith::SubFOp>(loc, oneVal, dy);
|
||||||
|
topAcc = rewriter.create<arith::MulFOp>(loc, topAcc, topPart);
|
||||||
|
bottomAcc = rewriter.create<arith::MulFOp>(loc, bottomAcc, bottomPart);
|
||||||
|
Value result = rewriter.create<arith::AddFOp>(loc, topAcc, bottomAcc);
|
||||||
|
|
||||||
|
rewriter.create<linalg::YieldOp>(loc, result);
|
||||||
|
return success();
|
||||||
|
}
|
||||||
|
y0x0 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y0x0);
|
||||||
|
y0x1 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y0x1);
|
||||||
|
y1x0 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y1x0);
|
||||||
|
y1x1 = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, y1x1);
|
||||||
|
|
||||||
|
if (resultElementTy.getIntOrFloatBitWidth() > 32) {
|
||||||
|
dx = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, dx);
|
||||||
|
dy = rewriter.create<arith::ExtSIOp>(loc, resultElementTy, dy);
|
||||||
|
}
|
||||||
|
|
||||||
|
auto unitVal = rewriter.create<arith::ConstantOp>(
|
||||||
|
loc, rewriter.getIntegerAttr(resultElementTy, 1 << shift));
|
||||||
|
Value rightPart = dx;
|
||||||
|
Value leftPart = rewriter.create<arith::SubIOp>(loc, unitVal, dx);
|
||||||
|
|
||||||
|
y0x0 = rewriter.create<arith::MulIOp>(loc, y0x0, leftPart);
|
||||||
|
y0x1 = rewriter.create<arith::MulIOp>(loc, y0x1, rightPart);
|
||||||
|
Value topAcc = rewriter.create<arith::AddIOp>(loc, y0x0, y0x1);
|
||||||
|
|
||||||
|
y1x0 = rewriter.create<arith::MulIOp>(loc, y1x0, leftPart);
|
||||||
|
y1x1 = rewriter.create<arith::MulIOp>(loc, y1x1, rightPart);
|
||||||
|
Value bottomAcc = rewriter.create<arith::AddIOp>(loc, y1x0, y1x1);
|
||||||
|
|
||||||
|
Value bottomPart = dy;
|
||||||
|
Value topPart = rewriter.create<arith::SubIOp>(loc, unitVal, dy);
|
||||||
|
topAcc = rewriter.create<arith::MulIOp>(loc, topAcc, topPart);
|
||||||
|
bottomAcc = rewriter.create<arith::MulIOp>(loc, bottomAcc, bottomPart);
|
||||||
|
Value result = rewriter.create<arith::AddIOp>(loc, topAcc, bottomAcc);
|
||||||
|
|
||||||
|
rewriter.create<linalg::YieldOp>(loc, result);
|
||||||
|
return success();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue