maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir][NFC] Fix typo in VectorOps.cpp

This commit is contained in:
Adrian Kuegel 2021-12-22 12:23:57 +01:00
parent 8a0e35f3a7
commit 35c0080333
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mlir/lib/Dialect/Vector/VectorOps.cpp
View File

@ -1125,11 +1125,11 @@ static Value foldExtractFromBroadcast(ExtractOp extractOp) {
auto getRank = [](Type type) {
return type.isa<VectorType>() ? type.cast<VectorType>().getRank() : 0;
};
unsigned broadcasrSrcRank = getRank(source.getType());
unsigned broadcastSrcRank = getRank(source.getType());
unsigned extractResultRank = getRank(extractOp.getType());
if (extractResultRank < broadcasrSrcRank) {
if (extractResultRank < broadcastSrcRank) {
auto extractPos = extractVector<int64_t>(extractOp.position());
unsigned rankDiff = broadcasrSrcRank - extractResultRank;
unsigned rankDiff = broadcastSrcRank - extractResultRank;
extractPos.erase(
extractPos.begin(),
std::next(extractPos.begin(), extractPos.size() - rankDiff));
@ -1236,12 +1236,12 @@ public:
auto getRank = [](Type type) {
return type.isa<VectorType>() ? type.cast<VectorType>().getRank() : 0;
};
unsigned broadcasrSrcRank = getRank(source.getType());
unsigned broadcastSrcRank = getRank(source.getType());
unsigned extractResultRank = getRank(extractOp.getType());
// We only consider the case where the rank of the source is smaller than
// the rank of the extract dst. The other cases are handled in the folding
// patterns.
if (extractResultRank <= broadcasrSrcRank)
if (extractResultRank <= broadcastSrcRank)
return failure();
rewriter.replaceOpWithNewOp<vector::BroadcastOp>(
extractOp, extractOp.getType(), source);