Skip to content

Commit c578d7b

Browse files
committed
name improvement
1 parent d708a63 commit c578d7b

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

mlir/lib/Dialect/Vector/Transforms/VectorLinearize.cpp

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -228,20 +228,20 @@ struct LinearizeVectorExtractStridedSlice final
228228
"strides other than 1 not supported");
229229
}
230230

231-
ArrayRef<int64_t> inputShape =
232-
extractStridedSliceOp.getSourceVectorType().getShape();
233-
234-
ArrayRef<int64_t> outputType = extractStridedSliceOp.getType().getShape();
235-
236-
auto maybeIntOffsets =
231+
FailureOr<SmallVector<int64_t>> offsets =
237232
intsFromArrayAttr(extractStridedSliceOp.getOffsets());
238-
if (failed(maybeIntOffsets)) {
233+
if (failed(offsets)) {
239234
return rewriter.notifyMatchFailure(extractStridedSliceOp,
240235
"failed to get integer offsets");
241236
}
242237

238+
ArrayRef<int64_t> inputShape =
239+
extractStridedSliceOp.getSourceVectorType().getShape();
240+
241+
ArrayRef<int64_t> outputShape = extractStridedSliceOp.getType().getShape();
242+
243243
SmallVector<int64_t> indices = getFlattenedStridedSliceIndices(
244-
outputType, inputShape, maybeIntOffsets.value());
244+
outputShape, inputShape, offsets.value());
245245

246246
Value srcVector = adaptor.getVector();
247247
rewriter.replaceOpWithNewOp<vector::ShuffleOp>(
@@ -299,13 +299,14 @@ struct LinearizeVectorInsertStridedSlice final
299299
ArrayRef<int64_t> outputShape = outputType.getShape();
300300
int64_t nOutputElements = outputType.getNumElements();
301301

302-
auto maybeIntOffsets = intsFromArrayAttr(insertStridedSliceOp.getOffsets());
303-
if (failed(maybeIntOffsets)) {
302+
FailureOr<SmallVector<int64_t>> offsets =
303+
intsFromArrayAttr(insertStridedSliceOp.getOffsets());
304+
if (failed(offsets)) {
304305
return rewriter.notifyMatchFailure(insertStridedSliceOp,
305306
"failed to get integer offsets");
306307
}
307308
SmallVector<int64_t> sliceIndices = getFlattenedStridedSliceIndices(
308-
inputShape, outputShape, maybeIntOffsets.value());
309+
inputShape, outputShape, offsets.value());
309310

310311
SmallVector<int64_t> indices(nOutputElements, 0);
311312
std::iota(indices.begin(), indices.end(), 0);

0 commit comments

Comments
 (0)