[mlir][vector] Fix missed return in ExtractStridedSliceOp::fold

[Luhaocong]

Fix missed return when folding splat ConstantOp, it could work well probably because of good compatibility of foldExtractStridedSliceNonSplatConstant.

[llvmbot]

@llvm/pr-subscribers-mlir-vector

@llvm/pr-subscribers-mlir

Author: Haocong Lu (Luhaocong)

Changes

Fix missed return when folding splat ConstantOp, it could work well probably because of good compatibility of foldExtractStridedSliceNonSplatConstant.

---

Full diff: https://github.com/llvm/llvm-project/pull/165669.diff

1 Files Affected:

• (modified) mlir/lib/Dialect/Vector/IR/VectorOps.cpp (+1-1)

diff --git a/mlir/lib/Dialect/Vector/IR/VectorOps.cpp b/mlir/lib/Dialect/Vector/IR/VectorOps.cpp
index 45c54c7587c69..cb94e9f92b917 100644
--- a/mlir/lib/Dialect/Vector/IR/VectorOps.cpp
+++ b/mlir/lib/Dialect/Vector/IR/VectorOps.cpp
@@ -4336,7 +4336,7 @@ OpFoldResult ExtractStridedSliceOp::fold(FoldAdaptor adaptor) {
   // ExtractStridedSliceOp(splat ConstantOp) -> ConstantOp.
   if (auto splat =
           llvm::dyn_cast_if_present<SplatElementsAttr>(adaptor.getSource()))
-    DenseElementsAttr::get(getType(), splat.getSplatValue<Attribute>());
+    return DenseElementsAttr::get(getType(), splat.getSplatValue<Attribute>());
 
   // ExtractStridedSliceOp(non-splat ConstantOp) -> ConstantOp.
   return foldExtractStridedSliceNonSplatConstant(*this, adaptor.getSource());