Skip to content

Commit 260e45c

Browse files
authored
[mlir][sparse] fix stack UAF (#79353)
1 parent e3ee376 commit 260e45c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

mlir/lib/Dialect/SparseTensor/Transforms/Utils/SparseTensorLevel.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -862,14 +862,13 @@ ValueRange FilterIterator::forward(OpBuilder &b, Location l) {
862862

863863
SmallVector<Value> whileArgs(getItVals().begin(), getItVals().end());
864864
whileArgs.push_back(isFirst);
865-
866865
auto whileOp = b.create<scf::WhileOp>(
867866
l, ValueRange(whileArgs).getTypes(), whileArgs,
868867
/*beforeBuilder=*/
869868
[this](OpBuilder &b, Location l, ValueRange ivs) {
870869
ValueRange isFirst = linkNewScope(ivs);
871870
assert(isFirst.size() == 1);
872-
ValueRange cont =
871+
scf::ValueVector cont =
873872
genWhenInBound(b, l, *wrap, C_FALSE,
874873
[this, isFirst](OpBuilder &b, Location l,
875874
Value wrapCrd) -> scf::ValueVector {

0 commit comments

Comments
 (0)