[Mlir-commits] [mlir] 411b1d8 - [mlir][Tensor] Make `TilingInterface` implementation only return handle to the created `pad` operation.

[Mahesh Ravishankar]

Author: Mahesh Ravishankar
Date: 2023-03-20T21:23:18Z
New Revision: 411b1d8f079533860a990ee615abae3b0e6dbd8b

URL: https://github.com/llvm/llvm-project/commit/411b1d8f079533860a990ee615abae3b0e6dbd8b
DIFF: https://github.com/llvm/llvm-project/commit/411b1d8f079533860a990ee615abae3b0e6dbd8b.diff

LOG: [mlir][Tensor] Make `TilingInterface` implementation only return handle to the created `pad` operation.

Pad tiling implementation only needs to return the tiled pad
operation. The rest of the generated code is related to handling
boundary conditions.

Reviewed By: hanchung

Differential Revision: https://reviews.llvm.org/D146439

Added: 
    

Modified: 
    mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp

Removed: 
    


################################################################################
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<TilingResult> tensor::bubbleUpPadSlice(OpBuilder &b,
           elseOp = createPadOfExtractSlice();
           b.create<scf::YieldOp>(loc, castResult(elseOp->getResult(0)));
         });
-    return TilingResult{{result}, SmallVector<Value>(result->getResults())};
+    return TilingResult{{elseOp}, SmallVector<Value>(result->getResults())};
   }
 
   Operation *newPadOp = createPadOfExtractSlice();