Skip to content

Commit f7e1ec7

Browse files
committed
Fix UAF
1 parent 3d91cac commit f7e1ec7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mlir/lib/Conversion/VectorToSCF/VectorToSCF.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1125,8 +1125,8 @@ struct ScalableTransposeTransferWriteConversion
11251125
ArrayRef<OpFoldResult>(*maskDims).drop_front());
11261126
}
11271127

1128-
ValueRange initLoopArgs =
1129-
isTensorOp(writeOp) ? writeOp.getSource() : ValueRange{};
1128+
Value initDest = isTensorOp(writeOp) ? writeOp.getSource() : Value{};
1129+
ValueRange initLoopArgs = initDest ? initDest : ValueRange{};
11301130
auto result = rewriter.create<scf::ForOp>(
11311131
loc, lb, ub, step, initLoopArgs,
11321132
[&](OpBuilder &b, Location loc, Value iv, ValueRange loopIterArgs) {

0 commit comments

Comments
 (0)