File tree 1 file changed +3
-3
lines changed
mlir/lib/Dialect/SCF/Transforms
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -950,10 +950,10 @@ LogicalResult mlir::scf::yieldReplacementForFusedProducer(
950
950
951
951
Location loc = originalOwner->getLoc ();
952
952
// a. collect all init Value to be appended
953
- ArrayRef <unsigned > initNumberList =
953
+ SmallVector <unsigned > initNumberList =
954
954
yieldResultNumber.empty () ? llvm::to_vector (llvm::seq<unsigned >(
955
955
0 , originalOwner->getNumResults ()))
956
- : yieldResultNumber;
956
+ : llvm::to_vector ( yieldResultNumber) ;
957
957
SmallVector<Value> initValueList;
958
958
for (const auto &resultNumber : initNumberList) {
959
959
FailureOr<Value> initValue = tensor::getOrCreateDestination (
@@ -1000,7 +1000,7 @@ LogicalResult mlir::scf::yieldReplacementForFusedProducer(
1000
1000
// on iteration Domain Tile
1001
1001
SmallVector<SmallVector<OpFoldResult>> offsetList, sizesList;
1002
1002
for (const auto &resultNumber : initNumberList) {
1003
- if (resultNumber == fusedProducerInfo. origProducer . getResultNumber () ) {
1003
+ if (resultNumber == sliceResultNumber ) {
1004
1004
offsetList.push_back (sliceOffset);
1005
1005
sizesList.push_back (sliceSizes);
1006
1006
} else {
You can’t perform that action at this time.
0 commit comments