Skip to content

Commit e5f1fe5

Browse files
author
Hariprasad Ravishankar
committed
Fix deprecated rewriter.create usages
1 parent da0f39c commit e5f1fe5

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

lib/Conversion/TorchToLinalg/Linear.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1628,23 +1628,23 @@ Value ConvertAtenConvolutionOp::createTransposedInputPadding(
16281628
Value outerDim = outerSizes[i + 2];
16291629
Value isPadding = rewriter.createOrFold<arith::CmpIOp>(
16301630
loc, arith::CmpIPredicate::ugt, outerDim, innerDim);
1631-
Value maxDim = rewriter.createOrFold<arith::SelectOp>(loc, isPadding,
1632-
outerDim, innerDim);
1631+
Value maxDim =
1632+
rewriter.createOrFold<arith::SelectOp>(loc, isPadding, outerDim, innerDim);
16331633
maxSizes.push_back(maxDim);
16341634
}
16351635

16361636
Value initMaxTensor =
16371637
createInitTensor(rewriter, loc, maxSizes, inputDTy, pad);
16381638

16391639
// Insert input
1640-
auto paddedTensor = rewriter.create<tensor::InsertSliceOp>(
1640+
auto paddedTensor = tensor::InsertSliceOp::create(rewriter,
16411641
loc, torch_to_linalg::removeSizeInformation(rewriter, loc, input),
16421642
initMaxTensor, insertSliceOffsets, inputSizes, strideIndexValues);
16431643

16441644
SmallVector<Value> allOnesStrides(inputSizes.size(), c1);
16451645

16461646
// Crop. Extract the final tensor from the "max" tensor
1647-
auto finalTensor = rewriter.create<tensor::ExtractSliceOp>(
1647+
auto finalTensor = tensor::ExtractSliceOp::create(rewriter,
16481648
loc,
16491649
torch_to_linalg::removeSizeInformation(rewriter, loc, paddedTensor),
16501650
extractSliceOffsets, outerSizes, allOnesStrides);
@@ -1657,7 +1657,7 @@ Value ConvertAtenConvolutionOp::createTransposedInputPadding(
16571657
createInitTensor(rewriter, loc, outerSizes, inputDTy, pad);
16581658

16591659
// Insert the original input into the outer tensor with calculated offsets
1660-
auto paddedInput = rewriter.create<tensor::InsertSliceOp>(
1660+
auto paddedInput = tensor::InsertSliceOp::create(rewriter,
16611661
loc, torch_to_linalg::removeSizeInformation(rewriter, loc, input),
16621662
initPaddedTensor, insertSliceOffsets, inputSizes, strideIndexValues);
16631663
return paddedInput;

0 commit comments

Comments
 (0)