Skip to content

Commit 7742c66

Browse files
committed
nits
Signed-off-by: Eric Feng <Eric.Feng@amd.com>
1 parent 49c8bbd commit 7742c66

1 file changed

Lines changed: 5 additions & 8 deletions

File tree

compiler/src/iree/compiler/Codegen/Dialect/GPU/IR/IREEGPUAttrs.cpp

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -980,20 +980,17 @@ static LogicalResult populateCanonicalOffsetsSizesAndStrides(
980980
for (size_t i = 1, e = vtidBasis.size(); i <= e; ++i) {
981981
if (!usedResults.contains(i)) {
982982
unusedBasisProduct *= vtidBasis[i - 1];
983+
unusedLaneIdComponents.push_back(splitLaneId.getResult(i));
984+
unusedBasisSizes.push_back(vtidBasis[i - 1]);
983985
}
984986
}
985987
int64_t overflowSize = broadcastFactor / unusedBasisProduct;
986988

987989
// Collect in most-significant-first order: overflow, then basis entries.
988990
if (overflowSize > 1) {
989-
unusedLaneIdComponents.push_back(splitLaneId.getResult(0));
990-
unusedBasisSizes.push_back(overflowSize);
991-
}
992-
for (size_t i = 1, e = vtidBasis.size(); i <= e; ++i) {
993-
if (!usedResults.contains(i)) {
994-
unusedLaneIdComponents.push_back(splitLaneId.getResult(i));
995-
unusedBasisSizes.push_back(vtidBasis[i - 1]);
996-
}
991+
unusedLaneIdComponents.insert(unusedLaneIdComponents.begin(),
992+
splitLaneId.getResult(0));
993+
unusedBasisSizes.insert(unusedBasisSizes.begin(), overflowSize);
997994
}
998995

999996
assert(llvm::product_of(unusedBasisSizes) == broadcastFactor &&

0 commit comments

Comments
 (0)