Skip to content

Commit 03eb768

Browse files
committed
Feedback
1 parent df53db0 commit 03eb768

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -449,10 +449,11 @@ struct LinearizeVectorExtract final
449449
linearizedOffset += offsets[i] * size;
450450
}
451451

452+
Value srcVector = adaptor.getVector();
452453
if (!isa<VectorType>(extractOp.getType())) {
453454
// Scalar case: generate a 1-D extract.
454455
Value result = rewriter.createOrFold<vector::ExtractOp>(
455-
extractOp.getLoc(), adaptor.getVector(), linearizedOffset);
456+
extractOp.getLoc(), srcVector, linearizedOffset);
456457
rewriter.replaceOp(extractOp, result);
457458
return success();
458459
}
@@ -461,8 +462,8 @@ struct LinearizeVectorExtract final
461462

462463
llvm::SmallVector<int64_t, 2> indices(size);
463464
std::iota(indices.begin(), indices.end(), linearizedOffset);
464-
rewriter.replaceOpWithNewOp<vector::ShuffleOp>(
465-
extractOp, dstTy, adaptor.getVector(), adaptor.getVector(), indices);
465+
rewriter.replaceOpWithNewOp<vector::ShuffleOp>(extractOp, dstTy, srcVector,
466+
srcVector, indices);
466467

467468
return success();
468469
}

0 commit comments

Comments
 (0)