Skip to content

Commit caf1fea

Browse files
LuhaocongDebadri Basak
authored andcommitted
[mlir][vector] Fix missed return in ExtractStridedSliceOp::fold (llvm#165669)
Fix missed `return` when folding splat ConstantOp, it could work well probably because of good compatibility of `foldExtractStridedSliceNonSplatConstant`.
1 parent 1b399fd commit caf1fea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/lib/Dialect/Vector/IR/VectorOps.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4336,7 +4336,7 @@ OpFoldResult ExtractStridedSliceOp::fold(FoldAdaptor adaptor) {
43364336
// ExtractStridedSliceOp(splat ConstantOp) -> ConstantOp.
43374337
if (auto splat =
43384338
llvm::dyn_cast_if_present<SplatElementsAttr>(adaptor.getSource()))
4339-
DenseElementsAttr::get(getType(), splat.getSplatValue<Attribute>());
4339+
return DenseElementsAttr::get(getType(), splat.getSplatValue<Attribute>());
43404340

43414341
// ExtractStridedSliceOp(non-splat ConstantOp) -> ConstantOp.
43424342
return foldExtractStridedSliceNonSplatConstant(*this, adaptor.getSource());

0 commit comments

Comments
 (0)