Skip to content

Commit 784df8a

Browse files
committed
Use named op helpers
1 parent 77f414a commit 784df8a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/lib/Dialect/ArmSME/Transforms/TileAllocation.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -169,14 +169,14 @@ static void findDependantOps(Value rootValue,
169169
if (auto branchOp = llvm::dyn_cast<cf::BranchOp>(user)) {
170170
// (CF) Follow branch.
171171
traverseCorrespondingValues(branchOp.getDestOperands(),
172-
user->getSuccessor(0)->getArguments());
172+
branchOp.getDest()->getArguments());
173173
} else if (auto condBranchOp = llvm::dyn_cast<cf::CondBranchOp>(user)) {
174174
// (CF) Follow true branch.
175175
traverseCorrespondingValues(condBranchOp.getTrueOperands(),
176-
user->getSuccessor(0)->getArguments());
176+
condBranchOp.getTrueDest()->getArguments());
177177
// (CF) Follow false branch.
178178
traverseCorrespondingValues(condBranchOp.getFalseOperands(),
179-
user->getSuccessor(1)->getArguments());
179+
condBranchOp.getFalseDest()->getArguments());
180180
} else if (auto loop = llvm::dyn_cast<LoopLikeOpInterface>(user)) {
181181
// (SCF) Follow iter_args of (basic) loops (e.g. for loops).
182182
traverseCorrespondingValues(loop.getInits(), loop.getRegionIterArgs());

0 commit comments

Comments
 (0)