File tree Expand file tree Collapse file tree
compiler/src/iree/compiler/Codegen/Dialect/GPU/IR Expand file tree Collapse file tree Original file line number Diff line number Diff 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 &&
You can’t perform that action at this time.
0 commit comments