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