Skip to content

Commit eda4bf3

Browse files
committed
fix CI
1 parent b9450f7 commit eda4bf3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/lib/Dialect/SCF/Transforms/TileUsingInterface.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -950,10 +950,10 @@ LogicalResult mlir::scf::yieldReplacementForFusedProducer(
950950

951951
Location loc = originalOwner->getLoc();
952952
// a. collect all init Value to be appended
953-
ArrayRef<unsigned> initNumberList =
953+
SmallVector<unsigned> initNumberList =
954954
yieldResultNumber.empty() ? llvm::to_vector(llvm::seq<unsigned>(
955955
0, originalOwner->getNumResults()))
956-
: yieldResultNumber;
956+
: llvm::to_vector(yieldResultNumber);
957957
SmallVector<Value> initValueList;
958958
for (const auto &resultNumber : initNumberList) {
959959
FailureOr<Value> initValue = tensor::getOrCreateDestination(
@@ -1000,7 +1000,7 @@ LogicalResult mlir::scf::yieldReplacementForFusedProducer(
10001000
// on iteration Domain Tile
10011001
SmallVector<SmallVector<OpFoldResult>> offsetList, sizesList;
10021002
for (const auto &resultNumber : initNumberList) {
1003-
if (resultNumber == fusedProducerInfo.origProducer.getResultNumber()) {
1003+
if (resultNumber == sliceResultNumber) {
10041004
offsetList.push_back(sliceOffset);
10051005
sizesList.push_back(sliceSizes);
10061006
} else {

0 commit comments

Comments
 (0)