Skip to content

Commit

Permalink
[mlir][Tensor] Make TilingInterface implementation only return hand…
Browse files Browse the repository at this point in the history
…le 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
  • Loading branch information
Mahesh Ravishankar committed Mar 20, 2023
1 parent d0e507f commit 411b1d8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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();
Expand Down

0 comments on commit 411b1d8

Please sign in to comment.