diff --git a/mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp b/mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp index 0faa29ade8047..cd3a6f680e972 100644 --- a/mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp +++ b/mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp @@ -648,7 +648,7 @@ FailureOr tensor::bubbleUpPadSlice(OpBuilder &b, elseOp = createPadOfExtractSlice(); b.create(loc, castResult(elseOp->getResult(0))); }); - return TilingResult{{result}, SmallVector(result->getResults())}; + return TilingResult{{elseOp}, SmallVector(result->getResults())}; } Operation *newPadOp = createPadOfExtractSlice();